Update app.py

app.py

@@ -1,10 +1,11 @@
 import logging
-from flask import Flask, request, jsonify
 import gradio as gr
 from transformers import pipeline
 import os
 import torch
 from huggingface_hub import login
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
 
 # Cấu hình logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -35,36 +36,37 @@ def gradio_answer(question, context):
     result = qa_pipeline(question=question, context=context)
     return result["answer"]
 
+# Tạo FastAPI app để thêm endpoint API
+app = FastAPI()
+
+@app.post("/api/answer")
+async def api_answer(request: Request):
+    try:
+        data = await request.json()
+        question = data.get("question")
+        context = data.get("context")
+        logging.info(f"Received request - Question: {question}, Context: {context[:200]}...")
+        if not question or not context:
+            logging.error("Missing question or context")
+            return JSONResponse({"error": "Missing question or context"}, status_code=400)
+        result = qa_pipeline(question=question, context=context)
+        logging.info(f"Response - Answer: {result['answer']}")
+        return JSONResponse({"answer": result["answer"]})
+    except Exception as e:
+        logging.error(f"API error: {e}")
+        return JSONResponse({"error": str(e)}, status_code=500)
+
 # Tạo Gradio Blocks
-with gr.Blocks() as demo:
+with gr.Blocks(app=app) 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.")
     
-    # Giao diện Gradio
     with gr.Row():
         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)
-    
-    # Endpoint API
-    @demo.route("/api/answer", methods=["POST"])
-    async def api_answer(request):
-        try:
-            data = await request.json()
-            question = data.get("question")
-            context = data.get("context")
-            logging.info(f"Received request - Question: {question}, Context: {context[:200]}...")
-            if not question or not context:
-                logging.error("Missing question or context")
-                return jsonify({"error": "Missing question or context"}), 400
-            result = qa_pipeline(question=question, context=context)
-            logging.info(f"Response - Answer: {result['answer']}")
-            return jsonify({"answer": result["answer"]})
-        except Exception as e:
-            logging.error(f"API error: {e}")
-            return jsonify({"error": str(e)}), 500
 
 # Chạy ứng dụng
 if __name__ == "__main__":