Duplicate from phenomenon1981/MagicPrompt-Stable-Diffusion

Co-authored-by: Dee Jay <[email protected]>

.gitattributes

@@ -0,0 +1,31 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: MagicPrompt Stable Diffusion
+emoji: 🍄
+colorFrom: red
+colorTo: indigo
+sdk: gradio
+sdk_version: 3.3.1
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: phenomenon1981/MagicPrompt-Stable-Diffusion
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,98 @@
+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')
+
+
+def generate(starting_text):
+    with open("ideas.txt", "r") as f:
+        line = f.readlines()
+    seed = random.randint(100, 1000000)
+    set_seed(seed)
+
+    if starting_text == "":
+        starting_text: str = line[random.randrange(0, len(line))].replace("\n", "").capitalize()
+        starting_text: str = re.sub(r"\.", '', starting_text)
+
+    response = gpt2_pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 80)), num_return_sequences=1)
+    response_list = []
+    for x in response:
+        resp = x['generated_text'].strip()
+        if resp != starting_text and len(resp) > (len(starting_text) + 4) and resp.endswith((":", "-", "—")) is False:
+            response_list.append(resp)
+
+    response_end = "\n".join(response_list)
+    response_end = re.sub('[^ ]+\.[^ ]+','', response_end)
+    response_end = response_end.replace("<", "").replace(">", "")
+
+    if response_end != "":
+        return response_end
+
+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 basic idea. 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(variant="compact"):
+            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)

requirements.txt

@@ -0,0 +1,5 @@
+jinja2==3.0.3
+transformers==4.22.2
+sentencepiece
+torch
+gradio

style.css

@@ -0,0 +1,84 @@
+#col-container {
+    max-width: 800px;
+    margin-left: auto;
+    margin-right: auto;
+}
+a {
+    color: inherit;
+    text-decoration: underline;
+}
+.gradio-container {
+    font-family: 'IBM Plex Sans', sans-serif;
+}
+.gr-button {
+    color: white;
+    border-color: #9d66e5;
+    background: #9d66e5;
+}
+input[type='range'] {
+    accent-color: #9d66e5;
+}
+.dark input[type='range'] {
+    accent-color: #dfdfdf;
+}
+.container {
+    max-width: 800px;
+    margin: auto;
+    padding-top: 1.5rem;
+}
+#gallery {
+    min-height: 22rem;
+    margin-bottom: 15px;
+    margin-left: auto;
+    margin-right: auto;
+    border-bottom-right-radius: .5rem !important;
+    border-bottom-left-radius: .5rem !important;
+}
+#gallery>div>.h-full {
+    min-height: 20rem;
+}
+.details:hover {
+    text-decoration: underline;
+}
+.gr-button {
+    white-space: nowrap;
+}
+.gr-button:focus {
+    border-color: rgb(147 197 253 / var(--tw-border-opacity));
+    outline: none;
+    box-shadow: var(--tw-ring-offset-shadow), var(--tw-ring-shadow), var(--tw-shadow, 0 0 #0000);
+    --tw-border-opacity: 1;
+    --tw-ring-offset-shadow: var(--tw-ring-inset) 0 0 0 var(--tw-ring-offset-width) var(--tw-ring-offset-color);
+    --tw-ring-shadow: var(--tw-ring-inset) 0 0 0 calc(3px var(--tw-ring-offset-width)) var(--tw-ring-color);
+    --tw-ring-color: rgb(191 219 254 / var(--tw-ring-opacity));
+    --tw-ring-opacity: .5;
+}
+#advanced-options {
+    margin-bottom: 20px;
+}
+.footer {
+    margin-bottom: 45px;
+    margin-top: 35px;
+    text-align: center;
+    border-bottom: 1px solid #e5e5e5;
+}
+.footer>p {
+    font-size: .8rem;
+    display: inline-block;
+    padding: 0 10px;
+    transform: translateY(10px);
+    background: white;
+}
+.dark .logo{ filter: invert(1); }
+.dark .footer {
+    border-color: #303030;
+}
+.dark .footer>p {
+    background: #0b0f19;
+}
+.acknowledgments h4{
+    margin: 1.25em 0 .25em 0;
+    font-weight: bold;
+    font-size: 115%;
+}
+