commit 00000004

app.py

@@ -9,6 +9,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from huggingface_hub import login
 from langchain_community.tools import DuckDuckGoSearchRun
 import re
+from fastapi import FastAPI
+from starlette.middleware.wsgi import WSGIMiddleware
 from gradio.routes import mount_gradio_app
 
 # ✅ Safe GPU decorator
@@ -178,8 +180,14 @@ with gr.Blocks() as demo:
 
 # # Start Flask in a background thread
 # threading.Thread(target=run_flask, daemon=True).start()
-app = gr.routes.App.create_app(demo)
-app.mount("/flask", app=app)  # Mount Flask under /flask
+# ---------------- Combine Flask + Gradio into one app ----------------
+fastapi_app = FastAPI()
+
+# Mount Flask under FastAPI (so /apidocs works)
+fastapi_app.mount("/", WSGIMiddleware(flask_app))
+
+# Mount Gradio at root path (overrides Flask's "/")
+app = mount_gradio_app(fastapi_app, demo, path="/") # Mount Flask under /flask
 # Gradio runs on port 7860 in HF Spaces
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+# if __name__ == "__main__":
+#     demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -14,4 +14,6 @@ sentencepiece
 nltk
 langchain_community
 duckduckgo-search
-pdfplumber
+pdfplumber
+fastapi
+uvicorn