add examples

app.py

@@ -45,21 +45,46 @@ def generate_image(prompt, model, num_iterations, learning_rate, progress = gr.P
         return None, f"An error occurred: {str(e)}"
 
 # Create Gradio interface
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[
-        gr.Textbox(label="Prompt"),
-        gr.Dropdown(["sd-turbo", "sdxl-turbo", "pixart", "hyper-sd"], label="Model"),
-        gr.Slider(minimum=10, maximum=100, step=10, value=50, label="Number of Iterations"),
-        gr.Slider(minimum=0.1, maximum=10.0, step=0.1, value=5.0, label="Learning Rate"),
-    ],
-    outputs=[
-        gr.Image(type="filepath", label="Generated Image"),
-        gr.Textbox(label="Status")
-    ],
-    title="ReNO Image Generation",
-    description="Enter a prompt to generate an image using ReNO. Adjust the model and parameters as needed."
-)
+title="# ReNO Image Generation",
+description="Enter a prompt to generate an image using ReNO. Adjust the model and parameters as needed."
+
+with gr.Blocks() as demo:
+    with gr.Column():
+        gr.Markdown(title)
+        gr.Markdown(description)
+
+        with gr.Row():
+            with gr.Column():
+                prompt = gr.Textbox(label="Prompt")
+                chosen_model = gr.Dropdown(["sd-turbo", "sdxl-turbo", "pixart", "hyper-sd"], label="Model")
+                
+                with gr.Row():
+                    n_iter = gr.Slider(minimum=10, maximum=100, step=10, value=50, label="Number of Iterations")
+                    learning_rate = gr.Slider(minimum=0.1, maximum=10.0, step=0.1, value=5.0, label="Learning Rate")
+
+                submit_btn = gr.Button("Submit")
+
+                gr.Examples(
+                    examples = [
+                        "A minimalist logo design of a reindeer, fully rendered. The reindeer features distinct, complete shapes using bold and flat colors. The design emphasizes simplicity and clarity, suitable for logo use with a sharp outline and white background.",
+                        "A blue scooter is parked near a curb in front of a green vintage car",
+                        "A impressionistic oil painting: a lone figure walking on a misty beach, a weathered lighthouse on a cliff, seagulls above crashing waves",
+                        "A bird with 8 legs",
+                        "An orange chair to the right of a black airplane",
+                        "A pink elephant and a grey cow",
+                    ],
+                    inputs = [prompt]     
+                )
+            
+            with gr.Column():
+                output_image = gr.Image(type="filepath", label="Generated Image")
+                status = gr.Textbox(label="Status")
+
+    submit_btn.click(
+        fn = generate_image,
+        inputs = [prompt, chosen_model, n_iter, learning_rate],
+        outputs = [output_image, status]
+    )
 
 # Launch the app
-iface.launch()
+demo.queue().launch(show_error=True)