8 loras added

app.py

@@ -43,6 +43,36 @@ loras = [
       "title": "Cakeify",
       "example_prompt": "The video opens on a woman. A knife, held by a hand, is coming into frame and hovering over the woman. The knife then begins cutting into the woman to c4k3 cakeify it. As the knife slices the woman open, the inside of the woman is revealed to be cake with chocolate layers. The knife cuts through and the contents of the woman are revealed."
     },
+      {
+      "image": "https://huggingface.co/Remade-AI/Muscle/resolve/main/example_videos/man2_muscle.gif",
+      "id": "3c6fd399-e558-43fa-8cd3-828300aac6f8",
+      "title": "Muscle",
+      "example_prompt": "A man t2k1s takes off clothes revealing a lean muscular body and shows off muscles, looking towards the camera."
+    },
+    {
+      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/crush_example.gif",
+      "id": "d8a2912b-94e4-4227-9c45-356679af34fd",
+      "title": "Crush",
+      "example_prompt": "The video begins with a cube saying closed source. A hydraulic press positioned above slowly descends towards the cube. Upon contact, the hydraulic press c5us4 crushes it, deforming and flattening the cube, causing the cube to collapse inward until the cube is no longer recognizable."
+    },
+    {
+      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/decay_example.gif",
+      "id": "3c6fd399-e558-43fa-8cd3-828300aac6f8",
+      "title": "Decay",
+      "example_prompt": "The video shows a man. The d3c4y decay time-lapse begins, causing the man to change. The man is initially whole, but soon he appears to be rotting. The man slowly becomes increasingly shriveled and discolored, and eventually, the man decomposes and falls apart. The man is rotting in the center and appears to be covered in mold, completing the d3c4y decay time-lapse."
+    },
+    {
+      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/jesus_example.gif",
+      "id": "615fe106-fec4-44bb-b28b-2864cb322027",
+      "title": "Jesus",
+      "example_prompt": "The video begins with a smiling woman with a pink shirt looking at the camera. Then jesus appears behind her as h54g hugs jesus. Jesus embraces the woman, and they both smile in front of a park."
+    },
+     {
+      "image": "https://storage.googleapis.com/remade-v2/huggingface_assets/inflate_example.gif",
+      "id": "da2b1c34-7be8-4161-a733-e8b19a98901c",
+      "title": "Inflate",
+      "example_prompt": "The large, bald man rides a gray donkey, then infl4t3 inflates it, both the man and the donkey expanding into giant, inflated figures against the desert landscape."
+    },
   
     
     
@@ -159,8 +189,46 @@ def build_lora_prompt(subject, lora_id):
             f"and the contents of the {subject} are revealed."
         )
     
+    elif lora_id == "3c6fd399-e558-43fa-8cd3-828300aac6f8":  # Muscle
+        return (
+            f"A {subject} t2k1s takes off clothes revealing a lean muscular body and shows off muscles, "
+            f"looking towards the camera."
+        )
+    
+    elif lora_id == "d8a2912b-94e4-4227-9c45-356679af34fd":  # Crush
+        return (
+            f"The video begins with a {subject}. A hydraulic press positioned above slowly descends "
+            f"towards the {subject}. Upon contact, the hydraulic press c5us4 crushes it, deforming and "
+            f"flattening the {subject}, causing the {subject} to collapse inward until the {subject} is "
+            f"no longer recognizable."
+        )
+    
+    elif lora_id == "3c6fd399-e558-43fa-8cd3-828300aac6f8":  # Decay
+        return (
+            f"The video shows a {subject}. The d3c4y decay time-lapse begins, causing the {subject} to change. "
+            f"The {subject} is initially whole, but soon it appears to be rotting. The {subject} slowly becomes "
+            f"increasingly shriveled and discolored, and eventually, the {subject} decomposes and falls apart. "
+            f"The {subject} is rotting in the center and appears to be covered in mold, completing the d3c4y decay time-lapse."
+        )
+    
+    elif lora_id == "615fe106-fec4-44bb-b28b-2864cb322027":  # Jesus
+        return (
+            f"The video begins with a {subject} looking at the camera. Then jesus appears behind the {subject} "
+            f"as h54g hugs jesus. Jesus embraces the {subject}, and they both smile."
+        )
+    
+    elif lora_id == "da2b1c34-7be8-4161-a733-e8b19a98901c":  # Inflate
+        return (
+            f"The {subject} infl4t3 inflates, expanding into a giant, inflated figure."
+        )
+    
     else:
-        raise ValueError(f"Unknown LoRA ID: {lora_id}")
+        # Fallback to using the example prompt from the LoRA config
+        if "example_prompt" in lora_config:
+            # Replace any specific subject in the example with the user's subject
+            return lora_config["example_prompt"].replace("rodent", subject).replace("woman", subject).replace("man", subject)
+        else:
+            raise ValueError(f"Unknown LoRA ID: {lora_id} and no example prompt available")
 
 def poll_generation_status(generation_id):
     """Poll generation status from database"""
