HamidOmarov commited on
Commit
26b3f58
·
1 Parent(s): 2200095

Add /generate alias for PDF-QA compatibility

Browse files
Files changed (1) hide show
  1. main.py +15 -0
main.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # --- ADD: /generate alias for compatibility ---
2
+ from fastapi import Form
3
+
4
+ @app.post("/generate")
5
+ async def generate_alias(
6
+ file: UploadFile = File(...),
7
+ num_questions: int = Query(20, ge=1, le=200),
8
+ language: str = Query("en")
9
+ ):
10
+ # simply call the real generator
11
+ return await generate_dataset(file=file, num_questions=num_questions, language=language)
12
+
13
+ @app.get("/health")
14
+ def health_alias():
15
+ return {"status": "ok"}