Update app.py

app.py

@@ -137,6 +137,56 @@ def gradio_load_model(selected_model):
         return "No model selected. Please select a model to load."
     return load_model(selected_model)
 
+
+
+with gr.Blocks(
+    css="""
+    .container {
+        max-width: 800px;
+        margin: auto;
+        padding: 20px;
+        background-color: #f8f8f8;
+        border-radius: 10px;
+        box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
+    }
+    """
+) as interface:
+    with gr.Tab("Image Generator"):
+        with gr.Column():
+            gr.Markdown("# Text-to-Image Generator")
+            model_dropdown = gr.Dropdown(choices=list(models.keys()), label="Select Model")
+            prompt_textbox = gr.Textbox(label="Enter Text Prompt")
+            seed_slider = gr.Slider(minimum=-1, maximum=1000, step=1, value=-1, label="Random Seed (-1 for random)")
+            generate_button = gr.Button("Generate Image")
+            output_image = gr.Image(label="Generated Image")
+            runtime_info_textbox = gr.Textbox(label="Runtime Information", lines=2, interactive=False)
+
+            # Add example prompts at the bottom
+            gr.Markdown("### Example Prompts")
+            examples = gr.Examples(
+                examples=[
+                    [list(models.keys())[0], "Sexy girl"],
+                    [list(models.keys())[0], "Beautiful Woman"],
+                    [list(models.keys())[0], "Future City"]
+                ],
+                inputs=[model_dropdown, prompt_textbox],
+            )
+
+            generate_button.click(gradio_generate, inputs=[model_dropdown, prompt_textbox, seed_slider], outputs=[output_image, runtime_info_textbox])
+
+    with gr.Tab("Model Information"):
+        for model_key, model_info in models.items():
+            gr.Markdown(f"## {model_key}")
+            gr.Markdown(model_info["description"])
+
+    gr.Markdown("""---
+    **Credits**: Created by Ruslan Magana Vsevolodovna. For more information, visit [https://ruslanmv.com/](https://ruslanmv.com/).""")
+
+if __name__ == "__main__":
+    interface.launch(debug=True)
+
+
+'''
 with gr.Blocks(
     css="""
     .container {
@@ -176,4 +226,5 @@ with gr.Blocks(
     **Credits**: Created by Ruslan Magana Vsevolodovna. For more information, visit [https://ruslanmv.com/](https://ruslanmv.com/).""")
 
 if __name__ == "__main__":
-    interface.launch(debug=True)
+    interface.launch(debug=True)
+'''