Update app.py

app.py

@@ -78,42 +78,54 @@ async def chat(request: ChatRequest):
 
 # Endpoint voice chat + TTS
 @app.post("/voice_chat")
-async def voice_chat(file: UploadFile = File(...)):
-    file_location = f"temp_{file.filename}"
-    with open(file_location, "wb") as f:
-        f.write(await file.read())
-
-    result = whisper_model.transcribe(file_location, language="vi")
-    user_text = result["text"]
-    os.remove(file_location)
-
-# Kiểm tra yêu cầu mở nhạc
-    if any(kw in user_text.lower() for kw in ["nghe nhạc", "mở bài hát", "bài hát", "bài"]):
-        song_name = extract_song_name(user_text)
-        if song_name:
-            wav_path = download_youtube_as_wav(song_name)
-            if wav_path:
-                return FileResponse(wav_path, media_type="audio/wav", filename="song.wav")
-            else:
-                return {"error": "Không tìm thấy hoặc tải được bài hát."}
-
-    # Nếu không phải yêu cầu mở nhạc → xử lý như cũ
-    conversation.append({"role": "user", "content": user_text})
-    text = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
-    model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
-    response_text = generate_full_response(model_inputs)
-    conversation.append({"role": "assistant", "content": response_text})
-
-    tts = gTTS(response_text, lang="vi")
-    audio_file = "response.mp3"
-    tts.save(audio_file)
-
-    return {
-        "user_text": user_text,
-        "response": response_text,
-        "audio_url": "/get_audio"
-    }
-
+async def voice_chat(request: Request):
+    try:
+        raw_audio = await request.body()
+        sample_rate = 16000
+        audio_np = np.frombuffer(raw_audio, dtype=np.int16)
+
+        # Chuyển thành WAV
+        wav_io = io.BytesIO()
+        wav.write(wav_io, sample_rate, audio_np)
+        wav_io.seek(0)
+
+        with open("temp_audio.wav", "wb") as f:
+            f.write(wav_io.read())
+
+        # Whisper nhận dạng
+        result = whisper_model.transcribe("temp_audio.wav", language="vi")
+        user_text = result["text"]
+
+        # Kiểm tra yêu cầu mở nhạc
+        if any(kw in user_text.lower() for kw in ["nghe nhạc", "mở bài hát", "bài hát", "bài"]):
+            song_name = extract_song_name(user_text)
+            if song_name:
+                wav_path = download_youtube_as_wav(song_name)
+                if wav_path:
+                    return FileResponse(wav_path, media_type="audio/wav", filename="song.wav")
+                else:
+                    return JSONResponse({"error": "Không tìm thấy hoặc tải được bài hát."}, status_code=404)
+
+        # Xử lý hội thoại
+        conversation.append({"role": "user", "content": user_text})
+        text = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
+        model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
+        response_text = generate_full_response(model_inputs)
+        conversation.append({"role": "assistant", "content": response_text})
+
+        # TTS
+        tts = gTTS(response_text, lang="vi")
+        audio_file = "response.mp3"
+        tts.save(audio_file)
+
+        return {
+            "user_text": user_text,
+            "response": response_text,
+            "audio_url": "/get_audio"
+        }
+
+    except Exception as e:
+        return JSONResponse({"error": str(e)}, status_code=500)
 # Endpoint trả về file âm thanh
 @app.get("/get_audio")
 async def get_audio():