Spaces:

awacke1
/

RealTimeAsyncASR

Sleeping

App Files Files Community

awacke1 commited on Nov 26, 2024

Commit

8e7963e

verified ·

1 Parent(s): 172a0f1

Create app.py

Browse files

Files changed (1) hide show

app.py +209 -0

app.py ADDED Viewed

	@@ -0,0 +1,209 @@

+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>
+    <head>
+        <title>Continuous Speech Demo</title>
+        <style>
+            body {
+                font-family: sans-serif;
+                padding: 20px;
+                max-width: 800px;
+                margin: 0 auto;
+            }
+            button {
+                padding: 10px 20px;
+                margin: 10px 5px;
+                font-size: 16px;
+            }
+            #status {
+                margin: 10px 0;
+                padding: 10px;
+                background: #e8f5e9;
+                border-radius: 4px;
+            }
+            #output {
+                white-space: pre-wrap;
+                padding: 15px;
+                background: #f5f5f5;
+                border-radius: 4px;
+                margin: 10px 0;
+                min-height: 100px;
+                max-height: 400px;
+                overflow-y: auto;
+            }
+            .controls {
+                margin: 10px 0;
+            }
+        </style>
+    </head>
+    <body>
+        <div class="controls">
+            <button id="start">Start Listening</button>
+            <button id="stop" disabled>Stop Listening</button>
+            <button id="clear">Clear Text</button>
+        </div>
+        <div id="status">Ready</div>
+        <div id="output"></div>
+        <script>
+            if (!('webkitSpeechRecognition' in window)) {
+                alert('Speech recognition not supported');
+            } else {
+                const recognition = new webkitSpeechRecognition();
+                const startButton = document.getElementById('start');
+                const stopButton = document.getElementById('stop');
+                const clearButton = document.getElementById('clear');
+                const status = document.getElementById('status');
+                const output = document.getElementById('output');
+                let fullTranscript = '';
+                let lastUpdateTime = Date.now();
+                // Configure recognition
+                recognition.continuous = true;
+                recognition.interimResults = true;
+                startButton.onclick = () => {
+                    try {
+                        recognition.start();
+                        status.textContent = 'Listening...';
+                        startButton.disabled = true;
+                        stopButton.disabled = false;
+                    } catch (e) {
+                        console.error(e);
+                        status.textContent = 'Error: ' + e.message;
+                    }
+                };
+                stopButton.onclick = () => {
+                    recognition.stop();
+                    status.textContent = 'Stopped';
+                    startButton.disabled = false;
+                    stopButton.disabled = true;
+                };
+                clearButton.onclick = () => {
+                    fullTranscript = '';
+                    output.textContent = '';
+                };
+                recognition.onresult = (event) => {
+                    let interimTranscript = '';
+                    let finalTranscript = '';
+                    for (let i = event.resultIndex; i < event.results.length; i++) {
+                        const transcript = event.results[i][0].transcript;
+                        if (event.results[i].isFinal) {
+                            finalTranscript += transcript + '\\n';
+                        } else {
+                            interimTranscript += transcript;
+                        }
+                    }
+                    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();
+                    }
+                    output.textContent = fullTranscript + (interimTranscript ? '... ' + interimTranscript : '');
+                    output.scrollTop = output.scrollHeight;
+                };
+                recognition.onend = () => {
+                    if (!stopButton.disabled) {
+                        try {
+                            recognition.start();
+                            console.log('Restarted recognition');
+                        } catch (e) {
+                            console.error('Failed to restart recognition:', e);
+                            status.textContent = 'Error restarting: ' + e.message;
+                            startButton.disabled = false;
+                            stopButton.disabled = true;
+                        }
+                    }
+                };
+                recognition.onerror = (event) => {
+                    console.error('Recognition error:', event.error);
+                    status.textContent = 'Error: ' + event.error;
+                    if (event.error === 'not-allowed' || event.error === 'service-not-allowed') {
+                        startButton.disabled = false;
+                        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.experimental_rerun()