update gradio interface

app.py

@@ -21,10 +21,6 @@ MAX_SEED = np.iinfo(np.int32).max
 IMAGE_SIZE = 1024
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 HF_TOKEN = os.environ.get("HF_TOKEN")
-trigger_word = "a_photo_of_TOK"
-# lora_path = "jiuface/boy-001"
-# lora_weights = "flux_train_replicate.safetensors"
-lora_scale = 0.9
 
 login(token=HF_TOKEN)
 
@@ -99,6 +95,8 @@ def process(
     input_image_editor: dict,
     lora_path: str,
     lora_weights: str,
+    lora_scale: float,
+    trigger_word: str,
     input_text: str,
     seed_slicer: int,
     randomize_seed_checkbox: bool,
@@ -161,35 +159,52 @@ with gr.Blocks() as demo:
                 layers=False,
                 brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"))
             
+        
+            with gr.Accordion("Prompt Settings", open=True):
+
+                input_text_component = gr.Textbox(
+                    label="Inpaint prompt",
+                    show_label=True,
+                    max_lines=1,
+                    placeholder="Enter your prompt",
+                )
+                trigger_word = gr.Textbox(
+                    label="Lora trigger word",
+                    show_label=True,
+                    max_lines=1,
+                    placeholder="Enter your lora trigger word here",
+                    value="a photo of TOK"
+                    
+                )
+
+                submit_button_component = gr.Button(
+                    value='Submit', variant='primary', scale=0)
+
             with gr.Accordion("Lora Settings", open=True):
-                lora_path = gr.Text(
+                lora_path = gr.Textbox(
                     label="Lora path",
                     show_label=True,
                     max_lines=1,
-                    placeholder="Enter your lora path",
-                    container=False,
+                    placeholder="Enter your lora's model path",
                 )
-                lora_weights = gr.Text(
+                lora_weights = gr.Textbox(
                     label="Lora weights",
                     show_label=True,
                     max_lines=1,
-                    placeholder="Enter your lora weights",
-                    container=False,
+                    placeholder="Enter your lora weights name",
                 )
-
-            with gr.Row():
-                input_text_component = gr.Text(
-                    label="Prompt",
+                lora_scale = gr.Slider(
+                    label="Lora scale",
                     show_label=True,
-                    max_lines=1,
-                    placeholder="Enter your prompt",
-                    container=False,
+                    minimum=0,
+                    maximum=1,
+                    step=0.1,
+                    value=0.8,
                 )
                 
-                submit_button_component = gr.Button(
-                    value='Submit', variant='primary', scale=0)
-
             with gr.Accordion("Advanced Settings", open=True):
+                
+                
                 seed_slicer_component = gr.Slider(
                     label="Seed",
                     minimum=0,
@@ -235,6 +250,8 @@ with gr.Blocks() as demo:
             input_image_editor_component,
             lora_path,
             lora_weights,
+            lora_scale,
+            trigger_word,
             input_text_component,
             seed_slicer_component,
             randomize_seed_checkbox_component,