Update app.py from anycoder

app.py

@@ -21,7 +21,7 @@ print("Loading GLM-Image model... This may take a few minutes.")
 pipe = GlmImagePipeline.from_pretrained(
     "zai-org/GLM-Image",
     torch_dtype=torch.bfloat16,
-    device_map="cuda"  # Changed from "auto" to "cuda" - "auto" is not supported
+    device_map="cuda"
 )
 print("Model loaded successfully!")
 
@@ -36,9 +36,7 @@ def calculate_duration(num_inference_steps: int) -> int:
     Returns:
         Estimated duration in seconds
     """
-    # Estimate ~3.75 seconds per inference step (typical for this model on H200)
     step_duration = 3.75
-    # Base time for image loading/preprocessing
     base_time = 15
     return base_time + (num_inference_steps * step_duration)
 
@@ -99,7 +97,7 @@ def process_image(
     num_inference_steps: int,
     guidance_scale: float,
     seed: int,
-    progress: gr.Progress = None  # Gradio 6: use function parameter, not click argument
+    progress: gr.Progress = None
 ) -> tuple:
     """
     Process the image through the GLM-Image pipeline.
@@ -180,6 +178,14 @@ def update_time_estimate(num_steps: int) -> str:
     """Update the estimated processing time display."""
     return f"**Estimated time:** {estimate_display_time(num_steps)}"
 
+def create_disabled_download_button() -> gr.DownloadButton:
+    """Create a disabled DownloadButton component."""
+    return gr.DownloadButton(
+        value=None,
+        interactive=False,
+        variant="secondary"
+    )
+
 custom_theme = gr.themes.Soft(
     primary_hue="indigo",
     secondary_hue="blue",
@@ -320,6 +326,7 @@ with gr.Blocks(fill_height=True) as demo:
                 show_label=True
             )
             
+            # Initialize download button as disabled
             download_btn = gr.DownloadButton(
                 "Download Image",
                 value=None,
@@ -378,8 +385,6 @@ with gr.Blocks(fill_height=True) as demo:
         api_visibility="private"
     )
     
-    # FIXED: Removed progress=gr.Progress() from click arguments
-    # In Gradio 6, progress is handled automatically via the function parameter
     generate_btn.click(
         fn=process_image,
         inputs=[
@@ -392,40 +397,49 @@ with gr.Blocks(fill_height=True) as demo:
             seed
         ],
         outputs=[output_image, status]
-        # NOTE: progress is handled automatically by Gradio 6
-        # Just add progress: gr.Progress as a parameter in your function
     )
     
     def enable_download(img):
         if img is not None:
-            return gr.DownloadButton(value=img, interactive=True)
-        return gr.DownloadButton(interactive=False)
+            return {
+                download_btn: gr.DownloadButton(
+                    value=img,
+                    interactive=True,
+                    variant="secondary"
+                )
+            }
+        return {
+            download_btn: gr.DownloadButton(
+                value=None,
+                interactive=False,
+                variant="secondary"
+            )
+        }
     
     output_image.change(
         fn=enable_download,
         inputs=output_image,
-        outputs=download_btn,
+        outputs=[download_btn],
         api_visibility="private"
     )
     
     def clear_all():
+        """Clear all inputs and outputs."""
         return {
             input_image: None,
             prompt: "",
             output_image: None,
             status: "Ready to generate! GPU will be allocated automatically.",
-            download_btn: gr.DownloadButton(interactive=False)
+            download_btn: gr.DownloadButton(
+                value=None,
+                interactive=False,
+                variant="secondary"
+            )
         }
     
     clear_btn.click(
         fn=clear_all,
-        outputs={
-            input_image: input_image,
-            prompt: prompt,
-            output_image: output_image,
-            status: status,
-            download_btn: download_btn
-        },
+        outputs=[input_image, prompt, output_image, status, download_btn],
         api_visibility="private"
     )
 
@@ -466,10 +480,6 @@ demo.launch(
     #input-image:hover, #output-image:hover {
         border-color: var(--primary-400);
     }
-    /* Make the generate button full width */
-    button#generate-btn {
-        width: 100% !important;
-    }
     """,
     footer_links=[
         {"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"},