Update app.py

app.py

@@ -1,5 +1,7 @@
 from transformers import pipeline, set_seed
 import gradio as grad, random, re
+import os
+import sys
 
 
 gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2')
@@ -29,23 +31,68 @@ def generate(starting_text):
     if response_end != "":
         return response_end
 
-
-txt = grad.Textbox(lines=1, label="Initial Text", placeholder="Enter a basic idea")
-out = grad.Textbox(lines=5, label="Generated Prompts")
-
-
-
-title = "The Stable Diffusion Prompt Generator - because your text needs a little more visual spice."
-description = 'Welcome to the MagicPrompt demo for Stable Diffusion! Ready to see some magic happen? Simply type in your text. Feeling lazy? No problem, just hit the Submit button and it will randomly pull from a list of thousands of ideas for you.<br>'
-
-grad.Interface(fn=generate,
-               inputs=txt,
-               outputs=out,
-               title=title,
-               description=description,
-               article='',
-               allow_flagging='never',
-               cache_examples=False,
-               theme="default").launch(enable_queue=False, debug=True)
-
-
+with grad.Blocks(css='style.css') as demo:
+    grad.HTML(
+        """
+            <div style="text-align: center; max-width: 650px; margin: 0 auto;">
+              <div>
+                <h1 style="font-weight: 900; font-size: 3rem; margin-bottom:20px;">
+                  The Stable Diffusion Prompt Generator - because your text needs a little more visual spice.
+                </h1>
+              </div>
+              <p style="margin-bottom: 10px; font-size: 96%">
+              Ready to see some magic happen? Simply type in your text. Feeling lazy? No problem, just hit the "Magic Prompt" button and it will randomly pull from a list of thousands of ideas for you.
+              </p>
+              <p style="margin-bottom: 10px; font-size: 98%">
+              ❤️ Press the Like Button if you enjoy my space! ❤️</a>
+              </p>
+            </div>
+        """
+    )
+    with grad.Column(elem_id="col-container"):
+        with grad.Row():
+            txt = grad.Textbox(
+                label="Initial Text",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter a basic idea",
+            ).style(
+                container=False,
+            )
+            run = grad.Button("✨ Magic Prompt ✨").style(full_width=False)
+
+
+
+        with grad.Row(variant="compact"):
+            out = grad.Textbox(
+                label="Generated Text",
+                show_label=False,
+                lines=5,
+            ).style(
+                container=False,
+            )
+
+        run.click(generate, inputs=[txt], outputs=[out])
+        
+
+
+        with grad.Row():
+                grad.HTML(
+    """
+        <div class="footer">
+        <p> Powered by <a href="https://huggingface.co/Gustavosta">Gustavosta</a> Stable Diffusion model
+</p>
+</div>
+        <div class="acknowledgments" style="font-size: 115%">
+            <p> Transform your boring ideas into creative masterpieces with just one click! Enter a spark of inspiration and let the "Magic Prompt" button work its magic.
+            </p>
+        </div>
+    """
+)
+
+
+    fn=generate,
+    run=generate,
+    inputs=txt,
+    outputs=out
+    demo.launch(enable_queue=False, inline=True)