Update app.py

app.py

@@ -88,24 +88,17 @@ HTML_CONTENT = """
                     body: JSON.stringify({ prompt: text })
                 });
 
-                if (response.ok) {
-                    const blob = await response.blob();
-                    const url = URL.createObjectURL(blob);
+                const data = await response.json();
+                if (data.audio) {
+                    const audioBlob = new Blob([new Uint8Array(atob(data.audio).split('').map(c => c.charCodeAt(0)))], { type: 'audio/wav' });
+                    const url = URL.createObjectURL(audioBlob);
                     const audio = document.getElementById('audioPlayer');
                     audio.src = url;
                     audio.style.display = 'block';
                     audio.play();
-
-                    const textResponse = await fetch('/llama', {
-                        method: 'POST',
-                        headers: { 'Content-Type': 'application/json' },
-                        body: JSON.stringify({ prompt: text })
-                    });
-                    const textData = await textResponse.json();
-                    addMessage('Bot: ' + (textData.text || 'Tôi không hiểu bạn nói gì.'));
+                    addMessage('Bot: ' + data.text);
                 } else {
-                    const errorData = await response.json();
-                    addMessage('Bot: Lỗi - ' + (errorData.error || 'Không có phản hồi'));
+                    addMessage('Bot: Lỗi - ' + (data.error || 'Không có phản hồi'));
                 }
             } catch (e) {
                 addMessage('Bot: Lỗi kết nối - ' + e.message);
@@ -141,32 +134,18 @@ HTML_CONTENT = """
                     body: formData
                 });
 
-                if (response.ok) {
-                    const blob = await response.blob();
-                    const url = URL.createObjectURL(blob);
+                const data = await response.json();
+                if (data.audio) {
+                    const audioBlob = new Blob([new Uint8Array(atob(data.audio).split('').map(c => c.charCodeAt(0)))], { type: 'audio/wav' });
+                    const url = URL.createObjectURL(audioBlob);
                     const audio = document.getElementById('audioPlayer');
                     audio.src = url;
                     audio.style.display = 'block';
                     audio.play();
-
-                    const sttResponse = await fetch('/stt', {
-                        method: 'POST',
-                        body: formData
-                    });
-                    const sttData = await sttResponse.json();
-                    if (sttData.text) {
-                        addMessage('Bạn: ' + sttData.text);
-                        const llamaResponse = await fetch('/llama', {
-                            method: 'POST',
-                            headers: { 'Content-Type': 'application/json' },
-                            body: JSON.stringify({ prompt: sttData.text })
-                        });
-                        const llamaData = await llamaResponse.json();
-                        addMessage('Bot: ' + (llamaData.text || 'Tôi không hiểu bạn nói gì.'));
-                    }
+                    addMessage('Bạn: ' + data.input);
+                    addMessage('Bot: ' + data.text);
                 } else {
-                    const errorData = await response.json();
-                    addMessage('Bot: Lỗi - ' + (errorData.error || 'Không có phản hồi'));
+                    addMessage('Bot: Lỗi - ' + (data.error || 'Không có phản hồi'));
                 }
             } catch (e) {
                 addMessage('Bot: Lỗi kết nối - ' + e.message);
@@ -198,7 +177,10 @@ async def text_to_speech(text: str):
         audio = client.text_to_speech(model=TTS_MODEL, text=text)
         with open(output_path, "wb") as f:
             f.write(audio)
-        return FileResponse(output_path, media_type="audio/wav", filename="output.wav")
+        with open(output_path, "rb") as f:
+            audio_data = f.read()
+        import base64
+        return base64.b64encode(audio_data).decode('utf-8')
     except Exception as e:
         print(f"❌ Lỗi TTS: {e}")
         return {"error": str(e)}
@@ -221,7 +203,6 @@ async def generate_text(prompt: str):
         print(f"🔄 Đang xử lý LLaMA với model {LLAMA_MODEL}...")
         output = llama_pipeline(prompt, max_new_tokens=100)[0]["generated_text"]
         print(f"Output từ LLaMA: {output}")
-        # Nếu output rỗng hoặc không có ý nghĩa, trả về mặc định
         if not output or output.strip() == prompt.strip():
             output = "Tôi không hiểu bạn nói gì."
         return {"text": output}
@@ -235,16 +216,12 @@ async def chat(prompt: str):
         # Gửi tới LLaMA
         llama_output = llama_pipeline(prompt, max_new_tokens=100)[0]["generated_text"]
         print(f"LLaMA output: {llama_output}")
-        # Xử lý nếu LLaMA không trả về kết quả hợp lệ
         if not llama_output or llama_output.strip() == prompt.strip():
             llama_output = "Tôi không hiểu bạn nói gì."
-        
+
         # Tạo TTS từ output của LLaMA
-        output_path = os.path.join(TEMP_DIR, "output.wav")
-        audio = client.text_to_speech(model=TTS_MODEL, text=llama_output)
-        with open(output_path, "wb") as f:
-            f.write(audio)
-        return FileResponse(output_path, media_type="audio/wav", filename="output.wav")
+        audio_base64 = await text_to_speech(llama_output)
+        return {"text": llama_output, "audio": audio_base64}
     except Exception as e:
         print(f"❌ Lỗi chat: {e}")
         return {"error": str(e)}
@@ -256,6 +233,8 @@ async def audio_chat(file: UploadFile = File(...)):
         audio_data = await file.read()
         stt_output = client.automatic_speech_recognition(model=STT_MODEL, data=audio_data).get("text", "")
         print(f"STT output: {stt_output}")
+        if not stt_output:
+            stt_output = "Không nghe được gì."
 
         # LLaMA: Sinh câu trả lời
         llama_output = llama_pipeline(stt_output, max_new_tokens=100)[0]["generated_text"]
@@ -264,11 +243,8 @@ async def audio_chat(file: UploadFile = File(...)):
             llama_output = "Tôi không hiểu bạn nói gì."
 
         # TTS: Chuyển câu trả lời thành âm thanh
-        output_path = os.path.join(TEMP_DIR, "output.wav")
-        audio = client.text_to_speech(model=TTS_MODEL, text=llama_output)
-        with open(output_path, "wb") as f:
-            f.write(audio)
-        return FileResponse(output_path, media_type="audio/wav", filename="output.wav")
+        audio_base64 = await text_to_speech(llama_output)
+        return {"input": stt_output, "text": llama_output, "audio": audio_base64}
     except Exception as e:
         print(f"❌ Lỗi audio_chat: {e}")
         return {"error": str(e)}