Update app.py

app.py

@@ -1,313 +1,123 @@
 import gradio as gr
-from datasets import load_dataset
-from PIL import Image  
-
-import re
 import os
-import requests
+import sys
+from pathlib import Path
+import random
+import string
+import time
+from queue import Queue
+from threading import Thread
+import emoji
 
-word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=False)
-word_list = word_list_dataset["train"]['text']
+text_gen=gr.Interface.load("spaces/phenomenon1981/MagicPrompt-Stable-Diffusion")
+def get_prompts(prompt_text):
+    if prompt_text:
+        return text_gen("dreamlikeart, " + prompt_text)
+    else:
+        return text_gen("")
+proc1=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
 
-is_gpu_busy = False
-def infer(prompt, negative, scale):
-    global is_gpu_busy
-    for filter in word_list:
-        if re.search(rf"\b{filter}\b", prompt):
-            raise gr.Error("Unsafe content found. Please try again with different prompts.")
-        
-    images = []
-    url = os.getenv('JAX_BACKEND_URL')
-    payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
-    images_request = requests.post(url, json = payload)
-    for image in images_request.json()["images"]:
-        image_b64 = (f"data:image/jpeg;base64,{image}")
-        images.append(image_b64)
-    
-    return images
-    
-    
-css = """
-        .gradio-container {
-            font-family: 'IBM Plex Sans', sans-serif;
-        }
-        .gr-button {
-            color: white;
-            border-color: black;
-            background: black;
-        }
-        input[type='range'] {
-            accent-color: black;
-        }
-        .dark input[type='range'] {
-            accent-color: #dfdfdf;
-        }
-        .container {
-            max-width: 730px;
-            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-btn {
-            font-size: .7rem !important;
-            line-height: 19px;
-            margin-top: 12px;
-            margin-bottom: 12px;
-            padding: 2px 8px;
-            border-radius: 14px !important;
-        }
-        #advanced-options {
-            display: none;
-            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 .footer {
-            border-color: #303030;
-        }
-        .dark .footer>p {
-            background: #0b0f19;
-        }
-        .acknowledgments h4{
-            margin: 1.25em 0 .25em 0;
-            font-weight: bold;
-            font-size: 115%;
-        }
-        .animate-spin {
-            animation: spin 1s linear infinite;
-        }
-        @keyframes spin {
-            from {
-                transform: rotate(0deg);
-            }
-            to {
-                transform: rotate(360deg);
-            }
-        }
-        #share-btn-container {
-            display: flex; padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; width: 13rem;
-            margin-top: 10px;
-            margin-left: auto;
-        }
-        #share-btn {
-            all: initial; color: #ffffff;font-weight: 600; cursor:pointer; font-family: 'IBM Plex Sans', sans-serif; margin-left: 0.5rem !important; padding-top: 0.25rem !important; padding-bottom: 0.25rem !important;right:0;
-        }
-        #share-btn * {
-            all: unset;
-        }
-        #share-btn-container div:nth-child(-n+2){
-            width: auto !important;
-            min-height: 0px !important;
-        }
-        #share-btn-container .wrap {
-            display: none !important;
-        }
-        
-        .gr-form{
-            flex: 1 1 50%; border-top-right-radius: 0; border-bottom-right-radius: 0;
-        }
-        #prompt-container{
-            gap: 0;
-        }
-        #prompt-text-input, #negative-prompt-text-input{padding: .45rem 0.625rem}
-        #component-16{border-top-width: 1px!important;margin-top: 1em}
-        .image_duplication{position: absolute; width: 100px; left: 50px}
-"""
+def restart_script_periodically():
+    while True:
+        random_time = random.randint(540, 600)
+        time.sleep(random_time)
+        os.execl(sys.executable, sys.executable, *sys.argv)
+
+
+restart_thread = Thread(target=restart_script_periodically, daemon=True)
+restart_thread.start()
+
+
+queue = Queue()
+queue_threshold = 100
+
+def add_random_noise(prompt, noise_level=0.00):
+    if noise_level == 0:
+        noise_level = 0.00
+    percentage_noise = noise_level * 5
+    num_noise_chars = int(len(prompt) * (percentage_noise/100))
+    noise_indices = random.sample(range(len(prompt)), num_noise_chars)
+    prompt_list = list(prompt)
+    noise_chars = list(string.ascii_letters + string.punctuation + ' ' + string.digits)
+    noise_chars.extend(['😍', '💩', '😂', '🤔', '😊', '🤗', '😭', '🙄', '😷', '🤯', '🤫', '🥴', '😴', '🤩', '🥳', '😔', '😩', '🤪', '😇', '🤢', '😈', '👹', '👻', '🤖', '👽', '💀', '🎃', '🎅', '🎄', '🎁', '🎂', '🎉', '🎈', '🎊', '🎮', '❤️', '💔', '💕', '💖', '💗', '🐶', '🐱', '🐭', '🐹', '🦊', '🐻', '🐨', '🐯', '🦁', '🐘', '🔥', '🌧️', '🌞', '🌈', '💥', '🌴', '🌊', '🌺', '🌻', '🌸', '🎨', '🌅', '🌌', '☁️', '⛈️', '❄️', '☀️', '🌤️', '⛅️', '🌥️', '🌦️', '🌧️', '🌩️', '🌨️', '🌫️', '☔️', '🌬️', '💨', '🌪️', '🌈'])
+    for index in noise_indices:
+        prompt_list[index] = random.choice(noise_chars)
+    return "".join(prompt_list)
 
