fix

app.py

@@ -266,6 +266,53 @@ css = """
 }
 """
 
+
+
+
+    @spaces.GPU(duration=400)
+    def generate_all(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
+                     progress=gr.Progress()):
+        outputs = [None] * (len(MODEL_CONFIGS) * 2)
+        for idx, model_name in enumerate(MODEL_CONFIGS.keys()):
+            try:
+                # Display progress for the specific model
+                # progress(0, desc=f"Starting generation for {model_name}...")
+                print(f"IMAGE GENERATING {model_name} ")
+                image, used_seed = generate_image(
+                    model_name, prompt, negative_prompt, seed,
+                    randomize_seed, width, height, guidance_scale,
+                    num_inference_steps, progress
+                )
+                print(f"IMAGE GENERATIED {model_name} ")
+                # Update the respective model's tab with the generated image
+                # results[model_name].update(image)
+                # seeds[model_name].update(used_seed)
+                outputs[idx * 2] = image  # Image slot
+                outputs[idx * 2 + 1] = seed  # Seed slot
+                # outputs.extend([image, used_seed])
+                # Add intermediate results to progress * (len(all_outputs) - len(all_outputs))
+                print("YELID")
+                yield outputs + [None]
+
+
+            except Exception as e:
+                print(f"Error generating with {model_name}: {str(e)}")
+                outputs[idx * 2] = None
+                outputs[idx * 2 + 1] = None
+
+        # Update the gallery after generation
+        gallery_images = update_gallery()
+        # file_gallery.update(value=gallery_images)
+        return outputs
+
+    def update_gallery():
+        """Update the file gallery"""
+        files = get_generated_images()
+        return [
+            (f["path"], f"{f['name']}\n{f['date']}")
+            for f in files
+        ]
+
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# Multi-Model Image Generation")
@@ -352,54 +399,11 @@ with gr.Blocks(css=css) as demo:
                     )
                     refresh_button = gr.Button("Refresh Gallery",visible=True)
 
-
-
-
-    def update_gallery():
-        """Update the file gallery"""
-        files = get_generated_images()
-        return [
-            (f["path"], f"{f['name']}\n{f['date']}")
-            for f in files
-        ]
-
-
-    @spaces.GPU(duration=400)
-    def generate_all(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
-                     progress=gr.Progress()):
-        outputs = [None] * (len(MODEL_CONFIGS) * 2)
-        for idx, model_name in enumerate(MODEL_CONFIGS.keys()):
-            try:
-                # Display progress for the specific model
-                # progress(0, desc=f"Starting generation for {model_name}...")
-                print(f"IMAGE GENERATING {model_name} ")
-                image, used_seed = generate_image(
-                    model_name, prompt, negative_prompt, seed,
-                    randomize_seed, width, height, guidance_scale,
-                    num_inference_steps, progress
-                )
-                print(f"IMAGE GENERATIED {model_name} ")
-                # Update the respective model's tab with the generated image
-                # results[model_name].update(image)
-                # seeds[model_name].update(used_seed)
-                outputs[idx * 2] = image  # Image slot
-                outputs[idx * 2 + 1] = seed  # Seed slot
-                # outputs.extend([image, used_seed])
-                # Add intermediate results to progress * (len(all_outputs) - len(all_outputs))
-                print("YELID")
-                yield outputs + [None]
-
-
-            except Exception as e:
-                print(f"Error generating with {model_name}: {str(e)}")
-                outputs[idx * 2] = None
-                outputs[idx * 2 + 1] = None
-
-        # Update the gallery after generation
-        gallery_images = update_gallery()
-        # file_gallery.update(value=gallery_images)
-        return outputs
-
+                    refresh_button.click(
+        fn=update_gallery,
+        inputs=[],
+        outputs=[file_gallery],
+    )
 
     output_components = []
     for model_name in MODEL_CONFIGS.keys():
@@ -420,12 +424,6 @@ with gr.Blocks(css=css) as demo:
         outputs=output_components,
     )
 
-    refresh_button.click(
-        fn=update_gallery,
-        inputs=[],
-        outputs=[file_gallery],
-    )
-
     demo.load(
         fn=update_gallery,
         inputs=[],