Update app.py

app.py

@@ -16,11 +16,15 @@ HF_API_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
 
 # Lấy API Key từ biến môi trường
 HF_API_KEY = os.getenv("HF_API_KEY")
+
 if not HF_API_KEY:
-    raise ValueError("Lỗi: API Key không tồn tại. Hãy thêm vào Hugging Face Secrets.")
+    logging.error("Lỗi: API Key không tồn tại. Hãy thêm vào Hugging Face Secrets.")
 
 # Hàm gửi request đến API Hugging Face
 def chat_with_tinyllama(prompt):
+    if not HF_API_KEY:
+        return "Lỗi: API Key không tồn tại."
+
     logging.info(f"Gửi request đến API: {HF_API_URL}")
     logging.info(f"Nội dung gửi: {prompt}")
 
@@ -33,7 +37,13 @@ def chat_with_tinyllama(prompt):
 
         if response.status_code == 200:
             result = response.json()
-            return result[0].get("generated_text", "Không có phản hồi từ mô hình.")
+            # Kiểm tra định dạng JSON trả về
+            if isinstance(result, list) and "generated_text" in result[0]:
+                return result[0]["generated_text"]
+            elif isinstance(result, dict) and "generated_text" in result:
+                return result["generated_text"]
+            else:
+                return "Lỗi: Định dạng phản hồi không hợp lệ."
         elif response.status_code == 503:
             return "Mô hình đang quá tải, vui lòng thử lại sau."
         else:
@@ -44,11 +54,17 @@ def chat_with_tinyllama(prompt):
     except requests.exceptions.RequestException as e:
         return f"Lỗi mạng: {str(e)}"
 
+# Route API kiểm tra server
+@app.route('/')
+def index():
+    return jsonify({'message': 'TinyLlama API is running!'})
+
 # Route API chat
 @app.route('/chat', methods=['POST'])
 def chat():
     data = request.json
-    user_message = data.get("message", "")
+    user_message = data.get("message", "").strip()
+
     if not user_message:
         return jsonify({'FROM': 'TinyLlama', 'MESSAGE': 'Lỗi: Tin nhắn không hợp lệ.'})
 
@@ -57,4 +73,4 @@ def chat():
 
 # Chạy Flask server
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860, debug=True)