Spaces:
Sleeping
Sleeping
Delete server.py
Browse files
server.py
DELETED
@@ -1,50 +0,0 @@
|
|
1 |
-
import asyncio
|
2 |
-
import websockets
|
3 |
-
import os
|
4 |
-
from io import BytesIO
|
5 |
-
from pydub import AudioSegment
|
6 |
-
from utils import transcribe_audio, get_response_llm, play_text_to_speech, load_whisper
|
7 |
-
|
8 |
-
# Load Whisper Model
|
9 |
-
model = load_whisper()
|
10 |
-
|
11 |
-
# Store connected clients
|
12 |
-
clients = set()
|
13 |
-
|
14 |
-
async def handle_audio_stream(websocket):
|
15 |
-
"""Handles incoming live audio stream from users."""
|
16 |
-
clients.add(websocket)
|
17 |
-
try:
|
18 |
-
while True:
|
19 |
-
audio_chunk = await websocket.recv()
|
20 |
-
if not audio_chunk:
|
21 |
-
break
|
22 |
-
|
23 |
-
# Convert received bytes to WAV file
|
24 |
-
audio = AudioSegment.from_file(BytesIO(audio_chunk))
|
25 |
-
audio.export("temp_audio_chunk.wav", format="wav")
|
26 |
-
|
27 |
-
# Transcribe
|
28 |
-
text = transcribe_audio(model, "temp_audio_chunk.wav")
|
29 |
-
if text:
|
30 |
-
print(f"User: {text}")
|
31 |
-
|
32 |
-
# Generate response
|
33 |
-
response_llm = get_response_llm(user_question=text)
|
34 |
-
|
35 |
-
# Play response
|
36 |
-
play_text_to_speech(text=response_llm)
|
37 |
-
|
38 |
-
# Send response to WebSocket client
|
39 |
-
await websocket.send(response_llm)
|
40 |
-
except:
|
41 |
-
pass
|
42 |
-
finally:
|
43 |
-
clients.remove(websocket)
|
44 |
-
|
45 |
-
async def main():
|
46 |
-
async with websockets.serve(handle_audio_stream, "0.0.0.0", 8765):
|
47 |
-
await asyncio.Future() # Keeps the server running
|
48 |
-
|
49 |
-
if __name__ == "__main__":
|
50 |
-
asyncio.run(main())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|