Spaces:

awacke1
/

RescuerOfStolenBikes

Running

App Files Files Community

awacke1 commited on Dec 1, 2024

Commit

b75652c

verified ·

1 Parent(s): cfa51fe

Update app.py

Browse files

Files changed (1) hide show

app.py +102 -200

app.py CHANGED Viewed

@@ -660,12 +660,100 @@ def display_file_manager():
             if st.button("🗑", key=f"delete_{unique_id}"):
                 os.remove(file)
                 st.rerun()
-# First, define the speech recognition HTML template at the global scope, before the main() function
 speech_recognition_html = """
 <!DOCTYPE html>
 <html>
 <head>
-    <title>Voice Recognition</title>
     <style>
         body {
             font-family: sans-serif;
@@ -709,8 +797,6 @@ speech_recognition_html = """
     <div id="output"></div>
     <script>
-        let messageTarget = 'textarea[aria-label="Message:"]';
         if (!('webkitSpeechRecognition' in window)) {
             alert('Speech recognition not supported');
         } else {
@@ -727,15 +813,6 @@ 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 {
@@ -766,7 +843,6 @@ speech_recognition_html = """
             clearButton.onclick = () => {
                 fullTranscript = '';
                 output.textContent = '';
-                updateTextArea('');
                 window.parent.postMessage({
                     type: 'clear_transcript',
                 }, '*');
@@ -788,11 +864,10 @@ speech_recognition_html = """
                 if (finalTranscript || (Date.now() - lastUpdateTime > 5000)) {
                     if (finalTranscript) {
                         fullTranscript += finalTranscript;
-                        updateTextArea(fullTranscript);
                         window.parent.postMessage({
                             type: 'final_transcript',
-                            text: finalTranscript,
-                            updateInput: true
                         }, '*');
                     }
                     lastUpdateTime = Date.now();
@@ -831,137 +906,6 @@ 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')
@@ -1107,11 +1051,7 @@ def get_media_html(media_path, media_type="video", width="100%"):
 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"],
@@ -1120,39 +1060,8 @@ def main():
     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:
@@ -1163,8 +1072,6 @@ def main():
                         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:")
@@ -1190,24 +1097,19 @@ def main():
                     elif data.get('type') == 'clear_transcript':
                         st.session_state.last_voice_input = ""
-                        st.session_state.current_text_input = ""
-                        st.rerun()
             except Exception as e:
                 st.error(f"Error processing voice input: {e}")
     # [Rest of the main function remains the same]
     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:
-                # Process the input as before
-                pass
     elif tab_main == "📸 Media Gallery":
         create_media_gallery()

             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")
+    # 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 = """
 <!DOCTYPE html>
 <html>
 <head>
+    <title>Continuous Speech Demo</title>
     <style>
         body {
             font-family: sans-serif;
     <div id="output"></div>
     <script>
         if (!('webkitSpeechRecognition' in window)) {
             alert('Speech recognition not supported');
         } else {
             recognition.continuous = true;
             recognition.interimResults = true;
             // Function to start recognition
             const startRecognition = () => {
                 try {
             clearButton.onclick = () => {
                 fullTranscript = '';
                 output.textContent = '';
                 window.parent.postMessage({
                     type: 'clear_transcript',
                 }, '*');
                 if (finalTranscript || (Date.now() - lastUpdateTime > 5000)) {
                     if (finalTranscript) {
                         fullTranscript += finalTranscript;
+                        // Send to Streamlit
                         window.parent.postMessage({
                             type: 'final_transcript',
+                            text: finalTranscript
                         }, '*');
                     }
                     lastUpdateTime = Date.now();
 </html>
 """
 # Helper Functions
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
 def main():
     st.sidebar.markdown("### 🚲BikeAI🏆 Claude and GPT Multi-Agent Research AI")
     # Main navigation
     tab_main = st.radio("Choose Action:",
                         ["🎤 Voice Input", "💬 Chat", "📸 Media Gallery", "🔍 Search ArXiv", "📝 File Editor"],
     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:
                         text = data.get('text', '').strip()
                         if text:
                             st.session_state.last_voice_input = text
                             # Process voice input with AI
                             st.subheader("AI Response to Voice Input:")
                     elif data.get('type') == 'clear_transcript':
                         st.session_state.last_voice_input = ""
+                        st.experimental_rerun()
             except Exception as e:
                 st.error(f"Error processing voice input: {e}")
+        # 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)
     # [Rest of the main function remains the same]
     elif tab_main == "💬 Chat":
+        # [Previous chat interface code]
+        pass
     elif tab_main == "📸 Media Gallery":
         create_media_gallery()