Update app.py

app.py

@@ -138,6 +138,26 @@ def safe_format_result(result) -> str:
     except Exception as e:
         return f"Error formatting result: {e}"
 
+
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
+
+api_app = FastAPI()
+
+@api_app.post("/multi_agent_chat")
+async def multi_agent_chat(request: Request):
+    data = await request.json()
+    query = data.get("query")
+    result = multi_agent_chat_advanced(query)
+    return JSONResponse(content={"result": result})
+
+@api_app.post("/multi_doc_qa")
+async def multi_doc_qa(request: Request):
+    data = await request.json()
+    query = data.get("query")
+    result = langgraph_tab6_main(query)
+    return JSONResponse(content={"result": result})
+    
 # Model and Device Setup 
 if torch.backends.mps.is_available():
     device = "mps"
@@ -1899,4 +1919,15 @@ demo = gr.TabbedInterface(
 )
 
 if __name__ == "__main__":
+    import threading
+    import uvicorn
+
+    # 開 FastAPI 在另一個 thread
+    def run_fastapi():
+        uvicorn.run("app:api_app", host="0.0.0.0", port=8000)
+
+    fastapi_thread = threading.Thread(target=run_fastapi)
+    fastapi_thread.start()
+
+    # 同時跑 Gradio
     demo.launch(server_name="0.0.0.0", server_port=7860, share=False)