22

app.py

@@ -121,7 +121,7 @@ css = """
 
 with open('examples/0_examples.json', 'r') as file: examples = json.load(file)
 with gr.Blocks() as demo:
-    has_lora = True
+    has_lora = gr.State(True)
     with gr.Column(elem_id="col-container"):
         with gr.Column():
             gr.HTML(read_file("static/header.html"))
@@ -140,67 +140,67 @@ with gr.Blocks() as demo:
             with gr.Column():
                 lora_path = gr.Textbox(label="Generated LoRA path",lines=2, interactive=False)
                 lora_download = gr.DownloadButton(label=f"Download LoRA", visible=has_lora)
-        
-        with gr.Row(visible=has_lora):
-            gr.Markdown("Your LoRA is ready! Now, try generating some images.")
-            with gr.Column():
-                prompt = gr.Textbox(
-                    label="Prompt",
-                    show_label=False,
-                    lines=2,
-                    placeholder="Enter your prompt",
-                    value="a man in a fishing boat. high quality, detailed"
-                    # container=False,
-                )
-                
-                imagen_button = gr.Button("Generate Image", variant="primary")
-
-                
-                with gr.Accordion("Advanced Settings", open=False):
-                    
-                    negative_prompt = gr.Textbox(
-                        label="Negative prompt",
+        with gr.Group(visible=has_lora) as my_container:
+            with gr.Row():
+                gr.Markdown("Your LoRA is ready! Now, try generating some images.")
+                with gr.Column():
+                    prompt = gr.Textbox(
+                        label="Prompt",
+                        show_label=False,
                         lines=2,
-                        container=False,
-                        placeholder="Enter your negative prompt",
-                        value="blurry, ugly, bad"
+                        placeholder="Enter your prompt",
+                        value="a man in a fishing boat. high quality, detailed"
+                        # container=False,
                     )
-                    with gr.Row():
-                        num_inference_steps = gr.Slider(
-                            label="Steps",
-                            minimum=1,
-                            maximum=30,
-                            step=1,
-                            value=9,
-                        )
-                        control_context_scale = gr.Slider(
-                            label="Context scale",
-                            minimum=0.0,
-                            maximum=1.0,
-                            step=0.01,
-                            value=0.75,
-                        )
+                    
+                    imagen_button = gr.Button("Generate Image", variant="primary")
 
-                    with gr.Row():
-                        guidance_scale = gr.Slider(
-                            label="Guidance scale",
-                            minimum=0.0,
-                            maximum=10.0,
-                            step=0.1,
-                            value=1.0,
+                    
+                    with gr.Accordion("Advanced Settings", open=False):
+                        
+                        negative_prompt = gr.Textbox(
+                            label="Negative prompt",
+                            lines=2,
+                            container=False,
+                            placeholder="Enter your negative prompt",
+                            value="blurry, ugly, bad"
                         )
+                        with gr.Row():
+                            num_inference_steps = gr.Slider(
+                                label="Steps",
+                                minimum=1,
+                                maximum=30,
+                                step=1,
+                                value=9,
+                            )
+                            control_context_scale = gr.Slider(
+                                label="Context scale",
+                                minimum=0.0,
+                                maximum=1.0,
+                                step=0.01,
+                                value=0.75,
+                            )
+
+                        with gr.Row():
+                            guidance_scale = gr.Slider(
+                                label="Guidance scale",
+                                minimum=0.0,
+                                maximum=10.0,
+                                step=0.1,
+                                value=1.0,
+                            )
+
+                        seed = gr.Slider(
+                            label="Seed",
+                            minimum=0,
+                            maximum=MAX_SEED,
+                            step=1,
+                            value=42,
+                        )
+                        randomize_seed = gr.Checkbox(label="Randomize seed", value=False)
 
-                    seed = gr.Slider(
-                        label="Seed",
-                        minimum=0,
-                        maximum=MAX_SEED,
-                        step=1,
-                        value=42,
-                    )
-                    randomize_seed = gr.Checkbox(label="Randomize seed", value=False)
-
-            with gr.Column():
-                output_image = gr.Image(label="Generated image", show_label=False)
+                with gr.Column():
+                    output_image = gr.Image(label="Generated image", show_label=False)
                     
         # gr.Examples(examples=examples, inputs=[input_image])
         gr.Markdown(read_file("static/footer.md"))