Update app.py

app.py

@@ -660,176 +660,12 @@ def display_file_manager():
             if st.button("🗑", key=f"delete_{unique_id}"):
                 os.remove(file)
                 st.rerun()
-
-def main():
-    st.sidebar.markdown("### 🚲BikeAI🏆 Claude and GPT Multi-Agent Research AI")
-    
-    # Initialize text input state if not exists
-    if 'current_text_input' not in st.session_state:
-        st.session_state.current_text_input = ""
-    
-    # Main navigation
-    tab_main = st.radio("Choose Action:", 
-                        ["🎤 Voice Input", "💬 Chat", "📸 Media Gallery", "🔍 Search ArXiv", "📝 File Editor"],
-                        horizontal=True)
-    
-    if tab_main == "🎤 Voice Input":
-        st.subheader("Voice Recognition")
-        
-        # Display speech recognition component with modified JavaScript
-        speech_recognition_html_modified = speech_recognition_html.replace(
-            'window.parent.postMessage({',
-            'window.parent.postMessage({'
-        ).replace(
-            """if (finalTranscript) {
-                        fullTranscript += finalTranscript;
-                        // Send to Streamlit
-                        window.parent.postMessage({
-                            type: 'final_transcript',
-                            text: finalTranscript
-                        }, '*');
-                    }""",
-            """if (finalTranscript) {
-                        fullTranscript += finalTranscript;
-                        // Send to Streamlit
-                        window.parent.postMessage({
-                            type: 'final_transcript',
-                            text: finalTranscript,
-                            updateInput: true
-                        }, '*');
-                        // Also update the text input area if it exists
-                        const textArea = document.querySelector('textarea[aria-label="Message:"]');
-                        if (textArea) {
-                            textArea.value = finalTranscript;
-                            // Trigger input event to update Streamlit
-                            const event = new Event('input', { bubbles: true });
-                            textArea.dispatchEvent(event);
-                        }
-                    }"""
-        )
-        
-        speech_component = st.components.v1.html(speech_recognition_html_modified, height=400)
-        
-        # Handle speech recognition output
-        if speech_component:
-            try:
-                data = speech_component
-                if isinstance(data, dict):
-                    if data.get('type') == 'final_transcript':
-                        text = data.get('text', '').strip()
-                        if text:
-                            st.session_state.last_voice_input = text
-                            # Update the current text input
-                            st.session_state.current_text_input = text
-                            
-                            # Process voice input with AI
-                            st.subheader("AI Response to Voice Input:")
-                            
-                            col1, col2, col3 = st.columns(3)
-                            with col2:
-                                st.write("Claude-3.5 Sonnet:")
-                                try:
-                                    claude_response = process_with_claude(text)
-                                except:
-                                    st.write('Claude 3.5 Sonnet out of tokens.')
-                            with col1:
-                                st.write("GPT-4o Omni:")
-                                try:
-                                    gpt_response = process_with_gpt(text)
-                                except:
-                                    st.write('GPT 4o out of tokens')
-                            with col3:
-                                st.write("Arxiv and Mistral Research:")
-                                with st.spinner("Searching ArXiv..."):
-                                    results = perform_ai_lookup(text)
-                                    st.markdown(results)
-                    
-                    elif data.get('type') == 'clear_transcript':
-                        st.session_state.last_voice_input = ""
-                        st.session_state.current_text_input = ""
-                        st.experimental_rerun()
-                        
-            except Exception as e:
-                st.error(f"Error processing voice input: {e}")
-    
-    elif tab_main == "💬 Chat":
-        # Use the current_text_input value in the text area
-        user_input = st.text_area("Message:", value=st.session_state.current_text_input, height=100)
-        
-        if st.button("Send 📨"):
-            if user_input:
-                if model_choice == "GPT-4o":
-                    gpt_response = process_with_gpt(user_input)
-                elif model_choice == "Claude-3":
-                    claude_response = process_with_claude(user_input)
-                else:  # Both
-                    col1, col2, col3 = st.columns(3)
-                    with col2:
-                        st.subheader("Claude-3.5 Sonnet:")
-                        try:
-                            claude_response = process_with_claude(user_input)
-                        except:
-                            st.write('Claude 3.5 Sonnet out of tokens.')
-                    with col1:
-                        st.subheader("GPT-4o Omni:")
-                        try:
-                            gpt_response = process_with_gpt(user_input)  
-                        except:
-                            st.write('GPT 4o out of tokens')
-                    with col3:
-                        st.subheader("Arxiv and Mistral Research:")
-                        with st.spinner("Searching ArXiv..."):
-                            #results = search_arxiv(user_input)
-                            results = perform_ai_lookup(user_input)
-                            
-                            st.markdown(results)
-                            
-        # Display Chat History
-        st.subheader("Chat History 📜")
-        tab1, tab2 = st.tabs(["Claude History", "GPT-4o History"])
-        
-        with tab1:
-            for chat in st.session_state.chat_history:
-                st.text_area("You:", chat["user"], height=100)
-                st.text_area("Claude:", chat["claude"], height=200)
-                st.markdown(chat["claude"])
-        
-        with tab2:
-            for message in st.session_state.messages:
-                with st.chat_message(message["role"]):
-                    st.markdown(message["content"])
-    
-    elif tab_main == "📸 Media Gallery":
-        create_media_gallery()
-    
-    elif tab_main == "🔍 Search ArXiv":
-        query = st.text_input("Enter your research query:")
-        if query:
-            with st.spinner("Searching ArXiv..."):
-                results = search_arxiv(query)
-                st.markdown(results)
-    
-    elif tab_main == "📝 File Editor":
-        if hasattr(st.session_state, 'current_file'):
-            st.subheader(f"Editing: {st.session_state.current_file}")
-            new_content = st.text_area("Content:", st.session_state.file_content, height=300)
-            if st.button("Save Changes"):
-                with open(st.session_state.current_file, 'w', encoding='utf-8') as file:
-                    file.write(new_content)
-                st.success("File updated successfully!")
-
-    # Always show file manager in sidebar
-    display_file_manager()
-
-if __name__ == "__main__":
-    main()
-
-# Speech Recognition HTML Component
+# First, define the speech recognition HTML template at the global scope, before the main() function
 speech_recognition_html = """
 <!DOCTYPE html>
 <html>
 <head>
-    <title>Continuous Speech Demo</title>
+    <title>Voice Recognition</title>
     <style>
         body { 
             font-family: sans-serif; 
@@ -873,6 +709,8 @@ speech_recognition_html = """
     <div id="output"></div>
 
     <script>
