Spaces:

awacke1
/

RescuerOfStolenBikes

Running

App Files Files Community

awacke1 commited on 27 days ago

Commit

47d3d04

•

1 Parent(s): 630aa9e

Update app.py

Browse files

Files changed (1) hide show

app.py +86 -117

app.py CHANGED Viewed

@@ -58,9 +58,8 @@ st.set_page_config(
 def get_cookie_manager():
     """Create and return a cookie manager."""
     return stx.CookieManager()
 def create_speech_component():
-    """Create speech recognition component with cookie-based storage."""
     speech_recognition_html = """
     <!DOCTYPE html>
@@ -95,6 +94,10 @@ def create_speech_component():
                 max-height: 400px;
                 overflow-y: auto;
             }
         </style>
     </head>
     <body>
@@ -105,16 +108,25 @@ def create_speech_component():
         </div>
         <div id="status">Ready</div>
         <div id="output"></div>
         <script>
-            function setCookie(name, value, days) {
-                let expires = "";
-                if (days) {
-                    const date = new Date();
-                    date.setTime(date.getTime() + (days * 24 * 60 * 60 * 1000));
-                    expires = "; expires=" + date.toUTCString();
                 }
-                document.cookie = name + "=" + encodeURIComponent(value) + expires + "; path=/";
             }
             if (!('webkitSpeechRecognition' in window)) {
@@ -133,24 +145,33 @@ def create_speech_component():
                 startButton.onclick = () => {
                     recognition.start();
-                    status.textContent = 'Listening...';
                     startButton.disabled = true;
                     stopButton.disabled = false;
                 };
                 stopButton.onclick = () => {
                     recognition.stop();
                     status.textContent = 'Stopped';
                     startButton.disabled = false;
                     stopButton.disabled = true;
-                    // Save transcript to cookie when stopping
-                    setCookie('voice_transcript', fullTranscript, 1);
                 };
                 clearButton.onclick = () => {
                     fullTranscript = '';
                     output.textContent = '';
-                    setCookie('voice_transcript', '', 1);
                 };
                 recognition.onresult = (event) => {
@@ -162,8 +183,6 @@ def create_speech_component():
                         if (event.results[i].isFinal) {
                             finalTranscript += transcript + '\\n';
                             fullTranscript += transcript + '\\n';
-                            // Update cookie with latest transcript
-                            setCookie('voice_transcript', fullTranscript, 1);
                         } else {
                             interimTranscript += transcript;
                         }
@@ -191,24 +210,29 @@ def create_speech_component():
     </html>
     """
-    return components.html(speech_recognition_html, height=400)
 def integrate_speech_component():
-    """Integrate speech component with cookie-based storage."""
-    # Initialize cookie manager
-    cookie_manager = get_cookie_manager()
-    # Create speech component
-    create_speech_component()
-    # Get transcript from cookie
-    transcript = cookie_manager.get('voice_transcript') or ""
-    # Update session state
-    if transcript:
-        st.session_state.voice_transcript = transcript
-    return transcript
 # 2. 🚲BikeAI🏆 Load environment variables and initialize clients
@@ -1037,108 +1061,53 @@ def main():
         # 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)
-        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()

 def get_cookie_manager():
     """Create and return a cookie manager."""
     return stx.CookieManager()
 def create_speech_component():
+    """Create speech recognition component with timer-based updates."""
     speech_recognition_html = """
     <!DOCTYPE html>
                 max-height: 400px;
                 overflow-y: auto;
             }
+            .listening {
+                color: green;
+                font-weight: bold;
+            }
         </style>
     </head>
     <body>
         </div>
         <div id="status">Ready</div>
         <div id="output"></div>
+        <div id="debug"></div>
         <script>
+            let lastTranscript = '';
+            let updateInterval;
+            function checkAndUpdateTranscript() {
+                const currentTranscript = document.getElementById('output').textContent;
+                if (currentTranscript !== lastTranscript) {
+                    lastTranscript = currentTranscript;
+                    window.parent.postMessage({
+                        type: 'streamlit:setComponentValue',
+                        data: currentTranscript
+                    }, '*');
+                    // Debug output
+                    document.getElementById('debug').textContent =
+                        'Last update: ' + new Date().toLocaleTimeString();
                 }
             }
             if (!('webkitSpeechRecognition' in window)) {
                 startButton.onclick = () => {
                     recognition.start();
+                    status.textContent = '🎤 Listening...';
+                    status.className = 'listening';
                     startButton.disabled = true;
                     stopButton.disabled = false;
+                    // Start periodic updates
+                    updateInterval = setInterval(checkAndUpdateTranscript, 500);
                 };
                 stopButton.onclick = () => {
                     recognition.stop();
                     status.textContent = 'Stopped';
+                    status.className = '';
                     startButton.disabled = false;
                     stopButton.disabled = true;
+                    // Stop periodic updates
+                    clearInterval(updateInterval);
                 };
                 clearButton.onclick = () => {
                     fullTranscript = '';
                     output.textContent = '';
+                    window.parent.postMessage({
+                        type: 'streamlit:setComponentValue',
+                        data: ''
+                    }, '*');
                 };
                 recognition.onresult = (event) => {
                         if (event.results[i].isFinal) {
                             finalTranscript += transcript + '\\n';
                             fullTranscript += transcript + '\\n';
                         } else {
                             interimTranscript += transcript;
                         }
     </html>
     """
+    # Return both the component value and height
+    return components.html(
+        speech_recognition_html,
+        height=400,
+    )
 def integrate_speech_component():
+    """Integrate speech component with timer-based updates."""
+    if "voice_transcript" not in st.session_state:
+        st.session_state.voice_transcript = ""
+    # Create the component and get its value
+    value = create_speech_component()
+    # If we got a value from the component, update session state
+    if value is not None and value != "":
+        st.session_state.voice_transcript = value
+        # Debug output
+        st.write("Updated transcript:", value)
+    return st.session_state.voice_transcript
 # 2. 🚲BikeAI🏆 Load environment variables and initialize clients
         # Get transcript from the speech component
         current_transcript = integrate_speech_component()
+        # Display the transcript with live updates
+        transcript_placeholder = st.empty()
+        transcript_placeholder.text_area(
+            "Voice Transcript (Live)",
+            value=current_transcript,
+            height=100
+        )
+        # Add a status indicator
+        status_placeholder = st.empty()
+        if current_transcript != st.session_state.get('last_transcript', ''):
+            status_placeholder.success("🔴 Recording in progress...")
+            st.session_state.last_transcript = current_transcript
         # Process buttons
         col1, col2, col3 = st.columns(3)
         with col1:
             if st.button("Process with GPT"):
+                if current_transcript:
+                    with st.spinner("Processing with GPT..."):
+                        gpt_response = process_with_gpt(current_transcript)
+                        st.markdown(gpt_response)
         with col2:
             if st.button("Process with Claude"):
+                if current_transcript:
+                    with st.spinner("Processing with Claude..."):
+                        claude_response = process_with_claude(current_transcript)
+                        st.markdown(claude_response)
         with col3:
             if st.button("Search ArXiv"):
+                if current_transcript:
+                    with st.spinner("Searching ArXiv..."):
+                        arxiv_results = perform_ai_lookup(current_transcript)
+                        st.markdown(arxiv_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()