-block = gr.Blocks(css=css)
 
-examples = [
-    [
-        'A high tech solarpunk utopia in the Amazon rainforest',
-        'low quality',
-        9
-    ],
-    [
-        'A pikachu fine dining with a view to the Eiffel Tower',
-        'low quality',
-        9
-    ],
-    [
-        'A mecha robot in a favela in expressionist style',
-        'low quality, 3d, photorealistic',
-        9
-    ],
-    [
-        'an insect robot preparing a delicious meal',
-        'low quality, illustration',
-        9
-    ],
-    [
-        "A small cabin on top of a snowy mountain in the style of Disney, artstation",
-        'low quality, ugly',
-        9
-    ],
-]
 
+def send_it1(inputs, noise_level, proc1=proc1):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
+    while queue.qsize() >= queue_threshold:
+        time.sleep(2)
+    queue.put(prompt_with_noise)
+    output1 = proc1(prompt_with_noise)
+    return output1
 
-with block:
-    gr.HTML(
-        """
-            <div style="text-align: center; margin: 0 auto;">
-              <div
-                style="
-                  display: inline-flex;
-                  align-items: center;
-                  gap: 0.8rem;
-                  font-size: 1.75rem;
-                "
-              >
-                <svg
-                  width="0.65em"
-                  height="0.65em"
-                  viewBox="0 0 115 115"
-                  fill="none"
-                  xmlns="http://www.w3.org/2000/svg"
-                >
-                  <rect width="23" height="23" fill="white"></rect>
-                  <rect y="69" width="23" height="23" fill="white"></rect>
-                  <rect x="23" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="23" y="69" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="46" width="23" height="23" fill="white"></rect>
-                  <rect x="46" y="69" width="23" height="23" fill="white"></rect>
-                  <rect x="69" width="23" height="23" fill="black"></rect>
-                  <rect x="69" y="69" width="23" height="23" fill="black"></rect>
-                  <rect x="92" width="23" height="23" fill="#D9D9D9"></rect>
-                  <rect x="92" y="69" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="115" y="46" width="23" height="23" fill="white"></rect>
-                  <rect x="115" y="115" width="23" height="23" fill="white"></rect>
-                  <rect x="115" y="69" width="23" height="23" fill="#D9D9D9"></rect>
-                  <rect x="92" y="46" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="92" y="115" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="92" y="69" width="23" height="23" fill="white"></rect>
-                  <rect x="69" y="46" width="23" height="23" fill="white"></rect>
-                  <rect x="69" y="115" width="23" height="23" fill="white"></rect>
-                  <rect x="69" y="69" width="23" height="23" fill="#D9D9D9"></rect>
-                  <rect x="46" y="46" width="23" height="23" fill="black"></rect>
-                  <rect x="46" y="115" width="23" height="23" fill="black"></rect>
-                  <rect x="46" y="69" width="23" height="23" fill="black"></rect>
-                  <rect x="23" y="46" width="23" height="23" fill="#D9D9D9"></rect>
-                  <rect x="23" y="115" width="23" height="23" fill="#AEAEAE"></rect>
-                  <rect x="23" y="69" width="23" height="23" fill="black"></rect>
-                </svg>
-                <h1 style="font-weight: 900; margin-bottom: 7px;margin-top:5px">
-                  Araç Rayiç Değeri Ölçüm Yazılımının Araç Oluşturucu Yapay Zeka Yazılımı
-                </h1>
-              </div>
-              <p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;">
-                Bu yazılım rayicdegeri.com tarafından oluşturulmuştur.
-              </p>
-            </div>
-        """
-    )
-    with gr.Group():
-        with gr.Box():
-            with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
-                with gr.Column():
-                    text = gr.Textbox(
-                        label="Enter your prompt",
-                        show_label=False,
-                        max_lines=1,
-                        placeholder="Enter your prompt",
-                        elem_id="prompt-text-input",
-                    ).style(
-                        border=(True, False, True, True),
-                        rounded=(True, False, False, True),
-                        container=False,
-                    )
-                    negative = gr.Textbox(
-                        label="Enter your negative prompt",
-                        show_label=False,
-                        max_lines=1,
-                        placeholder="Enter a negative prompt",
-                        elem_id="negative-prompt-text-input",
-                    ).style(
-                        border=(True, False, True, True),
-                        rounded=(True, False, False, True),
-                        container=False,
-                    )
-                btn = gr.Button("Generate image").style(
-                    margin=False,
-                    rounded=(False, True, True, False),
-                    full_width=False,
-                )
+def send_it2(inputs, noise_level, proc1=proc1):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
+    while queue.qsize() >= queue_threshold:
+        time.sleep(2)
+    queue.put(prompt_with_noise)
+    output2 = proc1(prompt_with_noise)
+    return output2
 
