Update app.py

app.py

@@ -1,11 +1,13 @@
 from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import StreamingResponse
+from fastapi.responses import StreamingResponse, FileResponse, RedirectResponse
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
 import io
 from PIL import Image
 from rembg import remove
 import os
 import sys
+import platform
 
 # Print debug information
 print("Python version:", sys.version)
@@ -23,9 +25,37 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Mount static files directory
+if os.path.exists("static"):
+    app.mount("/static", StaticFiles(directory="static"), name="static")
+    print("Static directory mounted successfully")
+else:
+    print("Warning: Static directory not found")
+
 @app.get("/")
 def read_root():
-    return {"message": "Background Removal API is running. Use POST /remove-bg to remove background from an image."}
+    # Redirect to static HTML if it exists
+    if os.path.exists("static/index.html"):
+        return RedirectResponse(url="/static/index.html")
+    else:
+        return {"message": "Background Removal API is running. Use POST /remove-bg to remove background from an image."}
+
+@app.get("/health")
+def health_check():
+    """Health check endpoint to verify the API is running correctly"""
+    return {
+        "status": "ok",
+        "python_version": platform.python_version(),
+        "system": platform.system(),
+        "dependencies": {
+            "fastapi": "installed",
+            "pillow": "installed",
+            "rembg": "installed"
+        },
+        "directories": os.listdir(),
+        "static_exists": os.path.exists("static"),
+        "static_files": os.listdir("static") if os.path.exists("static") else []
+    }
 
 @app.post("/remove-bg")
 async def remove_background(file: UploadFile = File(...)):