Delete main.py

main.py

@@ -1,90 +0,0 @@
-import os
-# Disable Numba caching to avoid the error
-os.environ["NUMBA_DISABLE_JIT"] = "1"
-
-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
-import sys
-from PIL import Image
-from rembg import remove
-
-# Print debug information
-print("Python version:", sys.version)
-print("Current directory:", os.getcwd())
-print("Directory contents:", os.listdir())
-
-# Create the FastAPI app
-app = FastAPI()
-
-# Add CORS middleware
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Mount static files directory if it exists
-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():
-    """Root endpoint that redirects to the static HTML or returns a status message"""
-    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",
-        "environment": {
-            "python_version": sys.version,
-            "directory": os.getcwd(),
-            "files": os.listdir(),
-            "numba_disabled": os.environ.get("NUMBA_DISABLE_JIT", "Not set")
-        }
-    }
-
-@app.post("/remove-bg")
-async def remove_background(file: UploadFile = File(...)):
-    """Remove the background from an uploaded image"""
-    # 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)}")
-
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)