Update app.py

app.py

@@ -1,45 +1,44 @@
-import gradio as gr
-import requests
-import io
-from PIL import Image
-
-def remove_background(image):
-    try:
-        # Convert image to bytes
-        img_bytes = io.BytesIO()
-        image.save(img_bytes, format="PNG")
-        img_bytes = img_bytes.getvalue()
-        
-        # Call your local API (will change to HF URL later)
-        response = requests.post(
-            "http://localhost:8001/api/v1/background-removal/remove-background",
-            files={"file": ("image.png", img_bytes, "image/png")}
-        )
-        
-        if response.status_code == 200:
-            # Get the response and return image
-            result_data = response.json()
-            download_url = result_data["download_url"]
-            
-            # Download the processed image
-            result_response = requests.get(f"http://localhost:8001{download_url}")
-            result_bytes = io.BytesIO(result_response.content)
-            return Image.open(result_bytes)
-        else:
-            raise Exception(f"API call failed: {response.status_code}")
-            
-    except Exception as e:
-        raise gr.Error(f"Background removal failed: {str(e)}")
-
-# Create Gradio interface
-demo = gr.Interface(
-    fn=remove_background,
-    inputs=gr.Image(type="pil", label="📷 Upload Image"),
-    outputs=gr.Image(type="pil", label="🎨 Background Removed"),
-    title="🎨 Professional Background Remover",
-    description="Upload any image (JPG, PNG, etc) to remove background automatically with AI",
-    examples=[["example_image.jpg"]]  # Add example images later
-)
-
-if __name__ == "__main__":
+import gradio as gr
+import requests
+import io
+from PIL import Image
+
+def remove_background(image):
+    try:
+        # Convert image to bytes
+        img_bytes = io.BytesIO()
+        image.save(img_bytes, format="PNG")
+        img_bytes = img_bytes.getvalue()
+        
+        # Call your local API (will change to HF URL later)
+        response = requests.post(
+            "http://localhost:8001/api/v1/background-removal/remove-background",
+            files={"file": ("image.png", img_bytes, "image/png")}
+        )
+        
+        if response.status_code == 200:
+            # Get the response and return image
+            result_data = response.json()
+            download_url = result_data["download_url"]
+            
+            # Download the processed image
+            result_response = requests.get(f"http://localhost:8001{download_url}")
+            result_bytes = io.BytesIO(result_response.content)
+            return Image.open(result_bytes)
+        else:
+            raise Exception(f"API call failed: {response.status_code}")
+            
+    except Exception as e:
+        raise gr.Error(f"Background removal failed: {str(e)}")
+
+# Create Gradio interface
+demo = gr.Interface(
+    fn=remove_background,
+    inputs=gr.Image(type="pil", label="📷 Upload Image"),
+    outputs=gr.Image(type="pil", label="🎨 Background Removed"),
+    title="🎨 Professional Background Remover",
+    description="Upload any image (JPG, PNG, etc) to remove background automatically with AI"
+)
+
+if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)