Update app.py

app.py

@@ -110,10 +110,22 @@ css = """
 }
 """
 
-demo = gr.Interface(generate_txt2img, 
-                    inputs=["text", "checkbox", gr.inputs.Dropdown(["None style", "Cinematic", "Digital Art", "Portrait"], label="Select Style")], 
-                    outputs="image",
-                    title="PixelCraft",
-                    css=css)
+with gr.Blocks(css=css) as demo:
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("<h1>NSFW-GEN</h1>")
+            text_prompt = gr.Textbox(label="Enter Prompt", placeholder="Example: a cute dog", lines=2)
+            generate_button = gr.Button("Generate Image", variant='primary')
 
-demo.launch()
+        with gr.Column():
+            gr.Markdown("<h4>Advanced Settings</h4>")
+            with gr.Accordion("Advanced Customizations", open=False):
+                negative_prompt = gr.Textbox(label="Negative Prompt (Optional)", placeholder="Example: blurry, unfocused", lines=2)
+                image_style = gr.Dropdown(label="Select Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], value="None style")
+
+    with gr.Row():
+        image_output = gr.Image(type="pil", label="Output Image")
+
+    generate_button.click(generate_txt2img, inputs=[current_model, text_prompt, negative_prompt, image_style], outputs=image_output)
+
+    demo.launch()