Update app.py

app.py

@@ -16,7 +16,7 @@ import threading
 import asyncio
 from queue import Queue as BlockingQueue
 from functools import partial
-from multiprocessing import Process
+from multiprocessing import Process, , Queue
 
 # external
 
@@ -137,10 +137,7 @@ def parallel(*pairs):
         return
     if len(pairs) == 1:
         pairs = pairs[0]
-
-    out = []
     running_tasks = []
-
     for pair in pairs:
         task = pair.pop(0)
         running_tasks.append(Process(target=task, args=(pair,)))
@@ -148,8 +145,6 @@ def parallel(*pairs):
         running_task.start()
     for running_task in running_tasks:
         running_task.join()
-        out.append(running_task.get())
-    return out
         
 # functionality
         
@@ -202,33 +197,37 @@ def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
 
-def calc(img,p1,p2,motion):
-    global last_motion
-    global pipe
-
-    if last_motion != motion:
-        if last_motion != "":
-            pipe.unload_lora_weights()
-        if motion != "":
-            pipe.load_lora_weights(motion, adapter_name="motion")
-            pipe.fuse_lora()
-            pipe.set_adapters("motion", [0.7])
-        last_motion = motion
-
-    pipe.to(device,dtype=dtype)
-
-    return pipe(
-        prompt=p1,
-        negative_prompt=p2,
-        height=height,
-        width=width,
-        ip_adapter_image=img.convert("RGB"),
-        num_inference_steps=step,
-        guidance_scale=accu,
-        num_frames=(fps*time)
-    )
-
-def handle(*inp):
+def handle_generate(*inp):
+
+    calc_out = []
+    
+    def calc(img,p1,p2,motion):
+        global last_motion
+        global pipe
+    
+        if last_motion != motion:
+            if last_motion != "":
+                pipe.unload_lora_weights()
+            if motion != "":
+                pipe.load_lora_weights(motion, adapter_name="motion")
+                pipe.fuse_lora()
+                pipe.set_adapters("motion", [0.7])
+            last_motion = motion
+    
+        pipe.to(device,dtype=dtype)
+    
+        calc_out.append(
+            pipe(
+                prompt=p1,
+                negative_prompt=p2,
+                height=height,
+                width=width,
+                ip_adapter_image=img.convert("RGB"),
+                num_inference_steps=step,
+                guidance_scale=accu,
+                num_frames=(fps*time)
+            )
+        )
 
     inp = list(inp)
     
@@ -251,9 +250,9 @@ def handle(*inp):
     ln = len(result)
 
     parargs = [[calc,*inp] for i in range(ln)]
-    out_pipe = parallel(parargs)
+    parallel(parargs)
     names = []
-    for i in out_pipe:
+    for i in calc_out:
         name = generate_random_string(12)+".png"
         export_to_gif(i.frames[0],name,fps=fps)
         names.append( name )
@@ -314,7 +313,7 @@ def ui():
                 prompt.submit,
                 prompt2.submit
             ],
-            fn=handle,
+            fn=handle_generate,
             inputs=[img,prompt,prompt2,motion],
             outputs=result
         )