+        let messageTarget = 'textarea[aria-label="Message:"]';
+        
         if (!('webkitSpeechRecognition' in window)) {
             alert('Speech recognition not supported');
         } else {
@@ -889,6 +727,15 @@ speech_recognition_html = """
             recognition.continuous = true;
             recognition.interimResults = true;
 
+            const updateTextArea = (text) => {
+                const textArea = document.querySelector(messageTarget);
+                if (textArea) {
+                    textArea.value = text;
+                    const event = new Event('input', { bubbles: true });
+                    textArea.dispatchEvent(event);
+                }
+            };
+
             // Function to start recognition
             const startRecognition = () => {
                 try {
@@ -919,6 +766,7 @@ speech_recognition_html = """
             clearButton.onclick = () => {
                 fullTranscript = '';
                 output.textContent = '';
+                updateTextArea('');
                 window.parent.postMessage({
                     type: 'clear_transcript',
                 }, '*');
@@ -940,10 +788,11 @@ speech_recognition_html = """
                 if (finalTranscript || (Date.now() - lastUpdateTime > 5000)) {
                     if (finalTranscript) {
                         fullTranscript += finalTranscript;
-                        // Send to Streamlit
+                        updateTextArea(fullTranscript);
                         window.parent.postMessage({
                             type: 'final_transcript',
-                            text: finalTranscript
+                            text: finalTranscript,
+                            updateInput: true
                         }, '*');
                     }
                     lastUpdateTime = Date.now();
@@ -982,6 +831,137 @@ speech_recognition_html = """
 </html>
 """
 
+def main():
+    st.sidebar.markdown("### 🚲BikeAI🏆 Claude and GPT Multi-Agent Research AI")
+    
+    # Initialize text input state if not exists
+    if 'current_text_input' not in st.session_state:
+        st.session_state.current_text_input = ""
+    
+    # Main navigation
+    tab_main = st.radio("Choose Action:", 
+                        ["🎤 Voice Input", "💬 Chat", "📸 Media Gallery", "🔍 Search ArXiv", "📝 File Editor"],
+                        horizontal=True)
+    
+    if tab_main == "🎤 Voice Input":
+        st.subheader("Voice Recognition")
+        
+        # Display speech recognition component
+        speech_component = st.components.v1.html(speech_recognition_html, height=400)
+        
+        # Handle speech recognition output
+        if speech_component:
+            try:
+                data = speech_component
+                if isinstance(data, dict) and data.get('type') == 'final_transcript':
+                    text = data.get('text', '').strip()
+                    if text:
+                        st.session_state.last_voice_input = text
+                        st.session_state.current_text_input = text
+                        
+                        # Process voice input with AI
+                        st.subheader("AI Response to Voice Input:")
+                        
+                        col1, col2, col3 = st.columns(3)
+                        with col2:
+                            st.write("Claude-3.5 Sonnet:")
+                            try:
+                                claude_response = process_with_claude(text)
+                            except:
+                                st.write('Claude 3.5 Sonnet out of tokens.')
+                        with col1:
+                            st.write("GPT-4o Omni:")
+                            try:
+                                gpt_response = process_with_gpt(text)
+                            except:
+                                st.write('GPT 4o out of tokens')
+                        with col3:
+                            st.write("Arxiv and Mistral Research:")
+                            with st.spinner("Searching ArXiv..."):
+                                results = perform_ai_lookup(text)
+                                st.markdown(results)
+                
+                elif isinstance(data, dict) and data.get('type') == 'clear_transcript':
+                    st.session_state.last_voice_input = ""
+                    st.session_state.current_text_input = ""
+                    st.experimental_rerun()
+                    
+            except Exception as e:
+                st.error(f"Error processing voice input: {e}")
+    
+    elif tab_main == "💬 Chat":
+        # Model Selection
+        model_choice = st.sidebar.radio(
+            "Choose AI Model:",
+            ["GPT-4o", "Claude-3", "GPT+Claude+Arxiv"]
+        )
+        
+        # Use the current_text_input value in the text area
+        user_input = st.text_area("Message:", value=st.session_state.current_text_input, height=100)
+        
+        if st.button("Send 📨"):
+            if user_input:
+                if model_choice == "GPT-4o":
+                    gpt_response = process_with_gpt(user_input)
+                elif model_choice == "Claude-3":
+                    claude_response = process_with_claude(user_input)
+                else:  # Both
+                    col1, col2, col3 = st.columns(3)
+                    with col2:
+                        st.subheader("Claude-3.5 Sonnet:")
+                        try:
+                            claude_response = process_with_claude(user_input)
+                        except:
+                            st.write('Claude 3.5 Sonnet out of tokens.')
+                    with col1:
+                        st.subheader("GPT-4o Omni:")
+                        try:
+                            gpt_response = process_with_gpt(user_input)  
+                        except:
+                            st.write('GPT 4o out of tokens')
+                    with col3:
+                        st.subheader("Arxiv and Mistral Research:")
+                        with st.spinner("Searching ArXiv..."):
+                            results = perform_ai_lookup(user_input)
+                            st.markdown(results)
+                            
+        # Display Chat History
+        st.subheader("Chat History 📜")
+        tab1, tab2 = st.tabs(["Claude History", "GPT-4o History"])
+        
+        with tab1:
+            for chat in st.session_state.chat_history:
+                st.text_area("You:", chat["user"], height=100)
+                st.text_area("Claude:", chat["claude"], height=200)
+                st.markdown(chat["claude"])
+        
+        with tab2:
+            for message in st.session_state.messages:
+                with st.chat_message(message["role"]):
+                    st.markdown(message["content"])
+    
+    elif tab_main == "📸 Media Gallery":
+        create_media_gallery()
+    
+    elif tab_main == "🔍 Search ArXiv":
+        query = st.text_input("Enter your research query:")
+        if query:
+            with st.spinner("Searching ArXiv..."):
+                results = search_arxiv(query)
+                st.markdown(results)
+    
+    elif tab_main == "📝 File Editor":
+        if hasattr(st.session_state, 'current_file'):
+            st.subheader(f"Editing: {st.session_state.current_file}")
+            new_content = st.text_area("Content:", st.session_state.file_content, height=300)
+            if st.button("Save Changes"):
+                with open(st.session_state.current_file, 'w', encoding='utf-8') as file:
+                    file.write(new_content)
+                st.success("File updated successfully!")
+
+    # Always show file manager in sidebar
+    display_file_manager()
+
 # Helper Functions
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')