Update app.py

app.py

@@ -1,91 +1,4 @@
 # Import app from main.py for backward compatibility
 from main import app
 
-# This file ensures compatibility with both app:app and main:app entrypoints
-
-from fastapi import FastAPI, File, UploadFile, HTTPException
-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)
-print("Current directory:", os.getcwd())
-print("Directory contents:", os.listdir())
-
-app = FastAPI(title="Background Removal API")
-
-# Add CORS middleware to allow requests from any origin
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    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():
-    # 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(...)):
-    # Check if the file is an image
-    if not file.content_type.startswith("image/"):
-        raise HTTPException(status_code=400, detail="File must be an image")
-    
-    try:
-        # Read the image
-        contents = await file.read()
-        input_image = Image.open(io.BytesIO(contents))
-        
-        # Remove the background
-        output_image = remove(input_image)
-        
-        # Convert to bytes
-        img_byte_arr = io.BytesIO()
-        output_image.save(img_byte_arr, format="PNG")
-        img_byte_arr.seek(0)
-        
-        # Return the processed image
-        return StreamingResponse(
-            content=img_byte_arr, 
-            media_type="image/png"
-        )
-    
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error processing image: {str(e)}") 
+# This file ensures compatibility with both app:app and main:app entrypoints