Update app.py

app.py

@@ -1,8 +1,12 @@
 from flask import Flask, render_template, request, jsonify
 import requests
 import os
+import logging
 from flask_cors import CORS
 
+# Cấu hình logging để debug
+logging.basicConfig(level=logging.INFO)
+
 app = Flask(__name__)
 CORS(app)
 
@@ -10,23 +14,32 @@ CORS(app)
 HF_MODEL = "google/gemma-2b"
 HF_API_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
 
-# Lấy API Key từ Hugging Face Secrets
+# 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.")
 
-# Gửi request API để chat
+# Hàm gửi request đến API Hugging Face
 def chat_with_gemma(prompt):
-    if not HF_API_KEY:
-        return "Lỗi: API Key không tồn tại. Hãy thêm vào Hugging Face Secrets."
-
-    response = requests.post(
-        HF_API_URL,
-        headers={"Authorization": f"Bearer {HF_API_KEY}"},
-        json={"inputs": prompt}
-    )
-
-    if response.status_code == 200:
-        return response.json()[0].get('generated_text', "Không có phản hồi.")
-    return f"Lỗi: {response.status_code} - {response.text}"
+    logging.info(f"Gửi request đến API: {HF_API_URL}")
+    logging.info(f"Nội dung gửi: {prompt}")
+
+    headers = {"Authorization": f"Bearer {HF_API_KEY}"}
+    payload = {"inputs": prompt}
+    
+    try:
+        response = requests.post(HF_API_URL, headers=headers, json=payload)
+        logging.info(f"Trạng thái API: {response.status_code}")
+        logging.info(f"Nội dung trả về: {response.text}")
+        
+        if response.status_code == 200:
+            result = response.json()
+            return result[0].get("generated_text", "Không có phản hồi.")
+        return f"Lỗi: {response.status_code} - {response.text}"
+    
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Lỗi khi gọi API: {str(e)}")
+        return "Lỗi: Không thể kết nối đến API."
 
 @app.route('/')
 def index():
@@ -40,4 +53,4 @@ def chat():
     return jsonify({'FROM': 'Gemma-2B', 'MESSAGE': bot_reply})
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)), debug=False)