Update app.py

app.py

@@ -10,18 +10,42 @@ from huggingface_hub import login
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Đăng nhập Hugging Face
-login(token=os.getenv("HF_TOKEN"))  # Lấy token từ biến môi trường
+try:
+    login(token=os.getenv("HF_TOKEN"))
+    logging.info("Logged in to Hugging Face Hub successfully")
+except Exception as e:
+    logging.error(f"Failed to login to Hugging Face Hub: {e}")
+    raise
 
 app = Flask(__name__)
 
 # Load mô hình
 logging.info("Loading nguyenvulebinh/vi-mrc-base...")
-qa_pipeline = pipeline(
-    "question-answering",
-    model="nguyenvulebinh/vi-mrc-base",
-    device=0 if torch.cuda.is_available() else -1
+try:
+    qa_pipeline = pipeline(
+        "question-answering",
+        model="nguyenvulebinh/vi-mrc-base",
+        device=0 if torch.cuda.is_available() else -1
+    )
+    logging.info("Model loaded successfully")
+except Exception as e:
+    logging.error(f"Failed to load model: {e}")
+    raise
+
+# Giao diện Gradio
+def gradio_answer(question, context):
+    result = qa_pipeline(question=question, context=context)
+    return result["answer"]
+
+iface = gr.Interface(
+    fn=gradio_answer,
+    inputs=["text", "text"],
+    outputs="text",
+    title="AgriBot: Hỏi đáp nông nghiệp",
+    description="Nhập câu hỏi và ngữ cảnh để nhận câu trả lời về nông nghiệp."
 )
 
+# Endpoint API
 @app.route("/api/answer", methods=["POST"])
 def answer():
     try:
@@ -39,18 +63,18 @@ def answer():
         logging.error(f"API error: {e}")
         return jsonify({"error": str(e)}), 500
 
-# Giao diện Gradio để thử nghiệm
-def gradio_answer(question, context):
-    result = qa_pipeline(question=question, context=context)
-    return result["answer"]
+# Health check
+@app.route("/", methods=["GET"])
+def health_check():
+    return jsonify({"status": "API is running, use /api/answer for API or Gradio interface"}), 200
 
-iface = gr.Interface(
-    fn=gradio_answer,
-    inputs=["text", "text"],
-    outputs="text",
-    title="AgriBot: Hỏi đáp nông nghiệp",
-    description="Nhập câu hỏi và ngữ cảnh để nhận câu trả lời về nông nghiệp."
-)
+# Phục vụ Gradio ở root
+@app.route("/", methods=["GET", "POST"])
+@app.route("/<path:path>", methods=["GET", "POST"])
+def serve_gradio(path=None):
+    return iface.__call__(request)
 
 if __name__ == "__main__":
+    logging.info("Starting Flask and Gradio app on port 7860...")
+    logging.info(f"Registered routes: {app.url_map}")
     app.run(host="0.0.0.0", port=7860)