Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
-import httpx  # This was missing
-from fastrtc import ReplyOnPause, Stream, get_stt_model, get_tts_model
+import httpx
+from fastrtc import ReplyOnPause, Stream, get_stt_model, get_tts_model, StreamHandlerBase
 from openai import OpenAI
 
 # Initialize Sambanova Client
@@ -12,20 +12,23 @@ sambanova_client = OpenAI(
 stt_model = get_stt_model()
 tts_model = get_tts_model()
 
-# Echo function to convert speech to text, then back to speech
-def echo(audio):
-    prompt = stt_model.stt(audio)
-    response = sambanova_client.chat.completions.create(
-        model="mistralai/Mistral-Small-24B-Instruct-2501",
-        messages=[{"role": "user", "content": prompt}],
-        max_tokens=200,
-    )
-    reply = response.choices[0].message.content
-    for audio_chunk in tts_model.stream_tts_sync(reply):
-        yield audio_chunk
+# Create a proper handler subclass
+class EchoHandler(StreamHandlerBase):
+    def __init__(self):
+        super().__init__()
 
-# Dummy implementation of get_cloudflare_turn_credentials
-# You’ll need to replace this with real logic if it's not a stub
+    def on_audio(self, audio):
+        prompt = stt_model.stt(audio)
+        response = sambanova_client.chat.completions.create(
+            model="mistralai/Mistral-Small-24B-Instruct-2501",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=200,
+        )
+        reply = response.choices[0].message.content
+        for audio_chunk in tts_model.stream_tts_sync(reply):
+            yield audio_chunk
+
+# Dummy TURN config
 def get_cloudflare_turn_credentials(
     turn_key_id=None,
     turn_key_api_token=None,
@@ -41,9 +44,9 @@ def get_cloudflare_turn_credentials(
         ]
     }
 
-# Create and start the stream
+# Launch stream with correct handler
 stream = Stream(
-    handler=echo,  # You had handler=... which is invalid
+    handler=EchoHandler(),
     rtc_configuration=get_cloudflare_turn_credentials,
     modality="audio",
     mode="send-receive"