Spaces:

awacke1
/

RealTimeAsyncASR

Running

App Files Files Community

awacke1 commited on Nov 26, 2024

Commit

07dd574

verified ·

1 Parent(s): 07a284c

Update app.py

Browse files

Files changed (1) hide show

app.py +61 -40

app.py CHANGED Viewed

@@ -1,19 +1,23 @@
 import streamlit as st
 import datetime
 import os
 # Initialize session state variables
 if 'transcript_history' not in st.session_state:
     st.session_state.transcript_history = []
-# Create containers at the top level
-st.title("Speech Recognition with Transcript History")
 # Create the main layout
 col1, col2 = st.columns([2, 1])
 with col1:
-    # Speech recognition component
     html = """
     <!DOCTYPE html>
     <html>
@@ -78,7 +82,8 @@ with col1:
                 recognition.continuous = true;
                 recognition.interimResults = true;
-                startButton.onclick = () => {
                     try {
                         recognition.start();
                         status.textContent = 'Listening...';
@@ -90,6 +95,13 @@ with col1:
                     }
                 };
                 stopButton.onclick = () => {
                     recognition.stop();
                     status.textContent = 'Stopped';
@@ -100,6 +112,8 @@ with col1:
                 clearButton.onclick = () => {
                     fullTranscript = '';
                     output.textContent = '';
                 };
                 recognition.onresult = (event) => {
@@ -118,15 +132,11 @@ with col1:
                     if (finalTranscript || (Date.now() - lastUpdateTime > 5000)) {
                         if (finalTranscript) {
                             fullTranscript += finalTranscript;
-                            // Try to communicate with Streamlit
-                            try {
-                                parent.postMessage({
-                                    type: 'transcriptUpdate',
-                                    text: finalTranscript
-                                }, '*');
-                            } catch (e) {
-                                console.error('Failed to send transcript:', e);
-                            }
                         }
                         lastUpdateTime = Date.now();
                     }
@@ -158,52 +168,63 @@ with col1:
                         stopButton.disabled = true;
                     }
                 };
             }
         </script>
     </body>
     </html>
     """
-    st.components.v1.html(html, height=400)
 with col2:
     # Display transcript history
     st.subheader("Transcript History")
-    transcript_text = st.empty()
-    # Save transcript function
     def save_transcript(text):
         if not os.path.exists('transcripts'):
             os.makedirs('transcripts')
         timestamp = datetime.datetime.now().strftime('%Y%m%d_%H%M%S')
         filename = f"transcripts/transcript_{timestamp}.md"
         with open(filename, 'w', encoding='utf-8') as f:
             f.write(text)
-    # Display full transcript
     if st.session_state.transcript_history:
         full_transcript = "\n".join(st.session_state.transcript_history)
-        transcript_text.text_area("Full Transcript", value=full_transcript, height=300)
-        # Save and download buttons
-        col1, col2 = st.columns(2)
-        with col1:
-            if st.button("Save Transcript"):
-                save_transcript(full_transcript)
-                st.success("Transcript saved!")
-        with col2:
-            st.download_button(
-                label="Download Transcript",
-                data=full_transcript,
-                file_name=f"transcript_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.md",
-                mime="text/markdown"
-            )
-# Listen for transcript updates
-if st.session_state.get('transcriptUpdate'):
-    new_text = st.session_state.transcriptUpdate
-    st.session_state.transcript_history.append(new_text)
-    st.session_state.transcriptUpdate = None  # Clear the update
-    st.rerun()

 import streamlit as st
 import datetime
 import os
+import base64
 # Initialize session state variables
 if 'transcript_history' not in st.session_state:
     st.session_state.transcript_history = []
+# Function to create a download link for a string
+def get_download_link(text, filename):
+    b64 = base64.b64encode(text.encode()).decode()
+    return f'<a href="data:text/plain;base64,{b64}" download="{filename}">Download Transcript</a>'
 # Create the main layout
+st.title("Speech Recognition with Transcript History")
 col1, col2 = st.columns([2, 1])
 with col1:
     html = """
     <!DOCTYPE html>
     <html>
                 recognition.continuous = true;
                 recognition.interimResults = true;
+                // Function to start recognition
+                const startRecognition = () => {
                     try {
                         recognition.start();
                         status.textContent = 'Listening...';
                     }
                 };
+                // Auto-start on load
+                window.addEventListener('load', () => {
+                    setTimeout(startRecognition, 1000); // Delay start by 1 second to ensure everything is loaded
+                });
+                startButton.onclick = startRecognition;
                 stopButton.onclick = () => {
                     recognition.stop();
                     status.textContent = 'Stopped';
                 clearButton.onclick = () => {
                     fullTranscript = '';
                     output.textContent = '';
+                    // Send clear signal to Streamlit
+                    window.parent.postMessage({type: 'clear'}, '*');
                 };
                 recognition.onresult = (event) => {
                     if (finalTranscript || (Date.now() - lastUpdateTime > 5000)) {
                         if (finalTranscript) {
                             fullTranscript += finalTranscript;
+                            // Send to Streamlit
+                            window.parent.postMessage({
+                                type: 'transcript',
+                                text: finalTranscript
+                            }, '*');
                         }
                         lastUpdateTime = Date.now();
                     }
                         stopButton.disabled = true;
                     }
                 };
+                // Listen for messages from Streamlit
+                window.addEventListener('message', (event) => {
+                    if (event.data.type === 'clear') {
+                        fullTranscript = '';
+                        output.textContent = '';
+                    }
+                });
             }
         </script>
     </body>
     </html>
     """
+    # Display the HTML component
+    component = st.components.v1.html(html, height=400)
 with col2:
     # Display transcript history
     st.subheader("Transcript History")
+    # Function to save transcript
     def save_transcript(text):
         if not os.path.exists('transcripts'):
             os.makedirs('transcripts')
         timestamp = datetime.datetime.now().strftime('%Y%m%d_%H%M%S')
         filename = f"transcripts/transcript_{timestamp}.md"
         with open(filename, 'w', encoding='utf-8') as f:
             f.write(text)
+        return filename
+    # Display transcript
     if st.session_state.transcript_history:
         full_transcript = "\n".join(st.session_state.transcript_history)
+        st.text_area("Full Transcript", value=full_transcript, height=300)
+        # Save transcript to file
+        timestamp = datetime.datetime.now().strftime('%Y%m%d_%H%M%S')
+        filename = f"transcript_{timestamp}.md"
+        # Create download link
+        st.markdown(get_download_link(full_transcript, filename), unsafe_allow_html=True)
+        # Save to file system
+        if st.button("Save to File"):
+            saved_file = save_transcript(full_transcript)
+            st.success(f"Saved to {saved_file}")
+# Handle transcript updates from JavaScript
+if component:
+    try:
+        data = component
+        if isinstance(data, dict) and data.get('type') == 'transcript':
+            st.session_state.transcript_history.append(data['text'])
+            st.experimental_rerun()
+        elif isinstance(data, dict) and data.get('type') == 'clear':
+            st.session_state.transcript_history = []
+            st.experimental_rerun()
+    except Exception as e:
+        st.error(f"Error processing transcript: {e}")