Update app.py

app.py

@@ -16,15 +16,19 @@ def generate_image(prompt, guidance_scale, num_inference_steps):
 
 # Create a Gradio interface for the text-to-image generation
 with gr.Blocks() as demo:
-    # Create a textbox for the user to input the text prompt
-    prompt_input = gr.Textbox(label="Text Prompt")
-    # Create sliders for the guidance scale and number of inference steps
-    guidance_scale_slider = gr.Slider(1, 20, value=7.5, step=0.1, label="Guidance Scale")
-    num_inference_steps_slider = gr.Slider(1, 100, value=50, step=1, label="Number of Inference Steps")
-    # Create an image output to display the generated image
-    image_output = gr.Image(label="Generated Image")
-    # Create a button to trigger the image generation
-    generate_button = gr.Button("Generate Image")
+    # Create a two-column layout
+    with gr.Row():
+        with gr.Column():
+            # Create a textbox for the user to input the text prompt
+            prompt_input = gr.Textbox(label="Text Prompt")
+            # Create sliders for the guidance scale and number of inference steps
+            guidance_scale_slider = gr.Slider(1, 20, value=7.5, step=0.1, label="Guidance Scale")
+            num_inference_steps_slider = gr.Slider(1, 100, value=50, step=1, label="Number of Inference Steps")
+            # Add a resolution input
+            #resolution_input = gr.Textbox(value="1536x1536", label="Resolution")
+        with gr.Column():
+            # Create an image output to display the generated image
+            image_output = gr.Image(label="Generated Image")
     
     # Add some documentation using HTML
     gr.HTML("""
@@ -33,8 +37,19 @@ with gr.Blocks() as demo:
     <p>Simply enter a text prompt, adjust the parameters, and click the 'Generate Image' button to see the generated image.</p>
     """)
     
+    # Create a button to trigger the image generation
+    generate_button = gr.Button("Generate Image")
+    
     # Define the event listener for the button click
     generate_button.click(fn=generate_image, inputs=[prompt_input, guidance_scale_slider, num_inference_steps_slider], outputs=image_output)
+    
+    # Add examples
+    examples = [
+        ["A fox drinking tea under a cherry blossom tree, anime style, 4K resolution"],
+        ["Night view of a futuristic city, cyberpunk style, neon lights"],
+        ["Medieval knight battling a dragon, epic scene, oil painting texture"]
+    ]
+    gr.Examples(examples, [prompt_input, guidance_scale_slider, num_inference_steps_slider])
 
 # Launch the Gradio app with share=True
 if __name__ == "__main__":