Update app.py

app.py

@@ -75,15 +75,13 @@ HTML_CONTENT = """
         let mediaRecorder;
         let audioChunks = [];
 
-        // Hàm gửi chat (văn bản -> LLaMA -> TTS)
         async function sendChat() {
             const text = document.getElementById('textInput').value;
             if (!text) return;
             addMessage('Bạn: ' + text);
-            document.getElementById('textInput').value = ''; // Xóa input sau khi gửi
+            document.getElementById('textInput').value = '';
 
             try {
-                // Gửi tới endpoint /chat (tích hợp LLaMA và TTS)
                 const response = await fetch('/chat', {
                     method: 'POST',
                     headers: { 'Content-Type': 'application/json' },
@@ -98,16 +96,13 @@ HTML_CONTENT = """
                     audio.style.display = 'block';
                     audio.play();
 
-                    // Lấy văn bản từ LLaMA để hiển thị
                     const textResponse = await fetch('/llama', {
                         method: 'POST',
                         headers: { 'Content-Type': 'application/json' },
                         body: JSON.stringify({ prompt: text })
                     });
                     const textData = await textResponse.json();
-                    if (textData.text) {
-                        addMessage('Bot: ' + textData.text);
-                    }
+                    addMessage('Bot: ' + (textData.text || 'Tôi không hiểu bạn nói gì.'));
                 } else {
                     const errorData = await response.json();
                     addMessage('Bot: Lỗi - ' + (errorData.error || 'Không có phản hồi'));
@@ -117,7 +112,6 @@ HTML_CONTENT = """
             }
         }
 
-        // Ghi âm (STT -> LLaMA -> TTS)
         async function startRecording() {
             const stream = await navigator.mediaDevices.getUserMedia({ audio: true });
             mediaRecorder = new MediaRecorder(stream);
@@ -142,7 +136,6 @@ HTML_CONTENT = """
             formData.append('file', audioBlob, 'recording.wav');
 
             try {
-                // Gửi tới endpoint /audio_chat (STT -> LLaMA -> TTS)
                 const response = await fetch('/audio_chat', {
                     method: 'POST',
                     body: formData
@@ -156,7 +149,6 @@ HTML_CONTENT = """
                     audio.style.display = 'block';
                     audio.play();
 
-                    // Lấy văn bản STT và LLaMA để hiển thị
                     const sttResponse = await fetch('/stt', {
                         method: 'POST',
                         body: formData
@@ -170,9 +162,7 @@ HTML_CONTENT = """
                             body: JSON.stringify({ prompt: sttData.text })
                         });
                         const llamaData = await llamaResponse.json();
-                        if (llamaData.text) {
-                            addMessage('Bot: ' + llamaData.text);
-                        }
+                        addMessage('Bot: ' + (llamaData.text || 'Tôi không hiểu bạn nói gì.'));
                     }
                 } else {
                     const errorData = await response.json();
@@ -231,18 +221,23 @@ 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}
     except Exception as e:
         print(f"❌ Lỗi LLaMA: {e}")
         return {"error": str(e)}
 
-# Endpoint tích hợp văn bản -> LLaMA -> TTS
 @app.post("/chat")
 async def chat(prompt: str):
     try:
         # 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")
@@ -254,7 +249,6 @@ async def chat(prompt: str):
         print(f"❌ Lỗi chat: {e}")
         return {"error": str(e)}
 
-# Endpoint tích hợp STT -> LLaMA -> TTS
 @app.post("/audio_chat")
 async def audio_chat(file: UploadFile = File(...)):
     try:
@@ -266,6 +260,8 @@ async def audio_chat(file: UploadFile = File(...)):
         # LLaMA: Sinh câu trả lời
         llama_output = llama_pipeline(stt_output, max_new_tokens=100)[0]["generated_text"]
         print(f"LLaMA output: {llama_output}")
+        if not llama_output or llama_output.strip() == stt_output.strip():
+            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")