Update app.py

app.py

@@ -101,6 +101,8 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history,lan
         print("❌ Invalid key.")
         return "", chatbot_display, internal_history
     prompt = prompt.replace(expected_key, "")
+    if lang not None:
+        prompt = translate_text(prompt,lang+"-en")
     # 1. Khởi tạo nếu đây là lần chạy đầu tiên
     # chatbot_display là [[user_msg, ai_msg], ...]
     if chatbot_display is None:
@@ -138,7 +140,9 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history,lan
     
     # 5. Decode *chỉ* phần trả lời mới
     response_text = tokenizer.decode(output_tokens[0][inputs.shape[-1]:], skip_special_tokens=True)
-    translated =  translate_text(response_text,lang)
+    translated = response_text
+    if(lang not None)
+        translated =  translate_text(response_text,"en-"+lang)
     print(response_text)
     # 6. Cập nhật "bộ nhớ" (gr.State) với lượt nói MỚI
     internal_history.append({"role": "user", "content": prompt})