Update app.py

app.py

@@ -1007,115 +1007,115 @@ def main():
     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")
-    
-    # Get transcript from the speech component
-    current_transcript = integrate_speech_component()
-    
-    # Display the transcript in a Streamlit text area
-    st.markdown("### Processed Voice Input:")
-    st.text_area("Voice Transcript", current_transcript, height=100)
-    
-    # Process buttons
-    col1, col2, col3 = st.columns(3)
-    with col1:
-        if st.button("Process with GPT"):
-            st.subheader("GPT-4o Response")
-            gpt_response = process_with_gpt(current_transcript)
-            st.markdown(gpt_response)
-    
-    with col2:
-        if st.button("Process with Claude"):
-            st.subheader("Claude Response")
-            claude_response = process_with_claude(current_transcript)
-            st.markdown(claude_response)
-    
-    with col3:
-        if st.button("Search ArXiv"):
-            st.subheader("ArXiv Search Results")
-            arxiv_results = perform_ai_lookup(current_transcript)
-            st.markdown(arxiv_results)
+        
+    if tab_main == "🎤 Voice Input":
+        st.subheader("Voice Recognition")
+        
+        # Get transcript from the speech component
+        current_transcript = integrate_speech_component()
+        
+        # Display the transcript in a Streamlit text area
+        st.markdown("### Processed Voice Input:")
+        st.text_area("Voice Transcript", current_transcript, height=100)
+        
+        # Process buttons
+        col1, col2, col3 = st.columns(3)
+        with col1:
+            if st.button("Process with GPT"):
+                st.subheader("GPT-4o Response")
+                gpt_response = process_with_gpt(current_transcript)
+                st.markdown(gpt_response)
+        
+        with col2:
+            if st.button("Process with Claude"):
+                st.subheader("Claude Response")
+                claude_response = process_with_claude(current_transcript)
+                st.markdown(claude_response)
+        
+        with col3:
+            if st.button("Search ArXiv"):
+                st.subheader("ArXiv Search Results")
+                arxiv_results = perform_ai_lookup(current_transcript)
+                st.markdown(arxiv_results)
+                
             
+            # Display last voice input
+            if st.session_state.last_voice_input:
+                st.text_area("Last Voice Input:", st.session_state.last_voice_input, height=100)
         
-        # Display last voice input
-        if st.session_state.last_voice_input:
-            st.text_area("Last Voice Input:", st.session_state.last_voice_input, height=100)
-    
-    
-    if tab_main == "💬 Chat":
-        # Model Selection
-        model_choice = st.sidebar.radio(
-            "Choose AI Model:",
-            ["GPT-4o", "Claude-3", "GPT+Claude+Arxiv"]
-        )
         
-        # Chat Interface
-        user_input = st.text_area("Message:", height=100)
+        if tab_main == "💬 Chat":
+            # Model Selection
+            model_choice = st.sidebar.radio(
+                "Choose AI Model:",
+                ["GPT-4o", "Claude-3", "GPT+Claude+Arxiv"]
+            )
+            
+            # Chat Interface
+            user_input = st.text_area("Message:", 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()
         
-        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"])
+        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)
         
-        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"])
+        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!")
         
-        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()