Update app.py

app.py

@@ -3,43 +3,104 @@ import os
 import sys
 import random
 from pathlib import Path
+from concurrent.futures import ThreadPoolExecutor
 
-# Load all models at once
-proc1 = gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-proc2 = gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
+#Load the models and the text generation model
+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")
 
-# Load text generation model
-text_gen = gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion")
+#Create a function to handle the errors
+def handle_errors(func):
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except Exception as e:
+            return "An error occurred: " + str(e)
+    return wrapper
 
-def generate_images(prompt_text):
-    # generate prompts
-    prompts = text_gen(prompt_text)
-    
-    # run both models on all prompts
-    output1 = proc1(prompts)
-    output2 = proc2(prompts)
-    
-    return output1, output2
 
-with gr.Blocks(title="DreamlikeArt Generator") as myface:
-    with gr.Row(title="Instructions", padding=20):
-        gr.Label("Enter a short prompt below and click 'Magic Prompt' to generate prettified prompts.")
-        gr.Label("Then, select the models to run and click 'Launch' to generate images.")
+# Define the functions for each model
+@handle_errors
+def send_it1(inputs):
+    return proc1(inputs)
 
+@handle_errors
+def send_it2(inputs):
+    return proc2(inputs)
+
+@handle_errors
+def send_it3(inputs):
+    return proc3(inputs)
+
+@handle_errors
+def send_it4(inputs):
+    return proc4(inputs)
+
+@handle_errors
+def send_it5(inputs):
+    return proc5(inputs)
+
+@handle_errors
+def send_it6(inputs):
+    return proc6(inputs)
+
+@handle_errors
+def send_it7(inputs):
+    return proc7(inputs)
+
+@handle_errors
+def send_it8(inputs):
+    return proc8(inputs)
+
+
+# Create a function to generate the prompts
+@handle_errors
+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", output=True)
+        prompt = gr.Textbox(label="Prettified Prompt")
         run = gr.Button("Launch")
-        
-    with gr.Row():
-        output1 = gr.Image(label="DreamlikeArt PhotoReal", output=True)
-        output2 = gr.Image(label="DreamlikeArt Diffusion", output=True)
-        
+
     with gr.Row():
-        model_select = gr.Checkbox(["DreamlikeArt PhotoReal", "DreamlikeArt Diffusion"])
+        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")
+
+    # Handle button clicks
+    see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt])
+
+    # Use a thread pool to run the models in parallel
+    executor = ThreadPoolExecutor()
+    run.click(lambda : [executor.submit(send_it1, prompt),
+                        executor.submit(send_it2, prompt),
+                        executor.submit(send_it3, prompt),
+                        executor.submit(send_it4, prompt),
+                        executor.submit(send_it5, prompt),
+                        executor.submit(send_it6, prompt),
+                        executor.submit(send_it7, prompt),
+                        executor.submit(send_it8, prompt)
+                        ],
+                        inputs=[prompt], 
+                        outputs=[output1, output2, output3, output4, output5, output6, output7, output8])
+
 
-    see_prompts.click(text_gen, inputs=[input_text], outputs=[prompt])
-    run.click(generate_images, inputs=[prompt], outputs=[output1, output2], control_inputs=[model_select], control_outputs=[output1, output2])