Update app.py

app.py

@@ -431,33 +431,38 @@ def generate_30(
         sd_image_a.resize((768,768), Image.LANCZOS)
         #sd_image_a.resize((height,width), Image.LANCZOS)
         caption=[]
-        caption.append(captioning(sd_image_a))
+        #caption.append(captioning(sd_image_a))
+        prompt1=captioning(sd_image_a)
         if latent_file_2 is not None:  # Check if a latent file is provided
             sd_image_b = Image.open(latent_file_2.name).convert('RGB')
             #sd_image_b.resize((height,width), Image.LANCZOS)
             sd_image_b.resize((768,768), Image.LANCZOS)
-            caption.append(captioning(sd_image_b))
+            #caption.append(captioning(sd_image_b))
+            prompt2=captioning(sd_image_b)
         else:
             sd_image_b = None
         if latent_file_3 is not None:  # Check if a latent file is provided
             sd_image_c = Image.open(latent_file_3.name).convert('RGB')
             #sd_image_c.resize((height,width), Image.LANCZOS)
             sd_image_c.resize((768,768), Image.LANCZOS)
-            caption.append(captioning(sd_image_c))
+            #caption.append(captioning(sd_image_c))
+            prompt3=captioning(sd_image_c)
         else:
             sd_image_c = None
         if latent_file_4 is not None:  # Check if a latent file is provided
             sd_image_d = Image.open(latent_file_4.name).convert('RGB')
             #sd_image_d.resize((height,width), Image.LANCZOS)
             sd_image_d.resize((768,768), Image.LANCZOS)
-            caption.append(captioning(sd_image_d))
+            #caption.append(captioning(sd_image_d))
+            prompt4=captioning(sd_image_d)
         else:
             sd_image_d = None
         if latent_file_5 is not None:  # Check if a latent file is provided
             sd_image_e = Image.open(latent_file_5.name).convert('RGB')
             #sd_image_e.resize((height,width), Image.LANCZOS)
             sd_image_e.resize((768,768), Image.LANCZOS)
-            caption.append(captioning(sd_image_e))
+            #caption.append(captioning(sd_image_e))
+            prompt5=captioning(sd_image_e)
         else:
             sd_image_e = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -482,7 +487,7 @@ def generate_30(
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")
         print("-- FINAL PROMPT --")
-        print(new_prompt)
+        print(prompt)
         print("-- FINAL PROMPT --")
         print("-- ------------ --")
         gc.collect()
@@ -498,7 +503,12 @@ def generate_30(
                 pil_image_3=sd_image_c,
                 pil_image_4=sd_image_d,
                 pil_image_5=sd_image_e,
-                prompt=new_prompt,
+                prompt=prompt,
+                prompt1=prompt1,
+                prompt2=prompt2,
+                prompt3=prompt3,
+                prompt4=prompt4,
+                prompt5=prompt5,
                 negative_prompt=negative_prompt,
                 text_scale=text_scale,
                 ip_scale=ip_scale,
@@ -557,36 +567,41 @@ def generate_60(
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
         sd_image_a = Image.open(latent_file.name).convert('RGB')
-        sd_image_a.resize((224,224), Image.LANCZOS)
+        sd_image_a.resize((768,768), Image.LANCZOS)
         #sd_image_a.resize((height,width), Image.LANCZOS)
         caption=[]
-        caption.append(captioning(sd_image_a))
+        #caption.append(captioning(sd_image_a))
+        prompt1=captioning(sd_image_a)
         if latent_file_2 is not None:  # Check if a latent file is provided
             sd_image_b = Image.open(latent_file_2.name).convert('RGB')
             #sd_image_b.resize((height,width), Image.LANCZOS)
-            sd_image_b.resize((224,224), Image.LANCZOS)
-            caption.append(captioning(sd_image_b))
+            sd_image_b.resize((768,768), Image.LANCZOS)
+            #caption.append(captioning(sd_image_b))
+            prompt2=captioning(sd_image_b)
         else:
             sd_image_b = None
         if latent_file_3 is not None:  # Check if a latent file is provided
             sd_image_c = Image.open(latent_file_3.name).convert('RGB')
             #sd_image_c.resize((height,width), Image.LANCZOS)
-            sd_image_c.resize((224,224), Image.LANCZOS)
-            caption.append(captioning(sd_image_c))
+            sd_image_c.resize((768,768), Image.LANCZOS)
+            #caption.append(captioning(sd_image_c))
+            prompt3=captioning(sd_image_c)
         else:
             sd_image_c = None
         if latent_file_4 is not None:  # Check if a latent file is provided
             sd_image_d = Image.open(latent_file_4.name).convert('RGB')
             #sd_image_d.resize((height,width), Image.LANCZOS)
-            sd_image_d.resize((224,224), Image.LANCZOS)
-            caption.append(captioning(sd_image_d))
+            sd_image_d.resize((768,768), Image.LANCZOS)
+            #caption.append(captioning(sd_image_d))
+            prompt4=captioning(sd_image_d)
         else:
             sd_image_d = None
         if latent_file_5 is not None:  # Check if a latent file is provided
             sd_image_e = Image.open(latent_file_5.name).convert('RGB')
             #sd_image_e.resize((height,width), Image.LANCZOS)
-            sd_image_e.resize((224,224), Image.LANCZOS)
-            caption.append(captioning(sd_image_e))
+            sd_image_e.resize((768,768), Image.LANCZOS)
+            #caption.append(captioning(sd_image_e))
+            prompt5=captioning(sd_image_e)
         else:
             sd_image_e = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -611,7 +626,7 @@ def generate_60(
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")
         print("-- FINAL PROMPT --")
-        print(new_prompt)
+        print(prompt)
         print("-- FINAL PROMPT --")
         print("-- ------------ --")
         gc.collect()
@@ -627,7 +642,12 @@ def generate_60(
                 pil_image_3=sd_image_c,
                 pil_image_4=sd_image_d,
                 pil_image_5=sd_image_e,
-                prompt=new_prompt,
+                prompt=prompt,
+                prompt1=prompt1,
+                prompt2=prompt2,
+                prompt3=prompt3,
+                prompt4=prompt4,
+                prompt5=prompt5,
                 negative_prompt=negative_prompt,
                 text_scale=text_scale,
                 ip_scale=ip_scale,
@@ -686,36 +706,41 @@ def generate_90(
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
         sd_image_a = Image.open(latent_file.name).convert('RGB')
-        sd_image_a.resize((224,224), Image.LANCZOS)
+        sd_image_a.resize((768,768), Image.LANCZOS)
         #sd_image_a.resize((height,width), Image.LANCZOS)
         caption=[]
-        caption.append(captioning(sd_image_a))
+        #caption.append(captioning(sd_image_a))
+        prompt1=captioning(sd_image_a)
         if latent_file_2 is not None:  # Check if a latent file is provided
             sd_image_b = Image.open(latent_file_2.name).convert('RGB')
             #sd_image_b.resize((height,width), Image.LANCZOS)
-            sd_image_b.resize((224,224), Image.LANCZOS)
-            caption.append(captioning(sd_image_b))
+            sd_image_b.resize((768,768), Image.LANCZOS)
+            #caption.append(captioning(sd_image_b))
+            prompt2=captioning(sd_image_b)
         else:
             sd_image_b = None
         if latent_file_3 is not None:  # Check if a latent file is provided
             sd_image_c = Image.open(latent_file_3.name).convert('RGB')
             #sd_image_c.resize((height,width), Image.LANCZOS)
-            sd_image_c.resize((224,224), Image.LANCZOS)
-            caption.append(captioning(sd_image_c))
+            sd_image_c.resize((768,768), Image.LANCZOS)
+            #caption.append(captioning(sd_image_c))
+            prompt3=captioning(sd_image_c)
         else:
             sd_image_c = None
         if latent_file_4 is not None:  # Check if a latent file is provided
             sd_image_d = Image.open(latent_file_4.name).convert('RGB')
             #sd_image_d.resize((height,width), Image.LANCZOS)
-            sd_image_d.resize((224,224), Image.LANCZOS)
-            caption.append(captioning(sd_image_d))
+            sd_image_d.resize((768,768), Image.LANCZOS)
+            #caption.append(captioning(sd_image_d))
+            prompt4=captioning(sd_image_d)
         else:
             sd_image_d = None
         if latent_file_5 is not None:  # Check if a latent file is provided
             sd_image_e = Image.open(latent_file_5.name).convert('RGB')
             #sd_image_e.resize((height,width), Image.LANCZOS)
-            sd_image_e.resize((224,224), Image.LANCZOS)
-            caption.append(captioning(sd_image_e))
+            sd_image_e.resize((768,768), Image.LANCZOS)
+            #caption.append(captioning(sd_image_e))
+            prompt5=captioning(sd_image_e)
         else:
             sd_image_e = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -740,7 +765,7 @@ def generate_90(
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")
         print("-- FINAL PROMPT --")
-        print(new_prompt)
+        print(prompt)
         print("-- FINAL PROMPT --")
         print("-- ------------ --")
         gc.collect()
@@ -756,7 +781,12 @@ def generate_90(
                 pil_image_3=sd_image_c,
                 pil_image_4=sd_image_d,
                 pil_image_5=sd_image_e,
-                prompt=new_prompt,
+                prompt=prompt,
+                prompt1=prompt1,
+                prompt2=prompt2,
+                prompt3=prompt3,
+                prompt4=prompt4,
+                prompt5=prompt5,
                 negative_prompt=negative_prompt,
                 text_scale=text_scale,
                 ip_scale=ip_scale,