Update check.py

check.py

@@ -21,7 +21,7 @@ from google.genai import types
 # Initialize the GenAI client with the API key.
 client = genai.Client(api_key=GENAI_API_KEY)
 
-app = FastAPI(title="Student Result Card API (Paper K only)")
+router = APIRouter(prefix="/check", tags=["check"])
 
 # Use system temporary directory to store the results file.
 TEMP_FOLDER = tempfile.gettempdir()
@@ -165,7 +165,7 @@ def load_answer_key(pdf_bytes: bytes) -> dict:
 # FastAPI Endpoints
 ##############################################################
 
-@app.post("/process")
+@@router.post("/process")
 async def process_pdfs(
     original_pdf: UploadFile = File(..., description="PDF with all student answer sheets (one page per student)"),
     paper_k_pdf:  UploadFile = File(..., description="Answer key PDF for Paper K")
@@ -227,7 +227,7 @@ async def process_pdfs(
         raise HTTPException(status_code=500, detail=str(e))
 
 
-@app.get("/download")
+@@router.get("/download")
 async def download_results():
     """
     Returns the result JSON file stored in the temporary folder.
@@ -241,7 +241,7 @@ async def download_results():
     )
 
 
-@app.get("/")
+@@router.get("/")
 async def root():
     return {
         "message": "Welcome to the Student Result Card API (Paper K only).",