Update app.py

app.py

@@ -6,7 +6,14 @@ from fastapi.middleware.cors import CORSMiddleware
 from rembg import remove
 import uvicorn
 
-app = FastAPI(title="AI ClearCut - Background Remover | JerryCoder")
+# --------------------------
+# Custom 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)
 
 # --------------------------
 # Enable CORS
@@ -29,7 +36,9 @@ async def root():
         "endpoints": {
             "POST /upload": "Upload an image file for background removal",
             "GET /json?url=": "Provide an image URL for background removal"
-        }
+        },
+        "creator": "Jerrycoder",
+        "telegram": "Oggy_workshop"
     }
 
 # --------------------------
@@ -41,20 +50,18 @@ def process_image(img_bytes: bytes):
         output_bytes = remove(img_bytes)
 
         # Upload to ar-hosting
-        files = {"file": ("jerrycoder.png", output_bytes, "image/png")}
+        files = {"file": ("output.png", output_bytes, "image/png")}
         r = requests.post("https://ar-hosting.pages.dev/upload", files=files, timeout=20)
         r.raise_for_status()
         data = r.json()
 
         if "url" in data:
-            response = {
+            return {
                 "status": "success",
                 "url": data["url"],
                 "creator": "Jerrycoder",
                 "telegram": "Oggy_workshop"
             }
-            # Pretty print JSON response
-            return json.loads(json.dumps(response, indent=4))
         return {"status": "error", "message": "Upload failed"}
     except Exception as e:
         return {"status": "error", "message": str(e)}
@@ -65,8 +72,7 @@ def process_image(img_bytes: bytes):
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
     img_bytes = await file.read()
-    result = process_image(img_bytes)
-    return JSONResponse(content=result)
+    return process_image(img_bytes)
 
 # --------------------------
 # GET: From image URL
@@ -83,10 +89,9 @@ async def from_url(url: str = Query(..., description="Image URL")):
         }
         resp = requests.get(url, headers=headers, timeout=15)
         resp.raise_for_status()
-        result = process_image(resp.content)
-        return JSONResponse(content=result)
+        return process_image(resp.content)
     except Exception as e:
-        return JSONResponse(content={"status": "error", "message": str(e)})
+        return {"status": "error", "message": str(e)}
 
 # --------------------------
 # HF Entry point