Update main.py

main.py

@@ -4,6 +4,7 @@ import uvicorn
 import io
 import json
 import os
+import tempfile
 import numpy as np
 import cv2
 from PIL import Image
@@ -22,10 +23,8 @@ 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)
+# Use the system temporary directory
+TEMP_FOLDER = tempfile.gettempdir()
 
 # -----------------------------
 # Preprocessing Methods
@@ -302,9 +301,9 @@ async def process_pdfs(
         response_data = {"result_cards": student_result_cards}
         
         if download:
-            # Create downloadable JSON file and save to disk
+            # Create downloadable JSON file and save to system temp folder
             json_bytes = json.dumps(response_data, indent=2).encode("utf-8")
-            file_path = os.path.join(DOWNLOADS_FOLDER, "result_cards.json")
+            file_path = os.path.join(TEMP_FOLDER, "result_cards.json")
             with open(file_path, "wb") as f:
                 f.write(json_bytes)
             return StreamingResponse(
@@ -324,9 +323,9 @@ async def process_pdfs(
 @app.get("/download")
 async def download_result_cards():
     """
-    Returns the previously generated result_cards.json file.
+    Returns the previously generated result_cards.json file from the system temporary folder.
     """
-    file_path = os.path.join(DOWNLOADS_FOLDER, "result_cards.json")
+    file_path = os.path.join(TEMP_FOLDER, "result_cards.json")
     if not os.path.exists(file_path):
          raise HTTPException(status_code=404, detail="File not found")
     return StreamingResponse(