Update main.py

main.py

@@ -22,6 +22,11 @@ client = genai.Client(api_key=GENAI_API_KEY)
 
 app = FastAPI(title="Student Result Card API")
 
+# Ensure downloads folder exists
+DOWNLOADS_FOLDER = "downloads"
+if not os.path.exists(DOWNLOADS_FOLDER):
+    os.makedirs(DOWNLOADS_FOLDER)
+
 # -----------------------------
 # Preprocessing Methods
 # -----------------------------
@@ -294,28 +299,48 @@ async def process_pdfs(
             result_card["Student Index"] = idx + 1
             student_result_cards.append(result_card)
         
+        response_data = {"result_cards": student_result_cards}
+        
         if download:
-            # Create downloadable JSON file
-            json_bytes = json.dumps({"result_cards": student_result_cards}, indent=2).encode("utf-8")
+            # Create downloadable JSON file and save to disk
+            json_bytes = json.dumps(response_data, indent=2).encode("utf-8")
+            file_path = os.path.join(DOWNLOADS_FOLDER, "result_cards.json")
+            with open(file_path, "wb") as f:
+                f.write(json_bytes)
             return StreamingResponse(
                 io.BytesIO(json_bytes),
                 media_type="application/json",
                 headers={"Content-Disposition": "attachment; filename=result_cards.json"}
             )
         else:
-            return JSONResponse(content={"result_cards": student_result_cards})
+            return JSONResponse(content=response_data)
     
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+# -----------------------------
+# New Download Endpoint
+# -----------------------------
+@app.get("/download")
+async def download_result_cards():
+    """
+    Returns the previously generated result_cards.json file.
+    """
+    file_path = os.path.join(DOWNLOADS_FOLDER, "result_cards.json")
+    if not os.path.exists(file_path):
+         raise HTTPException(status_code=404, detail="File not found")
+    return StreamingResponse(
+         open(file_path, "rb"),
+         media_type="application/json",
+         headers={"Content-Disposition": "attachment; filename=result_cards.json"}
+    )
 
 @app.get("/")
 async def root():
     return {
         "message": "Welcome to the Student Result Card API.",
-        "usage": "POST PDFs to /process with 'student_pdf' and 'answer_key_pdf' fields. Use ?download=true for file download."
+        "usage": "POST PDFs to /process with 'student_pdf' and 'answer_key_pdf' fields. Use ?download=true for file download or GET /download to re-download the JSON file."
     }
 
-
 if __name__ == "__main__":
     uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)