Update app.py

app.py

@@ -3,13 +3,11 @@ from fastapi.responses import JSONResponse
 import shutil
 import os
 
-# ✅ Fix config/cache issues in Hugging Face Docker Space
 os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
 os.environ["YOLO_CONFIG_DIR"] = "/tmp/ultralytics"
 os.environ["XDG_CACHE_HOME"] = "/tmp"
 os.environ["FONTCONFIG_PATH"] = "/tmp"
 
-# Create required directories if they don't exist
 os.makedirs("/tmp/matplotlib", exist_ok=True)
 os.makedirs("/tmp/ultralytics", exist_ok=True)
 os.makedirs("/tmp/fontconfig", exist_ok=True)
@@ -23,15 +21,12 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 @app.post("/extract-invoice")
 async def extract_invoice(file: UploadFile = File(...)):
-    # Save the uploaded image temporarily
     file_location = os.path.join(UPLOAD_DIR, file.filename)
     with open(file_location, "wb") as f:
         shutil.copyfileobj(file.file, f)
 
-    # Run OCR + detection
     extracted_data = extract_invoice_data_from_image(file_location)
 
-    # Optionally, clean up the uploaded file
     os.remove(file_location)
 
     return JSONResponse(content=extracted_data)