IAMTFRMZA commited on
Commit
442d49c
·
verified ·
1 Parent(s): 004ead9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import os
3
  import uuid
 
4
  from openai import OpenAI
5
  from realtime_transcriber import WebSocketClient, connections, WEBSOCKET_URI, WEBSOCKET_HEADERS
6
 
@@ -14,7 +15,7 @@ client = OpenAI(api_key=OPENAI_API_KEY)
14
  session_id = str(uuid.uuid4())
15
  if session_id not in connections:
16
  connections[session_id] = WebSocketClient(WEBSOCKET_URI, WEBSOCKET_HEADERS, session_id)
17
- connections[session_id].start()
18
 
19
  # Functions for Document Assistant
20
  def process_user_input(message, history):
@@ -98,4 +99,5 @@ with gr.Blocks(theme=gr.themes.Base(), css="""
98
  audio_in.stream(fn=send_audio_chunk_realtime, inputs=audio_in, outputs=live_transcript)
99
  clear_btn.click(fn=clear_transcript, outputs=live_transcript)
100
 
101
- demo.launch()
 
 
1
  import gradio as gr
2
  import os
3
  import uuid
4
+ import threading
5
  from openai import OpenAI
6
  from realtime_transcriber import WebSocketClient, connections, WEBSOCKET_URI, WEBSOCKET_HEADERS
7
 
 
15
  session_id = str(uuid.uuid4())
16
  if session_id not in connections:
17
  connections[session_id] = WebSocketClient(WEBSOCKET_URI, WEBSOCKET_HEADERS, session_id)
18
+ threading.Thread(target=connections[session_id].run, daemon=True).start()
19
 
20
  # Functions for Document Assistant
21
  def process_user_input(message, history):
 
99
  audio_in.stream(fn=send_audio_chunk_realtime, inputs=audio_in, outputs=live_transcript)
100
  clear_btn.click(fn=clear_transcript, outputs=live_transcript)
101
 
102
+ if __name__ == "__main__":
103
+ demo.launch()