Update app.py

app.py

@@ -582,91 +582,7 @@ def display_file_manager():
                 st.rerun()
 
 
-def main():
-    st.sidebar.markdown("### 🚲BikeAI🏆 Claude and GPT Multi-Agent Research AI")
-    
-    # Main navigation
-    tab_main = st.radio("Choose Action:", 
-                        ["💬 Chat", "📸 Media Gallery", "🔍 Search ArXiv", "📝 File Editor"],
-                        horizontal=True)
-    
-    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()
-    
-    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
 speech_recognition_html = """
@@ -950,6 +866,22 @@ def get_media_html(media_path, media_type="video", width="100%"):
             Your browser does not support the audio element.
         </audio>
         '''
+
+
+def main_old():
+    
+    # Main navigation
+    tab_main = st.radio("Choose Action:", 
+                        ["💬 Chat", "📸 Media Gallery", "🔍 Search ArXiv", "📝 File Editor"],
+                        horizontal=True)
+
+    # Always show file manager in sidebar
+    # display_file_manager()
+
+#if __name__ == "__main__":
+#    main()
+
+
         
 def main():
     st.sidebar.markdown("### 🚲BikeAI🏆 Claude and GPT Multi-Agent Research AI")
@@ -1008,10 +940,59 @@ def main():
         if st.session_state.last_voice_input:
             st.text_area("Last Voice Input:", st.session_state.last_voice_input, height=100)
     
-    # [Rest of the main function remains the same]
-    elif tab_main == "💬 Chat":
-        # [Previous chat interface code]
-        pass
+    
+    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()
@@ -1031,6 +1012,8 @@ def main():
                 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()