Update app.py from anycoder

app.py

@@ -1,9 +1,9 @@
 """
-GLM-Image to Image Editing App
-A Gradio 6 application for image-to-image editing using the GLM-Image model.
+GLM-Image to Image Editing App (ZeroGPU Version)
+A Gradio 6 application for image-to-image editing using the GLM-Image model with ZeroGPU support.
 
 This app allows users to upload an image and provide a prompt to transform
-the image using the GLM-Image diffusion model.
+the image using the GLM-Image diffusion model with dynamic GPU allocation.
 """
 
 import gradio as gr
@@ -13,6 +13,9 @@ from PIL import Image
 import os
 from datetime import datetime
 
+# Import ZeroGPU
+import spaces
+
 # Initialize the model (lazy loading for better startup performance)
 pipe = None
 
@@ -41,6 +44,34 @@ def get_image_dimensions(image: Image.Image) -> tuple:
     """Get the dimensions of an uploaded PIL image."""
     return image.size[1], image.size[0]  # height, width
 
+def estimate_duration(num_inference_steps: int, height: int, width: int) -> int:
+    """
+    Estimate the duration needed for the GPU task based on complexity.
+    
+    Args:
+        num_inference_steps: Number of diffusion steps
+        height: Image height
+        width: Image width
+    
+    Returns:
+        Estimated duration in seconds
+    """
+    # Base time per step (adjust based on testing)
+    base_time_per_step = 3.5  # seconds
+    
+    # Complexity factor based on image size (larger images take more time)
+    size_factor = (height * width) / (1024 * 1024)  # relative to 1024x1024
+    
+    # Estimate total time
+    estimated_time = num_inference_steps * base_time_per_step * size_factor
+    
+    # Add buffer for image processing overhead
+    total_duration = int(estimated_time) + 30  # +30 seconds buffer
+    
+    # Ensure minimum duration and cap at reasonable max
+    return max(60, min(total_duration, 180))  # Between 60s and 180s
+
+@spaces.GPU(duration=estimate_duration)
 def process_image(
     image: Image.Image,
     prompt: str,
@@ -53,6 +84,7 @@ def process_image(
 ) -> tuple:
     """
     Process the image through the GLM-Image pipeline.
+    Decorated with @spaces.GPU for ZeroGPU dynamic allocation.
     
     Args:
         image: Input PIL Image
@@ -152,7 +184,7 @@ custom_theme = gr.themes.Soft(
 )
 
 # Build the Gradio 6 application
-with gr.Blocks(theme=custom_theme, fill_height=True) as demo:
+with gr.Blocks(fill_height=True) as demo:
     
     # Header with branding
     gr.Markdown(
@@ -167,6 +199,16 @@ with gr.Blocks(theme=custom_theme, fill_height=True) as demo:
         elem_classes=["header-markdown"]
     )
     
+    # GPU Status indicator
+    gr.Markdown(
+        """
+        <div class="gpu-status">
+        🚀 <strong>ZeroGPU Enabled</strong> - Dynamic GPU allocation for optimal performance
+        </div>
+        """,
+        elem_classes=["gpu-status-markdown"]
+    )
+    
     # Main content in a row
     with gr.Row(equal_height=True):
         # Left column - Input controls
@@ -276,7 +318,7 @@ with gr.Blocks(theme=custom_theme, fill_height=True) as demo:
             # Status message
             status = gr.Textbox(
                 label="Status",
-                value="Ready to generate!",
+                value="Ready to generate! GPU will be allocated automatically.",
                 interactive=False,
                 show_label=True
             )
@@ -339,7 +381,7 @@ with gr.Blocks(theme=custom_theme, fill_height=True) as demo:
         api_visibility="private"
     )
     
-    # Generate button handler
+    # Generate button handler - uses ZeroGPU via @spaces.GPU decorator
     generate_btn.click(
         fn=process_image,
         inputs=[
@@ -374,7 +416,7 @@ with gr.Blocks(theme=custom_theme, fill_height=True) as demo:
             input_image: None,
             prompt: "",
             output_image: None,
-            status: "Ready to generate!",
+            status: "Ready to generate! GPU will be allocated automatically.",
             download_btn: gr.DownloadButton(interactive=False)
         }
     
@@ -413,6 +455,17 @@ demo.launch(
         color: #ffd700 !important;
         text-decoration: underline;
     }
+    .gpu-status-markdown {
+        background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%);
+        padding: 0.75rem;
+        border-radius: 0.5rem;
+        margin-bottom: 1rem;
+        text-align: center;
+        color: white;
+    }
+    .gpu-status-markdown strong {
+        color: #fff;
+    }
     #input-image, #output-image {
         border: 2px dashed var(--neutral-300);
         border-radius: var(--radius-lg);
@@ -424,6 +477,7 @@ demo.launch(
     footer_links=[
         {"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"},
         {"label": "GLM-Image Model", "url": "https://huggingface.co/zai-org/GLM-Image"},
+        {"label": "ZeroGPU", "url": "https://huggingface.co/spaces/zero-gpu-explorers/README"},
         {"label": "Diffusers Library", "url": "https://github.com/huggingface/diffusers"}
     ],
     server_name="0.0.0.0",