Update app.py

app.py

@@ -81,6 +81,8 @@ def generate(
     upscaler_strength: float = 0.55,
     upscale_by: float = 1.5,
     add_quality_tags: bool = True,
+    isImg2Img: bool = True,
+    
     progress=gr.Progress(track_tqdm=True),
 ):
     generator = utils.seed_everything(seed)
@@ -222,32 +224,14 @@ with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
     )
     with gr.Row():
         with gr.Column(scale=2):
-            with gr.Tab("Txt2img"):
+            with gr.Tab("Input"):
                 with gr.Group():
-                    prompt = gr.Text(
-                        label="Prompt",
-                        max_lines=5,
-                        placeholder="Enter your prompt",
-                    )
-                    negative_prompt = gr.Text(
-                        label="Negative Prompt",
-                        max_lines=5,
-                        placeholder="Enter a negative prompt",
+                    isImg2Img = gr.Checkbox(
+                        label="Img2Img", value=False
                     )
-                    with gr.Accordion(label="Quality Tags", open=True):
-                        add_quality_tags = gr.Checkbox(
-                            label="Add Quality Tags", value=True
-                        )
-                        quality_selector = gr.Dropdown(
-                            label="Quality Tags Presets",
-                            interactive=True,
-                            choices=list(quality_prompt.keys()),
-                            value="Standard v3.1",
-                        )
-            with gr.Tab("Img2Img"):
-                with gr.Group():
                     image = gr.Image(
                         label="Image Input",
+                        visible=False
                     )
                     prompt = gr.Text(
                         label="Prompt",
@@ -374,6 +358,13 @@ with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
         queue=False,
         api_name=False,
     )
+    isImg2Img.change(
+        fn=lambda x: [gr.update(visible=x)],
+        inputs=isImg2Img,
+        outputs=[image],
+        queue=False,
+        api_name=False,
+    )
     aspect_ratio_selector.change(
         fn=lambda x: gr.update(visible=x == "Custom"),
         inputs=aspect_ratio_selector,