Update app.py

app.py

@@ -4,19 +4,23 @@ import json
 import time
 import tempfile
 
+import gradio as gr
+from gradio.routes import mount_gradio_app
+
 import google.generativeai as genai
 from fastapi import FastAPI, HTTPException, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel, Field, validator
 
 from quiz import QuizManager, SUBJECTS, DIFFICULTIES
+from gradio_ui import build_gradio_app
 
 # ── Startup checks ───────────────────────────────────────────
 api_key = os.getenv("GEMINI_API_KEY")
 if not api_key:
     raise ValueError("GEMINI_API_KEY environment variable not set")
 
-# ── App ──────────────────────────────────────────────────────
+# ── FastAPI app ──────────────────────────────────────────────
 app = FastAPI(
     title="NEET/JEE Quiz Generator API",
     description="RAG-powered MCQ generator for competitive exam prep",
@@ -31,10 +35,10 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# ── Global quiz manager (model loaded once at startup) ────────
+# ── Global quiz manager ───────────────────────────────────────
 quiz_manager = QuizManager()
 
-# ── Request / Response models ─────────────────────────────────
+# ── Pydantic models ───────────────────────────────────────────
 class TextRequest(BaseModel):
     text: str = Field(..., min_length=100)
     subject: str = Field(...)
@@ -69,8 +73,8 @@ class HealthResponse(BaseModel):
     model: str
     version: str
 
-# ── Endpoints ─────────────────────────────────────────────────
-@app.get("/", response_model=HealthResponse)
+# ── API Endpoints ─────────────────────────────────────────────
+@app.get("/api/health", response_model=HealthResponse)
 async def health():
     return HealthResponse(
         status="healthy",
@@ -88,25 +92,22 @@ async def generate_from_text(req: TextRequest):
                 status_code=400,
                 detail="Notes too short. Please provide more content."
             )
-
         session = quiz_manager.create_quiz(
-            topic         = req.topic,
-            subject       = req.subject,
-            chapter       = req.chapter,
-            exam_type     = req.exam_type,
-            difficulty    = req.difficulty,
-            num_questions = req.num_questions
+            topic=req.topic,
+            subject=req.subject,
+            chapter=req.chapter,
+            exam_type=req.exam_type,
+            difficulty=req.difficulty,
+            num_questions=req.num_questions
         )
-
         return QuizResponse(
-            success              = True,
-            session_id           = session.session_id,
-            total_questions      = session.total_questions,
-            time_limit_minutes   = session.time_limit_minutes,
-            quiz                 = session.to_dict(),
-            generated_in_seconds = round(time.time() - start, 2)
+            success=True,
+            session_id=session.session_id,
+            total_questions=session.total_questions,
+            time_limit_minutes=session.time_limit_minutes,
+            quiz=session.to_dict(),
+            generated_in_seconds=round(time.time() - start, 2)
         )
-
     except HTTPException:
         raise
     except RuntimeError as e:
@@ -134,7 +135,9 @@ async def generate_from_pdf(
         )
 
     start = time.time()
-    with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as tmp:
+    with tempfile.NamedTemporaryFile(
+        suffix=".pdf", delete=False
+    ) as tmp:
         content  = await file.read()
         tmp.write(content)
         tmp_path = tmp.name
@@ -147,12 +150,12 @@ async def generate_from_pdf(
             num_questions=num_questions
         )
         return QuizResponse(
-            success              = True,
-            session_id           = session.session_id,
-            total_questions      = session.total_questions,
-            time_limit_minutes   = session.time_limit_minutes,
-            quiz                 = session.to_dict(),
-            generated_in_seconds = round(time.time() - start, 2)
+            success=True,
+            session_id=session.session_id,
+            total_questions=session.total_questions,
+            time_limit_minutes=session.time_limit_minutes,
+            quiz=session.to_dict(),
+            generated_in_seconds=round(time.time() - start, 2)
         )
     except Exception as e:
         raise HTTPException(
@@ -224,4 +227,14 @@ def _estimate_rank(percentage: float, exam_type: str) -> str:
         if percentage >= 70: return "IIT eligible"
         if percentage >= 55: return "NIT Top-tier eligible"
         if percentage >= 40: return "NIT eligible"
-        return "Below JEE cut-off range"
+        return "Below JEE cut-off range"
+
+# ── Mount Gradio at root "/" ──────────────────────────────────
+# API routes above still work at their paths
+# Gradio UI is served at "/"
+gradio_app = build_gradio_app()
+app = mount_gradio_app(
+    app=app,
+    blocks=gradio_app,
+    path="/"          # Gradio at root, API at /generate/... /score
+)