Update backend/app.py

backend/app.py

@@ -236,6 +236,26 @@ os.makedirs(IMAGES_DIR, exist_ok=True)
 
 app.mount("/outputs", StaticFiles(directory=OUTPUT_DIR), name="outputs")
 
+# Mount public sample images from frontend dist (Vite copies public/ to dist/ root)
+# Check both possible locations: frontend/dist (Docker) and ../frontend/dist (local dev)
+FRONTEND_DIST_CHECK = os.path.join(os.path.dirname(__file__), "frontend/dist")
+if not os.path.isdir(FRONTEND_DIST_CHECK):
+    FRONTEND_DIST_CHECK = os.path.abspath(os.path.join(os.path.dirname(__file__), "../frontend/dist"))
+
+for sample_dir in ["cyto", "colpo", "histo"]:
+    sample_path = os.path.join(FRONTEND_DIST_CHECK, sample_dir)
+    if os.path.isdir(sample_path):
+        app.mount(f"/{sample_dir}", StaticFiles(directory=sample_path), name=sample_dir)
+        print(f"✅ Mounted /{sample_dir} from {sample_path}")
+    else:
+        print(f"⚠️ Sample directory not found: {sample_path}")
+
+# Mount other static assets (logos, banners) from dist root
+for static_file in ["banner.jpeg", "white_logo.png", "black_logo.png", "manalife_LOGO.jpg"]:
+    static_path = os.path.join(FRONTEND_DIST_CHECK, static_file)
+    if os.path.isfile(static_path):
+        print(f"✅ Static file available: /{static_file}")
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # =====================================================
@@ -1021,6 +1041,25 @@ async def serve_frontend():
         return FileResponse(index_path)
     return JSONResponse({"message": "Backend is running. Frontend build not found."})
 
+@app.get("/{file_path:path}")
+async def serve_static_files(file_path: str):
+    """Serve static files from frontend dist (images, logos, etc.)"""
+    # Skip API routes
+    if file_path.startswith(("predict", "reports", "models", "health", "outputs", "assets", "cyto", "colpo", "histo")):
+        return JSONResponse({"error": "Not found"}, status_code=404)
+    
+    # Try to serve file from dist root
+    static_file = os.path.join(FRONTEND_DIST, file_path)
+    if os.path.isfile(static_file):
+        return FileResponse(static_file)
+    
+    # Fallback to index.html for client-side routing
+    index_path = os.path.join(FRONTEND_DIST, "index.html")
+    if os.path.isfile(index_path):
+        return FileResponse(index_path)
+    
+    return JSONResponse({"error": "Not found"}, status_code=404)
+
 if __name__ == "__main__":
     # Use PORT provided by the environment (Hugging Face Spaces sets PORT=7860)
     port = int(os.environ.get("PORT", 7860))