Update gradio_demo.py

gradio_demo.py

@@ -952,7 +952,7 @@ def process_image(input_image, input_text):
         # Run DINO-X detection
         task = DinoxTask(
             image_url=image_url,
-            prompts=[TextPrompt(text=input_text)]
+            prompts=[TextPrompt(text=input_text)],
             targets=[DetectionTarget.BBox, DetectionTarget.Mask]
         )
         
@@ -1079,7 +1079,7 @@ def process_image(input_image, input_text):
 
 block = gr.Blocks().queue()
 with block:
-    with gr.Tab("Text", visible=True):
+    with gr.Tab("Text", visible=False):
         with gr.Row():
             gr.Markdown("## Product Placement from Text")
         with gr.Row():
@@ -1255,11 +1255,8 @@ with block:
                         type='value'
                     )
 
-                    example_prompts = gr.Dataset(
-                        samples=quick_prompts, 
-                        label='Prompt Quick List', 
-                        components=[prompt]
-                    )
+                example_quick_subjects = gr.Dataset(samples=quick_subjects, label='Subject Quick List', samples_per_page=1000, components=[prompt])
+                example_quick_prompts = gr.Dataset(samples=quick_prompts, label='Lighting Quick List', samples_per_page=1000, components=[prompt])
                     # bg_gallery = gr.Gallery(
                     #     height=450, 
                     #     label='Background Quick List', 
@@ -1270,7 +1267,7 @@ with block:
                 relight_button_bg = gr.Button(value="Relight")
 
                 # Additional settings
-                with gr.Group():
+                with gr.Group(visible=False):
                     with gr.Row():
                         num_samples = gr.Slider(label="Images", minimum=1, maximum=12, value=1, step=1)
                         seed = gr.Number(label="Seed", value=12345, precision=0)
@@ -1280,13 +1277,13 @@ with block:
 
                 with gr.Accordion("Advanced options", open=False):
                     steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=20, step=1)
-                    cfg = gr.Slider(label="CFG Scale", minimum=1.0, maximum=32.0, value=7.0, step=0.01)
+                    cfg = gr.Slider(label="CFG Scale", minimum=1.0, maximum=32.0, value=7.0, step=0.01, visible=False)
                     highres_scale = gr.Slider(label="Highres Scale", minimum=1.0, maximum=2.0, value=1.2, step=0.01)
                     highres_denoise = gr.Slider(label="Highres Denoise", minimum=0.1, maximum=0.9, value=0.5, step=0.01)
-                    a_prompt = gr.Textbox(label="Added Prompt", value='best quality')
+                    a_prompt = gr.Textbox(label="Added Prompt", value='best quality', visible=False)
                     n_prompt = gr.Textbox(
                         label="Negative Prompt",
-                        value='lowres, bad anatomy, bad hands, cropped, worst quality'
+                        value='lowres, bad anatomy, bad hands, cropped, worst quality', visible=False
                     )
                 
             with gr.Column():
@@ -1314,6 +1311,10 @@ with block:
             outputs=[depth_image], show_progress=True
         )
 
+
+        example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
+        example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
+
         # def update_position(background, x_pos, y_pos, scale):
         #     """Update composite when position changes"""
         #     global original_bg