Update app.py

app.py

@@ -17,8 +17,6 @@ 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...")
 try:
@@ -32,20 +30,14 @@ except Exception as e:
     logging.error(f"Failed to load model: {e}")
     raise
 
-# Giao diện Gradio
+# Hàm xử lý cho Gradio và API
 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."
-)
+# Tạo Flask app để thêm endpoint API
+app = Flask(__name__)
 
-# Endpoint API
 @app.route("/api/answer", methods=["POST"])
 def answer():
     try:
@@ -63,15 +55,14 @@ def answer():
         logging.error(f"API error: {e}")
         return jsonify({"error": str(e)}), 500
 
-# Health check (chỉ dùng khi cần debug)
-@app.route("/health", methods=["GET"])
-def health_check():
-    return jsonify({"status": "API is running, use /api/answer for API or / for Gradio interface"}), 200
-
-# Phục vụ Gradio ở root
-app = gr.mount_gradio_app(app, iface, path="/")
+# Tạo Gradio interface
+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."
+)
 
-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)
+# Thêm Flask app vào Gradio
+iface.launch(server_name="0.0.0.0", server_port=7860, app=app)