Update app.py

app.py

@@ -183,6 +183,19 @@ Generate 768x768 multi-view images from a single image using SDXL <br>
             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
+
         imgs = gr.State()
 
         with gr.Column():
@@ -196,13 +209,30 @@ Generate 768x768 multi-view images from a single image using SDXL <br>
                 allow_preview=False,
             )
 
+            
+
             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=False)
-                result.select(use_orientation, None, selected)
-                add_button = gr.Button("Select orientation for processing")
-                add_button.click(add_orientation, [imgs, selected], input_fg)
-
+                # 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)
+                slider.change(
+                    fn=get_image_by_index,
+                    inputs=[result, slider],
+                    outputs=input_fg
+                )
+        
 
     # with gr.Row():
         # gr.Examples(