Update app.py

app.py

@@ -1,3 +1,6 @@
+ew Image-to-Image Generation
+drivecheck
+Optional tone and style instructions for the model
 import spaces
 import argparse
 import random
@@ -85,6 +88,28 @@ 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(
@@ -153,9 +178,44 @@ 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
+        
+        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
+
+            
         
         imgs = gr.State()
-    
+
         with gr.Column():
             result = gr.Gallery(
                 label="Result",
@@ -166,50 +226,61 @@ 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
-            )
 
-            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
+            if len(result) > 0:
+                # 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],
+                #    outputs=input_fg
+                #)
 
-            slider.change(
+                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],
+    run_button.click(
         fn=infer,
         inputs=[
             prompt,
-            input_image,
-            do_rembg,
-            seed,
-            randomize_seed,
-            guidance_scale,
-            num_inference_steps,
-            reference_conditioning_scale,
-            negative_prompt,
+            input_fg,
         ],
-        outputs=[result, preprocessed_image, seed, imgs],
+        outputs=[result],
     )
 
 demo.launch()