white balance api added

callaunchpad · Dec 7, 2024 · d3a5710 · d3a5710
1 parent 4a0a4db
commit d3a5710
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 0 deletions.
diff --git a/combined_demo/input-imgs/input.jpg b/combined_demo/input-imgs/input.jpg
diff --git a/combined_demo/intermediate-imgs/white-balanced.jpg b/combined_demo/intermediate-imgs/white-balanced.jpg
diff --git a/combined_demo/output-imgs/cropped.jpg b/combined_demo/output-imgs/cropped.jpg
diff --git a/combined_demo/output-imgs/redbox.jpg b/combined_demo/output-imgs/redbox.jpg
diff --git a/combined_demo/output-imgs/your-palette.jpg b/combined_demo/output-imgs/your-palette.jpg
diff --git a/main.py b/main.py
@@ -128,6 +128,14 @@ async def get_cropped():
 
     return FileResponse(cropped_path, media_type="image/jpg")
 
+@app.get("/aura_analyze/white-balanced")
+async def get_corrected():
+    corrected_path = "combined_demo/intermediate-imgs/white-balanced.jpg"
+    if not os.path.exists(corrected_path):
+        raise HTTPException(status_code=404, detail=f"Corrected image not found")
+
+    return FileResponse(corrected_path, media_type="image/jpg")
+
 @app.get("/aura_analyze/palette")
 async def get_palette():
     palette_path = "combined_demo/output-imgs/your-palette.jpg"