Update app.py

app.py

@@ -99,7 +99,7 @@ def load_model(model_key):
         model_load_status[model_key] = "Failed" # Update load status on error
         return f"Error loading model '{model_key}': {e}"
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=180)
 def generate_image(model, prompt, seed=-1):
     
     generator = torch.Generator(device="cuda" if torch.cuda.is_available() else "cpu")
@@ -138,7 +138,54 @@ def gradio_load_model(selected_model):
     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 Woman", "sample2.png"],
+                    [list(models.keys())[2], "Sexy girl", "sample3.png"],
+                    [list(models.keys())[1], "Future City", "sample1.png"]
+                ],
+                inputs=[model_dropdown, prompt_textbox, output_image],
+            )
+
+            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 {
@@ -184,7 +231,7 @@ with gr.Blocks(
 
 if __name__ == "__main__":
     interface.launch(debug=True)
-
+'''
 
 '''
 with gr.Blocks(