Update app.py

app.py

@@ -5,7 +5,6 @@ from pathlib import Path
 import random
 import string
 
-
 text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion")
 
 
@@ -19,71 +18,75 @@ proc6=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
 proc7=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
 proc8=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
 
+def add_random_noise(prompt):
+    noise = "".join(random.choices(string.ascii_letters + string.punctuation, k=random.randint(1, 10)))
+    return prompt + noise
 
 def send_it1(inputs,proc1=proc1):
-        output1=proc1(inputs)
-        return(output1)
+    prompt_with_noise = add_random_noise(inputs)
+    output1=proc1(prompt_with_noise)
+    return output1
+
 def send_it2(inputs,proc2=proc2):
-        output2=proc2(inputs)
-        return(output2)
+    prompt_with_noise = add_random_noise(inputs)
+    output2=proc2(prompt_with_noise)
+    return output2
+
 def send_it3(inputs,proc3=proc3):
-        output3=proc3(inputs)
-        return(output3)
+    prompt_with_noise = add_random_noise(inputs)
+    output3=proc3(prompt_with_noise)
+    return output3
+
 def send_it4(inputs,proc4=proc4):
-        output4=proc4(inputs)
-        return(output4)
+    prompt_with_noise = add_random_noise(inputs)
+    output4=proc4(prompt_with_noise)
+    return output4
 
 def send_it5(inputs,proc5=proc5):
-        output5=proc5(inputs)
-        return(output5)
+    prompt_with_noise = add_random_noise(inputs)
+    output5=proc5(prompt_with_noise)
+    return output5
+
 def send_it6(inputs,proc6=proc6):
-        output6=proc6(inputs)
-        return(output6)
+    prompt_with_noise = add_random_noise(inputs)
+    output6=proc6(prompt_with_noise)
+    return output6
+
 def send_it7(inputs,proc7=proc7):
-        output7=proc7(inputs)
-        return(output7)
-def send_it8(inputs,proc8=proc8):
-        output8=proc8(inputs)
-        return(output8)
+    prompt_with_noise = add_random_noise(inputs)
+    output7=proc7(prompt_with_noise)
+    return output7
 
+def send_it8(inputs,proc8=proc8):
+    prompt_with_noise = add_random_noise(inputs)
+    output8=proc8(prompt_with_noise)
+    return output8
 
 def get_prompts(prompt_text):
     return text_gen(prompt_text)
 
 
-
 with gr.Blocks() as myface:
     with gr.Row():
-            
-            
-            input_text=gr.Textbox(label="Short Prompt")
-            see_prompts=gr.Button("Magic Prompt")
-    with gr.Row():
-        
-            prompt=gr.Textbox(label="Prettified Prompt")
-            run=gr.Button("Launch")
-        
-    
-    with gr.Row():
-            output1=gr.Image(label="DreamlikeArt PhotoReal 1")
-            output2=gr.Image(label="DreamlikeArt PhotoReal 2")
-            output3=gr.Image(label="DreamlikeArt PhotoReal 3")
-            output4=gr.Image(label="DreamlikeArt PhotoReal 4")
-    with gr.Row():
-            output5=gr.Image(label="DreamlikeArt Diffusion 1")
-            output6=gr.Image(label="DreamlikeArt Diffusion 2")
-            output7=gr.Image(label="DreamlikeArt Diffusion 3")
-            output8=gr.Image(label="DreamlikeArt Diffusion 4")        
-
 
+        input_text=gr.Textbox(label="Short Prompt")
+        see_prompts=gr.Button("Magic Prompt")
+    with gr.Row():
 
+        prompt=gr.Textbox(label="Prettified Prompt")
+        run=gr.Button("Launch")
 
 
-    see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt])
-        
-def add_noise(input_text):
-    noise = ''.join(random.choice(string.ascii_letters) for i in range(5))
-    return input_text + noise
+    with gr.Row():
+        output1=gr.Image(label="DreamlikeArt PhotoReal 1")
+        output2=gr.Image(label="DreamlikeArt PhotoReal 2")
+        output3=gr.Image(label="DreamlikeArt PhotoReal 3")
+        output4=gr.Image(label="DreamlikeArt PhotoReal 4")
+    with gr.Row():
+        output5=gr.Image(label="DreamlikeArt Diffusion 1")
+        output6=gr.Image(label="DreamlikeArt Diffusion 2")
+        output7=gr.Image(label="DreamlikeArt Diffusion 3")
+        output8=gr.Image(label="DreamlikeArt Diffusion 4")        
 
     run.click(send_it1, inputs=[prompt], outputs=[output1])
     run.click(send_it2, inputs=[prompt], outputs=[output2])
@@ -93,7 +96,5 @@ def add_noise(input_text):
     run.click(send_it6, inputs=[prompt], outputs=[output6])
     run.click(send_it7, inputs=[prompt], outputs=[output7])
     run.click(send_it8, inputs=[prompt], outputs=[output8])
+    see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt])
 
-
-myface.queue(concurrency_count=8)
-myface.launch(enable_queue=True, inline=True)