Update app.py

app.py

@@ -204,7 +204,18 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="blue", secondar
                 output_image = gr.Image(label="Result", elem_id="gallery", show_label=False)
                 used_seed = gr.Number(label="Seed Used")
 
-    # ... existing code for update_llm_visibility function ...
+    def update_llm_visibility(use_llm):
+        return {
+            llm_provider: gr.update(visible=use_llm),
+            llm_model: gr.update(visible=use_llm),
+            prompt_type: gr.update(visible=use_llm)
+        }
+
+    use_llm_generator.change(
+        update_llm_visibility,
+        inputs=[use_llm_generator],
+        outputs=[llm_provider, llm_model, prompt_type]
+    )
 
     def generate_prompt(image, text_prompt, use_enhancer, use_llm_generator, llm_provider, llm_model, prompt_type):
         if image is not None: