working with both options

app.py

@@ -80,6 +80,7 @@ loras = [
     
     
     
+    
 ]
 
 # Initialize Supabase client with async support
@@ -256,8 +257,13 @@ async def generate_video(input_image, subject, duration, selected_index, progres
             workflow_path=str(WORKFLOW_PATH)
         )
         
-        # Upload image to GCS and get public URL
-        image_url = upload_to_gcs(input_image)
+        # Check if the input is a URL (example image) or a file path (user upload)
+        if input_image.startswith('http'):
+            # It's already a URL, use it directly
+            image_url = input_image
+        else:
+            # It's a file path, upload to GCS
+            image_url = upload_to_gcs(input_image)
         
         # Map duration selection to actual seconds
         duration_mapping = {
@@ -336,45 +342,13 @@ def update_selection(evt: gr.SelectData):
   sentence = f"Selected LoRA: {selected_lora['title']}"
   return selected_lora['id'], sentence
 
-def select_example_image(evt: gr.SelectData):
-    """Handle example image selection and return image URL and description"""
-    example_images = [
-        {
-            "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_1c2c6e4c-8938-4464-9355-84508bcca24e.jpg",
-            "description": "Old man"
-        },
-        {
-            "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_1f2c6ec9-823f-46d2-982f-73c494e51877.jpg",
-            "description": "Young woman"
-        },
-        {
-            "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_24d949f0-8699-4714-9c82-854e1b963063.jpg",
-            "description": "Puppy"
-        },
-        {
-            "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_af26651e-be1a-40c0-be18-c42b3bf6d211.png",
-            "description": "Mini toy dancers"
-        },
-        {
-            "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_d22a894e-a074-4742-9e23-787f001a3184.jpg",
-            "description": "Chair"
-        },
-        {
-            "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_e6472106-4e9d-4620-b41b-a9bbe4893415.png",
-            "description": "Cartoon boy on bike"
-        }
-    ]
-    
-    selected = example_images[evt.index]
-    return selected["url"], selected["description"]
-
-async def handle_generation(input_image, subject, duration, selected_index, selected_example_url, progress=gr.Progress(track_tqdm=True)):
+async def handle_generation(image_input, subject, duration, selected_index, progress=gr.Progress(track_tqdm=True)):
     try:
         if selected_index is None:
             raise gr.Error("You must select a LoRA before proceeding.")
             
         # Generate the video and get generation ID
-        generation_id = await generate_video(input_image, subject, duration, selected_index)
+        generation_id = await generate_video(image_input, subject, duration, selected_index)
         
         # Poll for status updates
         while True:
@@ -588,7 +562,6 @@ css = '''
 with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate", text_size="lg")) as demo:
     selected_index = gr.State(None)
     current_generation_id = gr.State(None)
-    selected_example_url = gr.State(None)
     
     gr.Markdown("# Remade AI - Wan 2.1 I2V effects LoRAs ")
     
@@ -645,8 +618,8 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
                    
                 
                 with gr.Column(scale=1):
-                
-                    input_image = gr.Image(type="filepath", label="")
+                    # Single image input component that will be used for both uploaded and example images
+                    image_input = gr.Image(type="filepath", label="")
             
             subject = gr.Textbox(label="Describe your subject", placeholder="Cat toy")
             
@@ -687,38 +660,71 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
             
             progress_bar = gr.Markdown(elem_id="progress", visible=False)
             output = gr.Video(interactive=False, label="Output video")
-            
-           
     
     gallery.select(
         update_selection,
         outputs=[selected_index, selected_info]
     )
     
-    # Connect example gallery selection to subject and selected_example_url
+    # Modified function to handle example image selection
+    def select_example_image(evt: gr.SelectData):
+        """Handle example image selection and return image URL, description, and update image source"""
+        example_images = [
+            {
+                "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_1c2c6e4c-8938-4464-9355-84508bcca24e.jpg",
+                "description": "Old man"
+            },
+            {
+                "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_1f2c6ec9-823f-46d2-982f-73c494e51877.jpg",
+                "description": "Young woman"
+            },
+            {
+                "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_24d949f0-8699-4714-9c82-854e1b963063.jpg",
+                "description": "Puppy"
+            },
+            {
+                "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_af26651e-be1a-40c0-be18-c42b3bf6d211.png",
+                "description": "Mini toy dancers"
+            },
+            {
+                "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_d22a894e-a074-4742-9e23-787f001a3184.jpg",
+                "description": "Chair"
+            },
+            {
+                "url": "https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_e6472106-4e9d-4620-b41b-a9bbe4893415.png",
+                "description": "Cartoon boy on bike"
+            }
+        ]
+        
+        selected = example_images[evt.index]
+        
+        # Return the URL, description, and update image source to "example"
+        return selected["url"], selected["description"], "example"
+    
+    # Connect example gallery selection to image_input and subject
     example_gallery.select(
         fn=select_example_image,
-        outputs=[selected_example_url, subject]
+        outputs=[image_input, subject]
     )
     
     # Add a custom handler to check if inputs are valid
-    def check_inputs(subject, input_image, selected_index, selected_example_url):
+    def check_inputs(subject, image_input, selected_index):
         if not selected_index:
             return gr.Error("You must select a LoRA before proceeding.")
         if not subject.strip():
             return gr.Error("Please describe your subject.")
-        if input_image is None:
-            return gr.Error("Please upload an image.")
+        if image_input is None:
+            return gr.Error("Please upload an image or select an example image.")
         return None
     
     # Use gr.on for the button click with validation
     button.click(
         fn=check_inputs,
-        inputs=[subject, input_image, selected_index, selected_example_url],
+        inputs=[subject, image_input, selected_index],
         outputs=None
     ).success(
         fn=handle_generation,
-        inputs=[input_image, subject, duration, selected_index, selected_example_url],
+        inputs=[image_input, subject, duration, selected_index],
         outputs=[output, current_generation_id, progress_bar]
     )