© 2025 by JerryCoder

app.py

@@ -1,80 +1,95 @@
 import requests
 import json
+import uuid
 from fastapi import FastAPI, UploadFile, File, Query
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, Response
 from fastapi.middleware.cors import CORSMiddleware
 from rembg import remove
 import uvicorn
 
 # --------------------------
-# Custom Pretty JSON Response
+# Temp Memory Storage
+# --------------------------
+TEMP_FILES = {}
+
+# --------------------------
+# Pretty JSON Response
 # --------------------------
 class PrettyJSONResponse(JSONResponse):
     def render(self, content) -> bytes:
         return json.dumps(content, indent=4, ensure_ascii=False).encode("utf-8")
 
-app = FastAPI(title="AI ClearCut - Background Remover | JerryCoder", default_response_class=PrettyJSONResponse)
+app = FastAPI(
+    title="AI ClearCut - HF UltraFast BG Remover",
+    default_response_class=PrettyJSONResponse
+)
 
 # --------------------------
-# Enable CORS
+# CORS
 # --------------------------
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],       # Allow all origins
+    allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["*"],       # Allow all HTTP methods
-    allow_headers=["*"],       # Allow all headers
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
 # --------------------------
-# Root endpoint
+# Root
 # --------------------------
 @app.get("/")
 async def root():
     return {
-        "message": "AI ClearCut Background Remover API is running! © 2025 by JerryCoder",
+        "message": "HF UltraFast Background Remover Running 🚀",
         "endpoints": {
-            "POST /upload": "Upload an image file for background removal",
-            "GET /json?url=": "Provide an image URL for background removal"
+            "POST /upload": "Upload image",
+            "GET /json?url=": "Image from URL",
+            "GET /file/{id}": "Get processed image"
         },
-        "creator": "Jerrycoder",
-        "telegram": "Oggy_workshop"
+        "creator": "Jerrycoder"
     }
 
 # --------------------------
-# Helper: process image bytes
+# Serve temp file
+# --------------------------
+@app.get("/file/{file_id}")
+async def get_file(file_id: str):
+    file_data = TEMP_FILES.get(file_id)
+    if not file_data:
+        return {"status": "error", "message": "File expired"}
+
+    return Response(content=file_data, media_type="image/png")
+
+# --------------------------
+# Process image
 # --------------------------
 def process_image(img_bytes: bytes):
     try:
-        # Fast background removal
+        # Remove background
         output_bytes = remove(img_bytes)
 
-        # Upload to Uguu
-        files = {
-            "files[]": ("output.png", output_bytes, "image/png")
-        }
+        # Generate unique ID
+        file_id = str(uuid.uuid4())
 
-        r = requests.post("https://uguu.se/upload", files=files, timeout=20)
-        r.raise_for_status()
-        data = r.json()
+        # Store temporarily
+        TEMP_FILES[file_id] = output_bytes
 
-        # Uguu returns a list
-        if isinstance(data, list) and len(data) > 0:
-            return {
-                "status": "success",
-                "url": data[0].get("url"),
-                "name": data[0].get("name"),
-                "size": data[0].get("size"),
-                "creator": "Jerrycoder",
-                "telegram": "Oggy_workshop"
-            }
+        # Build URL
+        file_url = f"/file/{file_id}"
 
-        return {"status": "error", "message": "Upload failed"}
+        return {
+            "status": "success",
+            "url": file_url,
+            "full_url": file_url,
+            "creator": "Jerrycoder"
+        }
 
     except Exception as e:
         return {"status": "error", "message": str(e)}
+
 # --------------------------
-# POST: Upload file
+# POST upload
 # --------------------------
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
@@ -82,26 +97,19 @@ async def upload_file(file: UploadFile = File(...)):
     return process_image(img_bytes)
 
 # --------------------------
-# GET: From image URL
+# GET from URL
 # --------------------------
 @app.get("/json")
-async def from_url(url: str = Query(..., description="Image URL")):
+async def from_url(url: str = Query(...)):
     try:
-        headers = {
-            "User-Agent": (
-                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
-                "AppleWebKit/537.36 (KHTML, like Gecko) "
-                "Chrome/115.0 Safari/537.36"
-            )
-        }
-        resp = requests.get(url, headers=headers, timeout=15)
+        resp = requests.get(url, timeout=10)
         resp.raise_for_status()
         return process_image(resp.content)
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
 # --------------------------
-# HF Entry point
+# Run
 # --------------------------
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)