Update app.py

app.py

@@ -30,10 +30,9 @@ def get_default_negative_prompt(existing_json: dict) -> str:
 def generate_json_prompt(
     prompt,
     prompt_inspire_image,
-    use_inspire,
 ):
     with torch.inference_mode():
-        if use_inspire and prompt_inspire_image is not None:
+        if prompt_inspire_image is not None:
             output = vlm_pipe(image=prompt_inspire_image, prompt="")
         else:
             output = vlm_pipe(prompt=prompt)
@@ -46,7 +45,6 @@ def generate_json_prompt(
 def generate_image(
     prompt,
     prompt_inspire_image,
-    use_inspire,
     prompt_in_json,
     negative_prompt="",
     seed=42,
@@ -62,7 +60,7 @@ def generate_image(
     with torch.inference_mode():
         # If JSON prompt is empty or None, generate it first
         if not prompt_in_json or prompt_in_json == "":
-            if use_inspire and prompt_inspire_image is not None:
+            if prompt_inspire_image is not None:
                 output = vlm_pipe(image=prompt_inspire_image, prompt="")
             else:
                 output = vlm_pipe(prompt=prompt)
@@ -160,12 +158,11 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
     with gr.Row(elem_id="col-container"):
         with gr.Column(scale=1):
             
-            with gr.Accordion("Inspire from Image", open=False) as inspire_accordion:
+            with gr.Accordion("Inspire from Image Reference", open=False) as inspire_accordion:
                 prompt_inspire_image = gr.Image(
                     label="Inspiration Image",
                     type="pil",
                 )
-                use_inspire = gr.Checkbox(label="Use inspiration image", value=False)
             
             prompt_generate = gr.Textbox(
                 label="Prompt",
@@ -180,10 +177,10 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
             
             with gr.Row():
                 generate_json_btn = gr.Button("Generate JSON Prompt", variant="secondary")
-                generate_image_btn = gr.Button("Generate Image", variant="primary")
+                generate_image_btn = gr.Button("Generate New Image", variant="primary")
             
             # Refine button - initially hidden
-            refine_btn = gr.Button("Refine existing image", variant="primary", visible=False)
+            refine_btn = gr.Button("Refine Existing Image", variant="primary", visible=False)
 
             with gr.Accordion("Advanced Settings", open=False):
                 with gr.Row():
@@ -209,7 +206,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
         inputs=[
             prompt_generate,
             prompt_inspire_image,
-            use_inspire,
         ],
         outputs=[prompt_in_json],
     )
@@ -220,7 +216,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
         inputs=[
             prompt_generate,
             prompt_inspire_image,
-            use_inspire,
             prompt_in_json,
             negative_prompt,
             seed,