Update app.py

app.py

@@ -85,28 +85,6 @@ def infer(
     return images, preprocessed_image, seed
 
 
-# examples = [
-#     [
-#         "A decorative figurine of a young anime-style girl",
-#         "assets/demo/i2mv/A_decorative_figurine_of_a_young_anime-style_girl.png",
-#         True,
-#         21,
-#     ],
-#     [
-#         "A juvenile emperor penguin chick",
-#         "assets/demo/i2mv/A_juvenile_emperor_penguin_chick.png",
-#         True,
-#         0,
-#     ],
-#     [
-#         "A striped tabby cat with white fur sitting upright",
-#         "assets/demo/i2mv/A_striped_tabby_cat_with_white_fur_sitting_upright.png",
-#         True,
-#         0,
-#     ],
-# ]
-
-
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown(
@@ -175,35 +153,9 @@ Generate 768x768 multi-view images from a single image using SDXL <br>
                         placeholder="Enter your negative prompt",
                         value="watermark, ugly, deformed, noisy, blurry, low contrast",
                     )
-
-        def use_orientation(selected_image:gr.SelectData):
-            return selected_image.index
-
-        def add_orientation(imgs, index):
-            index = int(index)
-            return imgs[index]
-
-        # def get_image_by_index(imgs, index):
-        #     """
-        #     Retrieves the image from the imgs list based on the provided index.
-        #     """
-        #     try:
-        #         index = int(index)
-        #         if 0 <= index < len(imgs):
-        #             return imgs[index]
-        #         else:
-        #             return None  # Or return a placeholder image
-        #     except (ValueError, TypeError):
-        #         return None  # Or return a placeholder image
-
-
-        def set_slider_index(selected_image):
-            return selected_image.index
-        
-    
         
         imgs = gr.State()
-
+    
         with gr.Column():
             result = gr.Gallery(
                 label="Result",
@@ -214,46 +166,34 @@ Generate 768x768 multi-view images from a single image using SDXL <br>
                 height="auto",
                 allow_preview=False,
             )
-
             
+            # Slider to select image index
+            slider = gr.Slider(
+                minimum=0,
+                maximum=5,
+                step=1,
+                label="Select Image Index",
+                value=0,  # Default value
+                interactive=True
+            )
 
-            if result:
-                # Slider to select image index
-                slider = gr.Slider(
-                    minimum=0,
-                    maximum=5,
-                    step=1,
-                    label="Select Image Index",
-                    value=0,  # Default value
-                    interactive=True
-                )
-                
-                input_fg = gr.Image(type="numpy", label="Selected Image", height=480)
-                # selected = gr.Number(visible=True)
-                # result.select(use_orientation, None, selected).then()
-                # add_button = gr.Button("Select orientation for processing")
-                # add_button.click(add_orientation, [imgs, selected], input_fg)
-                result.select(
-                    fn=set_slider_index,
-                    inputs=result,
-                    outputs=slider
-                )
-                
-                slider.change(
-                    fn=use_orientation,
-                    inputs=[result, slider],
+            input_fg = gr.Image(type="numpy", label="Selected Image", height=480)
+
+            def update_selected_image(selected_index:int, images):
+                """Selects the appropriate image"""
+                try:
+                    selected_index=int(selected_index)
+                    if 0 <= selected_index < len(images):
+                        return images[selected_index]
+                except (ValueError, TypeError):
+                    return None
+
+            slider.change(
+                    fn=update_selected_image,
+                    inputs=[slider, imgs],
                     outputs=input_fg
-                )
-        
+            )
 
-    # with gr.Row():
-        # gr.Examples(
-        #     examples=examples,
-        #     fn=infer,
-        #     inputs=[prompt, input_image, do_rembg, seed],
-        #     outputs=[result, preprocessed_image, seed],
-        #     cache_examples=True,
-        # )
 
     gr.on(
         triggers=[run_button.click, prompt.submit],
@@ -269,7 +209,7 @@ Generate 768x768 multi-view images from a single image using SDXL <br>
             reference_conditioning_scale,
             negative_prompt,
         ],
-        outputs=[result, preprocessed_image, seed],
+        outputs=[result, preprocessed_image, seed, imgs],
     )
 
 demo.launch()