-        gallery = gr.Gallery(
-            label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[2], height="auto")
+#def send_it3(inputs, noise_level, proc1=proc1):
+    #prompt_with_noise = add_random_noise(inputs, noise_level)
+    #while queue.qsize() >= queue_threshold:
+        #time.sleep(2)
+    #queue.put(prompt_with_noise)
+    #output3 = proc1(prompt_with_noise)
+    #return output3
 
-        with gr.Accordion("Advanced settings", open=False):
-        #    gr.Markdown("Advanced settings are temporarily unavailable")
-        #    samples = gr.Slider(label="Images", minimum=1, maximum=4, value=4, step=1)
-        #    steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=45, step=1)
-             guidance_scale = gr.Slider(
-                label="Guidance Scale", minimum=0, maximum=50, value=9, step=0.1
-             )
-        #    seed = gr.Slider(
-        #        label="Seed",
-        #        minimum=0,
-        #        maximum=2147483647,
-        #        step=1,
-        #        randomize=True,
-        #    )
+#def send_it4(inputs, noise_level, proc1=proc1):
+    #prompt_with_noise = add_random_noise(inputs, noise_level)
+    #while queue.qsize() >= queue_threshold:
+        #time.sleep(2)
+    #queue.put(prompt_with_noise)
+    #output4 = proc1(prompt_with_noise)
+    #return output4
 
-        ex = gr.Examples(examples=examples, fn=infer, inputs=[text, negative, guidance_scale], outputs=[gallery, community_icon, loading_icon, share_button], cache_examples=False)
-        ex.dataset.headers = [""]
-        negative.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
-        text.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
-        btn.click(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
+
+with gr.Blocks(css='style.css') as demo:
+
+    with gr.Column(elem_id="col-container"):
+        with gr.Row(variant="compact"):
+            input_text = gr.Textbox(
+                label="Short Prompt",
+                show_label=False,
+                max_lines=2,
+                placeholder="Araç bilginizi giriniz.",
+            ).style(
+                container=False,
+            )
+            see_prompts = gr.Button("Oluştur [Çok detaylı]").style(full_width=False)
+
+        
+        with gr.Row(variant="compact"):
+            prompt = gr.Textbox(
+                label="Enter your prompt",
+                show_label=False,
+                max_lines=2,
+                placeholder="Araç bilginizi giriniz.",
+            ).style(
+                container=False,
+            )
+            run = gr.Button("Oluştur [Az detaylı]").style(full_width=False)
         
-        #advanced_button.click(
-        #    None,
-        #    [],
-        #    text,
-        #    _js="""
-        #    () => {
-        #        const options = document.querySelector("body > gradio-app").querySelector("#advanced-options");
-        #        options.style.display = ["none", ""].includes(options.style.display) ? "flex" : "none";
-        #    }""",
-        #)
-        share_button.click(
-            None,
-            [],
-            [],
-            _js=share_js,
-        )
-block.queue(concurrency_count=80, max_size=100).launch(max_threads=150)
+        with gr.Row():
+            with gr.Row():
+                noise_level = gr.Slider(minimum=0.0, maximum=3, step=0.1, label="Noise Level")
+        with gr.Row():
+            with gr.Row():
+                output1=gr.Image(label="Dreamlike Diffusion 1.0",show_label=False)
+                output2=gr.Image(label="Dreamlike Diffusion 1.0",show_label=False)
+        
+
+        see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt], queue=False)
+        run.click(send_it1, inputs=[prompt, noise_level], outputs=[output1])
+        run.click(send_it2, inputs=[prompt, noise_level], outputs=[output2])
+
+
+    demo.launch(enable_queue=True, inline=True)
+    block.queue(concurrency_count=100)