Update app.py

app.py

@@ -36,7 +36,7 @@ def gradio_answer(question, context):
     result = qa_pipeline(question=question, context=context)
     return result["answer"]
 
-# Tạo FastAPI app để thêm endpoint API
+# Tạo FastAPI app
 app = FastAPI()
 
 @app.post("/api/answer")
@@ -44,11 +44,11 @@ async def api_answer(request: Request):
     try:
         data = await request.json()
         question = data.get("question")
-        context = data.get("context")
+        context = data.get("context", "Cây lúa là một loại cây trồng phổ biến ở Việt Nam, cần điều kiện đất và nước phù hợp.")
         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)
+        if not question:
+            logging.error("Missing question")
+            return JSONResponse({"error": "Missing question"}, status_code=400)
         result = qa_pipeline(question=question, context=context)
         logging.info(f"Response - Answer: {result['answer']}")
         return JSONResponse({"answer": result["answer"]})
@@ -56,7 +56,7 @@ async def api_answer(request: Request):
         logging.error(f"API error: {e}")
         return JSONResponse({"error": str(e)}, status_code=500)
 
-# Tạo Gradio Blocks
+# Tạo Gradio Blocks và mount vào FastAPI
 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.")
@@ -68,11 +68,12 @@ with gr.Blocks() as demo:
     submit_btn = gr.Button("Gửi")
     submit_btn.click(fn=gradio_answer, inputs=[question_input, context_input], outputs=output)
 
+# Mount Gradio vào FastAPI
+app = gr.integration.FastAPI(app)
+app.mount("/", demo)
+
 # Chạy ứng dụng
 if __name__ == "__main__":
-    logging.info("Starting Gradio on port 7860...")
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        inline=False,
-    )
+    import uvicorn
+    logging.info("Starting app on port 80...")
+    uvicorn.run(app, host="0.0.0.0", port=80)