Add endpoint to list all extracted images

app/api/routes/images.py

@@ -0,0 +1,23 @@
+from fastapi import APIRouter
+from pathlib import Path
+import os
+
+router = APIRouter(prefix="/images", tags=["Images"])
+
+
+@router.get("/all")
+def list_images():
+
+    BASE_DATA_DIR = os.getenv("HF_HOME", ".")
+    OUTPUT_DIR = os.path.join(BASE_DATA_DIR, "outputs")
+
+    if not os.path.exists(OUTPUT_DIR):
+        return {"images": []}
+
+    image_urls = [
+        f"outputs/{file.name}"
+        for file in Path(OUTPUT_DIR).glob("*")
+        if file.suffix.lower() in [".png", ".jpg", ".jpeg"]
+    ]
+
+    return {"images": image_urls}

app/main.py

@@ -4,6 +4,9 @@ from fastapi.staticfiles import StaticFiles
 
 from app.api.routes import rag, visualize, grading, upload
 
+from app.api.routes import images
+
+
 
 # ✅ Use persistent storage on HF, local folder otherwise
 BASE_DATA_DIR = os.getenv("HF_HOME", ".")
@@ -20,6 +23,8 @@ app.include_router(rag.router)
 app.include_router(visualize.router)
 app.include_router(grading.router)
 app.include_router(upload.router)
+app.include_router(images.router)
+
 
 
 # ✅ Mount the real path (not hardcoded "outputs")