Create fallback.py

fallback.py

@@ -0,0 +1,95 @@
+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 os
+import sys
+from PIL import Image
+
+# 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(),
+            "mode": "fallback - no background removal"
+        }
+    }
+
+@app.post("/remove-bg")
+async def remove_background(file: UploadFile = File(...)):
+    """
+    This is a fallback version that just adds a white background
+    instead of removing the background, in case rembg doesn't work
+    """
+    # 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))
+        
+        # Create a white background image
+        white_bg = Image.new("RGBA", input_image.size, (255, 255, 255, 255))
+        
+        # Paste the input image on the white background
+        if input_image.mode == 'RGBA':
+            white_bg.paste(input_image, (0, 0), input_image)
+        else:
+            white_bg.paste(input_image, (0, 0))
+        
+        # Convert to bytes
+        img_byte_arr = io.BytesIO()
+        white_bg.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)