small UI layout improvements (#3)

- small UI layout improvements (79fe32938357895466ee185ae29c42bb44db884d)
- Update app.py (2fa27441602d6dd948787089289ae125c2daf3de)


Co-authored-by: Linoy Tsaban <[email protected]>

app.py

@@ -123,6 +123,13 @@ def create_interface():
         "Snoopy", "Van_Gogh", "Vector"
     ]
 
+    def update_trigger_word(lora_name, prompt):
+      for name in demo_lora_names:
+        trigger = " ".join(name.split("_")) + " style,"
+        prompt = prompt.replace(trigger, "")
+      new_trigger = " ".join(lora_name.split("_"))+ " style,"
+      return new_trigger + prompt
+
     # Example data
     examples = [
         ["3D_Chibi", "", "3D Chibi style, Two smiling colleagues enthusiastically high-five in front of a whiteboard filled with technical notes about multimodal learning, reflecting a moment of success and collaboration at OpenAI.", 
@@ -152,29 +159,33 @@ def create_interface():
 
         with gr.Row():
             with gr.Column(scale=1):
-                lora_dropdown = gr.Dropdown(
+               image_input = gr.Image(type="pil", label="Upload Image")
+               prompt_box = gr.Textbox(label="Prompt",
+                                        value="3D Chibi style,",
+                                        info="Remember to include the necessary trigger words if you're using a custom LoRA."
+                )
+               lora_dropdown = gr.Dropdown(
                     demo_lora_names, label="Select built-in LoRA")
-                custom_repo_box = gr.Textbox(
+               custom_repo_box = gr.Textbox(
                     label="Enter Custom LoRA",
                     placeholder="LoRA Hugging Face path (e.g., 'username/repo_name')",
                     info="If you want to use a custom LoRA, enter its Hugging Face repo ID here and built-in LoRA will be Overridden. Leave empty to use built-in LoRAs. [Check the list of FLUX LoRAs](https://huggingface.co/models?other=base_model:adapter:black-forest-labs/FLUX.1-dev)"
                 )
-                prompt_box = gr.Textbox(label="Prompt",
-                                        placeholder="Enter your prompt here",
-                                        info="Remember to include the necessary trigger words if you're using a custom LoRA."
-                )
-                image_input = gr.Image(type="pil", label="Upload Image")
+               gen_btn = gr.Button("Generate")
             with gr.Column(scale=1):
                 output_image = gr.ImageSlider(label="Generated Image")
-                height_box = gr.Textbox(value="1024", label="Height")
-                width_box  = gr.Textbox(value="1024", label="Width")
-                guidance_slider = gr.Slider(
-                    0.1, 20, value=3.5, step=0.1, label="Guidance Scale")
-                steps_slider = gr.Slider(
-                    1, 50, value=25, step=1, label="Inference Steps")
-                seed_slider = gr.Slider(
-                    1, 2_147_483_647, value=42, step=1, label="Seed")
-                gen_btn = gr.Button("Generate")
+        with gr.Accordion("Advanced Options", open=False):
+          height_box = gr.Textbox(value="1024", label="Height")
+          width_box  = gr.Textbox(value="1024", label="Width")
+          guidance_slider = gr.Slider(
+              0.1, 20, value=3.5, step=0.1, label="Guidance Scale")
+          steps_slider = gr.Slider(
+              1, 50, value=25, step=1, label="Inference Steps")
+          seed_slider = gr.Slider(
+              1, 2_147_483_647, value=42, step=1, label="Seed")
+
+        lora_dropdown.select(fn=update_trigger_word, inputs=[lora_dropdown,prompt_box], 
+                             outputs=prompt_box)
 
         gr.Examples(
             examples=examples,