Update app.py

app.py

@@ -15,26 +15,17 @@ def remove_background(input_image):
             device="cpu"
         )
         
-        # Process the image and get result
-        result = segmentor(input_image, return_mask=True)
-        
-        # Convert result to RGBA
-        if isinstance(result, Image.Image):
-            # Create transparent background
-            output = Image.new('RGBA', result.size, (0, 0, 0, 0))
-            output.paste(input_image, mask=result)
-        else:
-            output = result['output_image']
-        
-        return output
-
+        # Process the image
+        result = segmentor(input_image)
+        return result['output_image']
     except Exception as e:
         raise gr.Error(f"Error processing image: {str(e)}")
 
-# Custom theme and styling
+# Custom theme with valid colors
 theme = gr.themes.Soft(
-    primary_hue="gold",
-    secondary_hue="orange",
+    primary_hue="orange",    # Changed from "gold" to "orange"
+    secondary_hue="blue",
+    neutral_hue="gray"
 ).set(
     body_background_fill="linear-gradient(135deg, #1a1a1a 0%, #2d2d2d 100%)",
     body_text_color="#ffffff",
@@ -43,50 +34,16 @@ theme = gr.themes.Soft(
     border_color_primary="#FFD700"
 )
 
-css = """
-.gradio-container {
-    max-width: 1200px !important;
-    margin: 0 auto !important;
-    padding: 20px !important;
-}
-.image-container {
-    border-radius: 15px !important;
-    border: 2px solid rgba(255, 215, 0, 0.3) !important;
-    padding: 10px !important;
-    background: rgba(255, 255, 255, 0.1) !important;
-    transition: transform 0.3s ease !important;
-}
-.image-container:hover {
-    transform: translateY(-5px) !important;
-}
-.gr-button {
-    min-width: 200px !important;
-    height: 45px !important;
-    font-size: 16px !important;
-    margin: 10px !important;
-    transition: all 0.3s ease !important;
-}
-.gr-button:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 5px 15px rgba(255, 215, 0, 0.3) !important;
-}
-.footer {
-    text-align: center;
-    margin-top: 20px;
-    color: #666;
-}
-"""
-
 # Create Gradio interface
-with gr.Blocks(theme=theme, css=css) as demo:
+with gr.Blocks(theme=theme) as demo:
     gr.HTML(
         """
-        <div style="text-align: center; margin-bottom: 2rem;">
-            <h1 style="font-size: 3rem; margin-bottom: 1rem; background: linear-gradient(45deg, #FFD700, #FFA500); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">
-                AI Background Remover Pro
+        <div style="text-align: center; max-width: 800px; margin: 0 auto; padding: 20px;">
+            <h1 style="font-size: 2.5rem; margin-bottom: 1rem; background: linear-gradient(45deg, #FFD700, #FFA500); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">
+                AI Background Remover
             </h1>
-            <p style="color: #cccccc; font-size: 1.2rem;">
-                Remove backgrounds instantly using advanced AI technology
+            <p style="color: #cccccc; font-size: 1.2rem; margin-bottom: 2rem;">
+                Remove backgrounds instantly using RMBG V1.4 model
             </p>
         </div>
         """
@@ -95,23 +52,21 @@ with gr.Blocks(theme=theme, css=css) as demo:
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(
-                label="Upload Your Image",
+                label="Upload Image",
                 type="pil",
-                elem_classes="image-container"
+                sources=["upload", "clipboard"]
             )
             
-            with gr.Row():
-                clear_btn = gr.Button("Clear", variant="secondary")
-                process_btn = gr.Button("Remove Background", variant="primary")
-                download_btn = gr.Button("Download", variant="primary", visible=False)
-
         with gr.Column():
             output_image = gr.Image(
                 label="Result",
-                type="pil",
-                elem_classes="image-container"
+                type="pil"
             )
-            
+    
+    with gr.Row():
+        clear_btn = gr.Button("Clear", variant="secondary")
+        process_btn = gr.Button("Remove Background", variant="primary")
+
     # Status message
     status_msg = gr.Textbox(
         label="Status",
@@ -122,34 +77,22 @@ with gr.Blocks(theme=theme, css=css) as demo:
     # Event handlers
     def process_and_update(image):
         if image is None:
-            return None, "Please upload an image first", gr.Button.update(visible=False)
+            return None, "Please upload an image first"
         try:
             result = remove_background(image)
-            return (
-                result, 
-                "✨ Background removed successfully!", 
-                gr.Button.update(visible=True)
-            )
+            return result, "✨ Background removed successfully!"
         except Exception as e:
-            return None, f"❌ Error: {str(e)}", gr.Button.update(visible=False)
+            return None, f"❌ Error: {str(e)}"
 
     process_btn.click(
         fn=process_and_update,
         inputs=[input_image],
-        outputs=[output_image, status_msg, download_btn],
+        outputs=[output_image, status_msg],
     )
     
     clear_btn.click(
-        fn=lambda: (None, None, "Ready to process your image...", gr.Button.update(visible=False)),
-        outputs=[input_image, output_image, status_msg, download_btn],
-    )
-
-    gr.HTML(
-        """
-        <div class="footer">
-            <p>Powered by BRIA AI's RMBG V1.4 Model</p>
-        </div>
-        """
+        fn=lambda: (None, None, "Ready to process your image..."),
+        outputs=[input_image, output_image, status_msg],
     )
 
 # Launch the app