capcut8128 commited on
Commit
058b118
·
verified ·
1 Parent(s): 8b8c6e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -37
app.py CHANGED
@@ -1,41 +1,13 @@
1
  import gradio as gr
2
- import spaces
3
- import torch
4
- from fastapi import FastAPI
5
- from fastapi.middleware.cors import CORSMiddleware
6
 
7
- # FastAPI app banao
8
- app = FastAPI()
9
 
10
- # CORS enable karo taaki frontend call kar sake
11
- app.add_middleware(
12
- CORSMiddleware,
13
- allow_origins=["*"],
14
- allow_methods=["*"],
15
- allow_headers=["*"],
16
- )
17
 
18
- # --- API Routes ---
19
- @app.get("/documents")
20
- def get_documents():
21
- # Yahan apne actual documents return karo
22
- return {"documents": ["doc1.pdf", "doc2.pdf"]}
23
-
24
- @app.post("/ask")
25
- @spaces.GPU
26
- def ask_question(payload: dict):
27
- question = payload.get("question", "")
28
- # Yahan apna RAG / QA logic likho
29
- zero = torch.Tensor([0]).to("cuda")
30
- return {"answer": f"Answer for: {question}", "tensor": str(zero)}
31
-
32
- # --- Gradio UI (optional) ---
33
- @spaces.GPU
34
- def greet(n):
35
- zero = torch.Tensor([0]).to("cuda")
36
- return f"Hello {zero + n} Tensor"
37
-
38
- demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
39
-
40
- # Gradio ko FastAPI ke andar mount karo
41
- app = gr.mount_gradio_app(app, demo, path="/ui")
 
1
  import gradio as gr
2
+ import uvicorn
3
+ from app.main import app as fastapi_app
 
 
4
 
5
+ # Dummy Gradio (SDK detection ke liye)
6
+ demo = gr.Interface(fn=lambda x: x, inputs="text", outputs="text")
7
 
8
+ # FastAPI app ko Gradio ke saath mount karo
9
+ app = gr.mount_gradio_app(fastapi_app, demo, path="/ui")
 
 
 
 
 
10
 
11
+ # Gradio SDK ki jagah khud uvicorn chalao
12
+ if __name__ == "__main__":
13
+ uvicorn.run(app, host="0.0.0.0", port=7860)