Update app.py

app.py

@@ -51,7 +51,7 @@ base = "black-forest-labs/FLUX.1-schnell"
 
 # variable data
 
-
+additional_image = None
 
 # precision data
 
@@ -183,6 +183,7 @@ def pipe_generate(img,p1,p2,time,title):
             max_sequence_length=seq,
             generator=torch.Generator(device).manual_seed(int(str(random.random()).split(".")[1]))
         ).images[0]
+        additional_image = True
 
     if title != "":
         draw = ImageDraw.Draw(img)
@@ -192,6 +193,8 @@ def pipe_generate(img,p1,p2,time,title):
         x = (width - textwidth) // 2
         y = (height - textheight) // 2
         draw.text((x, y), title,font=font)
+
+    additional_image = img if additional_image else None
     
     if time == 0.0:
         return img
@@ -209,6 +212,8 @@ def pipe_generate(img,p1,p2,time,title):
 
 def handle_generate(*_inp):
 
+    additional_image = None
+
     inp = list(_inp)
     
     inp[1] = translate(inp[1],"english")
@@ -242,7 +247,7 @@ def handle_generate(*_inp):
     if inp[3] == 0.0:
         return name, None
     else:
-        return None, name
+        return additional_image, name
 
 def ui():
     global result