@@ -268,7 +336,39 @@ def update_selection(evt: gr.SelectData):
   sentence = f"Selected LoRA: {selected_lora['title']}"
   return selected_lora['id'], sentence
 
-async def handle_generation(input_image, subject, duration, selected_index, progress=gr.Progress(track_tqdm=True)):
+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)):
     try:
         if selected_index is None:
             raise gr.Error("You must select a LoRA before proceeding.")
@@ -432,11 +532,63 @@ css = '''
     color: #e0e0e0;
     font-weight: bold;
 }
+
+/* Example images and upload section styling */
+.upload-section {
+  display: flex;
+  gap: 20px;
+  margin: 20px 0;
+}
+
+.example-images-container {
+  flex: 1;
+}
+
+.upload-container {
+  flex: 1;
+  display: flex;
+  flex-direction: column;
+  justify-content: center;
+}
+
+.section-title {
+  font-weight: bold;
+  margin-bottom: 10px;
+  color: #7289DA;
+}
+
+.example-images-grid {
+  display: grid;
+  grid-template-columns: repeat(3, 1fr);
+  gap: 10px;
+}
+
+.example-image-item {
+  border-radius: 8px;
+  overflow: hidden;
+  cursor: pointer;
+  transition: all 0.2s ease;
+  border: 2px solid transparent;
+}
+
+.example-image-item:hover {
+  transform: scale(1.05);
+  box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
+}
+
+.example-image-item.selected {
+  border-color: #7289DA;
+}
+
+.upload-button {
+  margin-top: 15px;
+}
 '''
 
 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 ")
     
@@ -467,11 +619,35 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
             # Discord feature callout for LoRAs with better contrast
             gr.HTML(
                 """<div class="discord-feature">
-                    <span class="discord-feature-title">✨ Discord Members:</span> Access 100+ additional LoRAs beyond these 3 samples!
+                    <span class="discord-feature-title">✨ Discord Members:</span> Access 100+ additional LoRAs beyond these 8 samples!
                 </div>"""
             )
+
+            gr.HTML('<div class="section-description">Click an example image or upload your own</div>')
+            
+            # Reorganized image input section - example images and upload side by side
+            with gr.Row():
+                with gr.Column(scale=1):
+                   
+                    example_gallery = gr.Gallery(
+                        [
+                            ("https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_1c2c6e4c-8938-4464-9355-84508bcca24e.jpg", "Old man"),
+                            ("https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_1f2c6ec9-823f-46d2-982f-73c494e51877.jpg", "Young woman"),
+                            ("https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_24d949f0-8699-4714-9c82-854e1b963063.jpg", "Puppy"),
+                            ("https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_af26651e-be1a-40c0-be18-c42b3bf6d211.png", "Mini toy dancers"),
+                            ("https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_d22a894e-a074-4742-9e23-787f001a3184.jpg", "Chair"),
+                            ("https://storage.googleapis.com/remade-v2/huggingface_assets/uploads_e6472106-4e9d-4620-b41b-a9bbe4893415.png", "Cartoon boy on bike")
+                        ],
+                        columns=3,
+                        height="300px",
+                        object_fit="cover"
+                    )
+                   
+                
+                with gr.Column(scale=1):
+                
+                    input_image = gr.Image(type="filepath", label="")
             
-            input_image = gr.Image(type="filepath")
             subject = gr.Textbox(label="Describe your subject", placeholder="Cat toy")
             
             # Modified duration options - only one active option
@@ -519,8 +695,14 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
         outputs=[selected_index, selected_info]
     )
     
+    # Connect example gallery selection to subject and selected_example_url
+    example_gallery.select(
+        fn=select_example_image,
+        outputs=[selected_example_url, subject]
+    )
+    
     # Add a custom handler to check if inputs are valid
-    def check_inputs(subject, input_image, selected_index):
+    def check_inputs(subject, input_image, selected_index, selected_example_url):
         if not selected_index:
             return gr.Error("You must select a LoRA before proceeding.")
         if not subject.strip():
@@ -532,11 +714,11 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="
     # Use gr.on for the button click with validation
     button.click(
         fn=check_inputs,
-        inputs=[subject, input_image, selected_index],
+        inputs=[subject, input_image, selected_index, selected_example_url],
         outputs=None
     ).success(
         fn=handle_generation,
-        inputs=[input_image, subject, duration, selected_index],
+        inputs=[input_image, subject, duration, selected_index, selected_example_url],
         outputs=[output, current_generation_id, progress_bar]
     )