Add code

app.py

@@ -3,18 +3,18 @@ import logging
 # Configure the root logger to WARNING to suppress debug messages from other libraries
 logging.basicConfig(level=logging.WARNING)
 
-# Create a console handler
-console_handler = logging.StreamHandler()
-console_handler.setLevel(logging.DEBUG)
+# Create a file handler instead of console handler
+file_handler = logging.FileHandler("gradio_webrtc.log")
+file_handler.setLevel(logging.DEBUG)
 
-# Create a formatter
-formatter = logging.Formatter("%(name)s - %(levelname)s - %(message)s")
-console_handler.setFormatter(formatter)
+# Create a formatter (you might want to add timestamp to file logs)
+formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+file_handler.setFormatter(formatter)
 
 # Configure the logger for your specific library
 logger = logging.getLogger("gradio_webrtc")
 logger.setLevel(logging.DEBUG)
-logger.addHandler(console_handler)
+logger.addHandler(file_handler)
 
 
 import base64
@@ -24,8 +24,7 @@ import tempfile
 import time
 import traceback
 from dataclasses import dataclass
-from queue import Queue
-from threading import Thread
+from threading import Event, Thread
 
 import gradio as gr
 import librosa
@@ -47,11 +46,11 @@ snapshot_download(repo_id, local_dir="./checkpoint", revision="main")
 IP = "0.0.0.0"
 PORT = 60808
 
-# thread = Thread(target=serve, daemon=True)
-# thread.start()
+thread = Thread(target=serve, daemon=True)
+thread.start()
 
 
-API_URL = "https://freddyaboulton-omni-mini-webrtc-backend.hf.space/chat"
+API_URL = "http://0.0.0.0:60808/chat"
 
 account_sid = os.environ.get("TWILIO_ACCOUNT_SID")
 auth_token = os.environ.get("TWILIO_AUTH_TOKEN")
@@ -231,7 +230,7 @@ class OmniHandler(StreamHandler):
         super().__init__(
             expected_layout="mono", output_sample_rate=OUT_RATE, output_frame_size=480
         )
-        self.chunk_queue = Queue()
+        self.event = Event()
         self.state = AppState()
         self.generator = None
         self.duration = 0
@@ -241,22 +240,25 @@ class OmniHandler(StreamHandler):
             return
         process_audio(frame, self.state)
         if self.state.pause_detected:
-            self.chunk_queue.put(True)
+            self.event.set()
 
     def reset(self):
         self.generator = None
+        self.event.clear()
         self.state = AppState()
         self.duration = 0
 
     def emit(self):
-        if not self.generator:
-            self.chunk_queue.get()
+        if not self.event.is_set():
+            return None
+        else:
+            if not self.generator:
+                self.generator = response(self.state)
             self.state.responding = True
-            self.generator = response(self.state)
-        try:
-            return next(self.generator)
-        except StopIteration:
-            self.reset()
+            try:
+                return next(self.generator)
+            except StopIteration:
+                self.reset()
 
 
 with gr.Blocks() as demo:

requirements.txt

@@ -12,5 +12,5 @@ fastapi==0.112.4
 librosa==0.10.2.post1
 flask==3.0.3
 fire
-gradio_webrtc==0.0.5a1
+gradio_webrtc==0.0.6a3
 twilio