Update app.py from anycoder

app.py

@@ -15,6 +15,11 @@ from diffusers.pipelines.glm_image import GlmImagePipeline
 from PIL import Image
 import time
 import random
+import os
+import tempfile
+
+# Create a temp directory for saving images
+TEMP_DIR = tempfile.mkdtemp(prefix="glm_image_")
 
 # Load the GLM-Image model directly with bfloat16 precision
 print("Loading GLM-Image model... This may take a few minutes.")
@@ -122,7 +127,8 @@ def process_image(
         progress: Gradio progress tracker (handled automatically by Gradio 6)
     
     Returns:
-        Tuple of (output_image, status_message)
+        Tuple of (output_image, status_message, file_path)
+        file_path: Path to the saved image file for download (or None if failed)
     """
     try:
         if image is None:
@@ -159,15 +165,20 @@ def process_image(
         
         output_image = result.images[0]
         
+        # Save image to temp file for download button
+        timestamp = int(time.time() * 1000)
+        temp_path = os.path.join(TEMP_DIR, f"glm_output_{timestamp}.png")
+        output_image.save(temp_path, format="PNG")
+        
         if progress:
             progress(1.0, desc="Complete!")
         
         status = f"Successfully generated! ({height}x{width}, {num_inference_steps} steps)"
-        return output_image, status
+        return output_image, status, temp_path
         
     except Exception as e:
         error_msg = f"Error: {str(e)}"
-        return None, error_msg
+        return None, error_msg, None
 
 def update_dimensions_from_image(image: Image.Image) -> tuple:
     """Update height and width based on uploaded image dimensions."""
@@ -404,14 +415,24 @@ with gr.Blocks(fill_height=True) as demo:
             guidance_scale,
             seed
         ],
-        outputs=[output_image, status]
+        outputs=[output_image, status, download_btn]
     )
     
-    def enable_download(img):
-        if img is not None:
+    def enable_download(img, file_path):
+        """
+        Enable the download button with the file path.
+        
+        Args:
+            img: The PIL Image (unused, kept for compatibility)
+            file_path: Path to the saved image file
+        
+        Returns:
+            Update dict for the download button
+        """
+        if file_path is not None and os.path.exists(file_path):
             return {
                 download_btn: gr.DownloadButton(
-                    value=img,
+                    value=file_path,  # <- Use file path, not Image object
                     interactive=True,
                     variant="secondary"
                 )
@@ -424,13 +445,6 @@ with gr.Blocks(fill_height=True) as demo:
             )
         }
     
-    output_image.change(
-        fn=enable_download,
-        inputs=output_image,
-        outputs=[download_btn],
-        api_visibility="private"
-    )
-    
     def clear_all():
         """Clear all inputs and outputs."""
         return {