Update app.py

app.py

@@ -1,24 +1,32 @@
 from flask import Flask, render_template, request, jsonify
 import requests
+import os
 from flask_cors import CORS
 
 app = Flask(__name__)
 CORS(app)
 
-# Model Hugging Face (Không cần tải về)
+# Model Hugging Face
 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
+HF_API_KEY = os.getenv("HF_API_KEY")
+
 # Gửi request API để chat
 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 YOUR_HF_API_KEY"},  # Thay YOUR_HF_API_KEY bằng API Key của bạn
+        headers={"Authorization": f"Bearer {HF_API_KEY}"},
         json={"inputs": prompt}
     )
+
     if response.status_code == 200:
-        return response.json()[0]['generated_text']
-    return "Lỗi: Không thể lấy phản hồi từ API."
+        return response.json()[0].get('generated_text', "Không có phản hồi.")
+    return f"Lỗi: {response.status_code} - {response.text}"
 
 @app.route('/')
 def index():