Update app.py

app.py

@@ -77,6 +77,8 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
     image = Image.open(io.BytesIO(image_bytes))
     return image
 
+import gradio as gr
+
 css = """
 /* General Container Styles */
 .gradio-container {
@@ -122,21 +124,29 @@ def generate_txt2img(current_model, text_prompt, negative_prompt, image_style):
     # Your generation code here
     pass
 
-current_model = gr.inputs.Dropdown(label="Current Model", choices=list_models, default=list_models[1])
+favicon = '<img src="" width="48px" style="display: inline">'
+title = f"""<h1><center>{favicon} AI Diffusion</center></h1>"""
+markdown_title = gr.Markdown(title)
+
+current_model = gr.inputs.Dropdown(label="Current Model", choices=list_models, value=list_models[1])
 text_prompt = gr.inputs.Textbox(label="Prompt", placeholder="Enter a prompt", lines=1)
 negative_prompt = gr.inputs.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", lines=1)
-image_style = gr.inputs.Dropdown(label="Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], default="None style")
-generate_button = gr.outputs.Button(label="Generate", type="button")
+image_style = gr.inputs.Dropdown(label="Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], value="None style")
 
-image_output = gr.outputs.Image(label="Generated Image")
-
-interface = gr.Interface(
-    fn=generate_txt2img,
-    inputs=[current_model, text_prompt, negative_prompt, image_style, generate_button],
-    outputs=image_output,
-    layout="vertical",
-    css=css,
-    title="AI Diffusion"
-)
-
-interface.launch()
+generate_button = gr.outputs.Button(label="Generate", type="button")
+image_output = gr.outputs.Image(label="Output Image")
+
+with gr.Blocks(css=css) as demo:
+    gr.Grid(col_width="auto", col_gap="10px").push(
+        markdown_title,
+        current_model,
+        text_prompt,
+        negative_prompt,
+        image_style,
+        generate_button,
+        image_output
+    )
+
+    generate_button.click(generate_txt2img, inputs=[current_model, text_prompt, negative_prompt, image_style], outputs=image_output)
+
+demo.launch(show_api=False)