Update app.py

app.py

@@ -5,6 +5,8 @@ import gradio as gr
 import os
 import torch
 from huggingface_hub import login
+from werkzeug.middleware.dispatcher import DispatcherMiddleware
+from werkzeug.serving import run_simple
 
 # Cấu hình logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -35,7 +37,7 @@ def gradio_answer(question, context):
     result = qa_pipeline(question=question, context=context)
     return result["answer"]
 
-# Tạo Flask app để thêm endpoint API
+# Tạo Flask app cho API
 app = Flask(__name__)
 
 @app.route("/api/answer", methods=["POST"])
@@ -55,14 +57,19 @@ def answer():
         logging.error(f"API error: {e}")
         return jsonify({"error": str(e)}), 500
 
-# 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."
-)
+# Tạo Gradio Blocks
+with gr.Blocks() as demo:
+    gr.Markdown("# AgriBot: Hỏi đáp nông nghiệp")
+    gr.Markdown("Nhập câu hỏi và ngữ cảnh để nhận câu trả lời về nông nghiệp.")
+    question_input = gr.Textbox(label="Câu hỏi", placeholder="Nhập câu hỏi của bạn...")
+    context_input = gr.Textbox(label="Ngữ cảnh", placeholder="Nhập ngữ cảnh liên quan...")
+    output = gr.Textbox(label="Câu trả lời")
+    submit_btn = gr.Button("Gửi")
+    submit_btn.click(fn=gradio_answer, inputs=[question_input, context_input], outputs=output)
 
-# Thêm Flask app vào Gradio
-iface.launch(server_name="0.0.0.0", server_port=7860, app=app)
+# Tích hợp Flask và Gradio trên cùng port
+application = DispatcherMiddleware(demo.app, {"/api": app})
+
+if __name__ == "__main__":
+    logging.info("Starting Gradio and Flask on port 7860...")
+    run_simple("0.0.0.0", 7860, application)