Update app.py

app.py

@@ -9,63 +9,56 @@ text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion")
 
 
 proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-proc2=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-proc3=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-proc4=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-
 proc5=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
-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)))
+
+def add_random_noise(prompt, noise_level=0.1):
+    noise = "".join(random.choices(string.ascii_letters + string.punctuation, k=int(noise_level * 10)))
     return prompt + noise
 
-def send_it1(inputs,proc1=proc1):
-    prompt_with_noise = add_random_noise(inputs)
+def send_it1(inputs, noise_level, proc1=proc1):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
     output1=proc1(prompt_with_noise)
     return output1
 
-def send_it2(inputs,proc2=proc2):
-    prompt_with_noise = add_random_noise(inputs)
-    output2=proc2(prompt_with_noise)
+def send_it2(inputs, noise_level, proc1=proc1):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
+    output2=proc1(prompt_with_noise)
     return output2
 
-def send_it3(inputs,proc3=proc3):
-    prompt_with_noise = add_random_noise(inputs)
-    output3=proc3(prompt_with_noise)
+def send_it3(inputs, noise_level, proc1=proc1):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
+    output3=proc1(prompt_with_noise)
     return output3
 
-def send_it4(inputs,proc4=proc4):
-    prompt_with_noise = add_random_noise(inputs)
-    output4=proc4(prompt_with_noise)
+def send_it4(inputs, noise_level, proc1=proc1):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
+    output4=proc1(prompt_with_noise)
     return output4
 
-def send_it5(inputs,proc5=proc5):
-    prompt_with_noise = add_random_noise(inputs)
+def send_it5(inputs, noise_level, proc5=proc5):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
     output5=proc5(prompt_with_noise)
     return output5
 
-def send_it6(inputs,proc6=proc6):
-    prompt_with_noise = add_random_noise(inputs)
-    output6=proc6(prompt_with_noise)
+def send_it6(inputs, noise_level, proc5=proc5):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
+    output6=proc5(prompt_with_noise)
     return output6
 
-def send_it7(inputs,proc7=proc7):
-    prompt_with_noise = add_random_noise(inputs)
-    output7=proc7(prompt_with_noise)
+def send_it7(inputs, noise_level, proc5=proc5):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
+    output7=proc5(prompt_with_noise)
     return output7
 
-def send_it8(inputs,proc8=proc8):
-    prompt_with_noise = add_random_noise(inputs)
-    output8=proc8(prompt_with_noise)
+def send_it8(inputs, noise_level, proc5=proc5):
+    prompt_with_noise = add_random_noise(inputs, noise_level)
+    output8=proc5(prompt_with_noise)
     return output8
 
 def get_prompts(prompt_text):
     return text_gen(prompt_text)
 
-
 with gr.Blocks() as myface:
     with gr.Row():
 
@@ -73,7 +66,8 @@ with gr.Blocks() as myface:
         see_prompts=gr.Button("Magic Prompt")
     with gr.Row():
 
-        prompt=gr.Textbox(label="Enhanced Prompt")
+        prompt=gr.Textbox(label="Enter Prompt")
+        noise_level=gr.Slider(minimum=0.1, maximum=1, step=0.1, label="Noise Level: Adjust noise level to explore different outputs. More noise, more variation. The noise level controls the level of randomness or variations added to the prompt before it is sent to the model. A higher noise level would introduce more variations to the prompt, while a lower noise level would introduce fewer variations.")
         run=gr.Button("Generate")
 
 
@@ -90,16 +84,15 @@ with gr.Blocks() as myface:
         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])
-    run.click(send_it3, inputs=[prompt], outputs=[output3])
-    run.click(send_it4, inputs=[prompt], outputs=[output4])
-    run.click(send_it5, inputs=[prompt], outputs=[output5])
-    run.click(send_it6, inputs=[prompt], outputs=[output6])
-    run.click(send_it7, inputs=[prompt], outputs=[output7])
-    run.click(send_it8, inputs=[prompt], outputs=[output8])
+    run.click(send_it1, inputs=[prompt, noise_level], outputs=[output1])
+    run.click(send_it2, inputs=[prompt, noise_level], outputs=[output2])
+    run.click(send_it3, inputs=[prompt, noise_level], outputs=[output3])
+    run.click(send_it4, inputs=[prompt, noise_level], outputs=[output4])
+    run.click(send_it5, inputs=[prompt, noise_level], outputs=[output5])
+    run.click(send_it6, inputs=[prompt, noise_level], outputs=[output6])
+    run.click(send_it7, inputs=[prompt, noise_level], outputs=[output7])
+    run.click(send_it8, inputs=[prompt, noise_level], outputs=[output8])
     see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt])
 
-
-myface.queue(concurrency_count=2)
-myface.launch(enable_queue=True, inline=True)
+    myface.queue(concurrency_count=8)
+    myface.launch(enable_queue=True, inline=True)