Update app.py

app.py

@@ -128,11 +128,11 @@ def upsample_prompt_logic(prompt, image_list):
         print(f"Upsampling failed: {e}")
         return prompt
 
-def update_size_from_image(image_list, width, height):
+def update_dimensions_from_image(image_list):
     """Update width/height sliders based on uploaded image aspect ratio.
     Keeps one side at 1024 and scales the other proportionally, with both sides as multiples of 8."""
     if image_list is None or len(image_list) == 0:
-        return width, height  # Default dimensions
+        return 1024, 1024  # Default dimensions
     
     # Get the first image to determine dimensions
     img = image_list[0][0]  # Gallery returns list of tuples (image, caption)
@@ -191,9 +191,6 @@ def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024,
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-
-    # Auto-update dimensions when images are uploaded
-    width, height = update_size_from_image(input_images,width,height)
     
     # Prepare image list (convert None or empty gallery to None)
     image_list = None
@@ -256,6 +253,7 @@ title = "# AI Image Editor"
 description = "AI Image Editor transforms your photos using simple text prompts only. Describe the change you want—from subtle tweaks to major edits—and watch AI instantly deliver professional, high-quality results. Unleash your creativity without complex software! Try unlimited version at [AI image editor](https://www.image2image.ai)."
 
 with gr.Blocks() as demo:
+    
     with gr.Column(elem_id="col-container"):
         gr.Markdown(title)
         gr.Markdown(description)
@@ -357,6 +355,13 @@ with gr.Blocks() as demo:
             cache_mode="lazy"
         )
 
+    # Auto-update dimensions when images are uploaded
+    input_images.upload(
+        fn=update_dimensions_from_image,
+        inputs=[input_images],
+        outputs=[width, height]
+    )
+
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,