Update app.py

app.py

@@ -135,14 +135,17 @@ def load_lora(sdxl_loras, random_lora_index, lora_scale = 1.0, progress=gr.Progr
 
 
 ## SEGA ##
-def shuffle_lora(sdxl_loras, selected_lora=None):
+def shuffle_lora(sdxl_loras, selected_lora=None, chosen_prompt):
     print("selected_lora in shuffle_lora", selected_lora)
     if(selected_lora is not None):
         random_lora_index = selected_lora
     else:
         random_lora_index = random.randrange(0, len(sdxl_loras), 1)
     print("random_lora_index in shuffle_lora: ", random_lora_index)
-    spooky_concept = random.choice([' spooky witch', ' spooky vampire', ' spooky werewolf', ' spooky ghost', ' spooky wizard', ' spooky pumpkin'])
+    if(chosen_prompt):
+        spooky_concept = chosen_prompt
+    else:
+        spooky_concept = random.choice([' spooky witch', ' spooky vampire', ' spooky werewolf', ' spooky ghost', ' spooky wizard', ' spooky pumpkin', ' spooky wizard', 'spooky skeleton'])
     lora_repo = sdxl_loras[random_lora_index]["repo"]
     lora_title = sdxl_loras[random_lora_index]["title"]
     lora_desc = f"""#### LoRA used to edit this image:
@@ -164,13 +167,16 @@ def block_if_removed(input_image):
     if(input_image is None):
         raise gr.Warning("Photo removed. Upload a new one!")
     
-def select_lora(selected_state: gr.SelectData, sdxl_loras):
+def select_lora(selected_state: gr.SelectData, sdxl_loras, chosen_prompt):
     random_lora_index = selected_state.index
-    spooky_concept = random.choice([' spooky witch', ' spooky vampire', ' spooky werewolf', ' spooky ghost', ' spooky wizard', ' spooky pumpkin'])
+    if(chosen_prompt):
+        spooky_concept = chosen_prompt
+    else:
+        spooky_concept = random.choice([' spooky witch', ' spooky vampire', ' spooky werewolf', ' spooky ghost', ' spooky wizard', ' spooky pumpkin', ' spooky wizard', 'spooky skeleton'])
     lora_repo = sdxl_loras[random_lora_index]["repo"]
     lora_title = sdxl_loras[random_lora_index]["title"]
-    lora_desc = f"""#### LoRA used to edit this image: 
-                    ## {lora_title}
+    lora_desc = f"""#### LoRA used to edit this image:
+                    ### {lora_title}
                     prompt: {spooky_concept}
                     by `{lora_repo.split('/')[0]}`
                     """
@@ -328,7 +334,8 @@ with gr.Blocks(css="style.css") as demo:
                 run_button = gr.Button("Regenerate with the same picture 🖼️🎲", elem_id="run_again")
 
 
-            with gr.Accordion("Tired of randomizing? Pick your LoRA", open=False, elem_id="pick", ):
+            with gr.Accordion("Tired of randomizing? Pick your prompt and LoRA", open=False, elem_id="pick", ):
+                choose_prompt = gr.Textbox(label="Spooky Prompt", value="")
                 choose_gallery = gr.Gallery(
                     value=[(item["image"], item["title"]) for item in sdxl_loras_raw],
                     allow_preview=False,
@@ -339,14 +346,14 @@ with gr.Blocks(css="style.css") as demo:
             
         choose_gallery.select(
             fn=select_lora,
-            inputs=gr_sdxl_loras,
+            inputs=[gr_sdxl_loras, choose_prompt],
             outputs=[gr_picked_lora, lora_image, lora_desc, spooky_concept],
             queue=False
         )
         
         run_button.click(
                 fn=shuffle_lora,
-                inputs=[gr_sdxl_loras, gr_picked_lora],
+                inputs=[gr_sdxl_loras, gr_picked_lora, choose_prompt],
                 outputs=[sega_edited_image, gr_lora_index, lora_image, lora_desc, loaded_lora, sega_edited_image, spooky_concept],
                 queue=False
             ).then(
@@ -380,7 +387,7 @@ with gr.Blocks(css="style.css") as demo:
             outputs = [do_inversion],
             queue = False).then(
                 fn=shuffle_lora,
-                inputs=[gr_sdxl_loras, gr_picked_lora],
+                inputs=[gr_sdxl_loras, gr_picked_lora, choose_prompt],
                 outputs=[sega_edited_image, gr_lora_index, lora_image, lora_desc, loaded_lora, sega_edited_image, spooky_concept],
                 queue=False
             ).then(