Update app.py

app.py

@@ -243,8 +243,8 @@ def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
         f.write(f"Model UNET: ford442/RealVisXL_V5.0_BF16 \n")
     upload_to_ftp(filename)
 
-def captioning(img):
-    prompts_array = [
+def captioning(img, prompts_array):
+    '''prompts_array = [
         "Adjectives describing this scene are:",
       #  "The color scheme of this image is",
       #  "This scene could be described in detail as",
@@ -255,6 +255,7 @@ def captioning(img):
         "The setting of this scene must be located",
     # Add more prompts here
     ]
+    '''
     output_prompt=[]
     cap_prompt = (
             "Describe this image with a caption to be used for image generation."
@@ -442,7 +443,8 @@ def generate_30(
     latent_file_3_scale: float = 1.0,
     latent_file_4_scale: float = 1.0,
     latent_file_5_scale: float = 1.0,
-    samples=1,
+    samples: int = 1,
+    prompt_array: list[str] = None,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     prompt1=None
@@ -460,14 +462,14 @@ def generate_30(
         #sd_image_a.resize((height,width), Image.LANCZOS)
         caption=[]
         #caption.append(captioning(sd_image_a))
-        prompt1, cap=captioning(sd_image_a)
+        prompt1, cap=captioning(sd_image_a,prompt_array)
         capt.append(cap)
         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))
-            prompt2, cap=captioning(sd_image_b)
+            prompt2, cap=captioning(sd_image_b,prompt_array)
             capt.append(cap)
         else:
             sd_image_b = None
@@ -476,7 +478,7 @@ def generate_30(
             #sd_image_c.resize((height,width), Image.LANCZOS)
             sd_image_c.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_c))
-            prompt3, cap=captioning(sd_image_c)
+            prompt3, cap=captioning(sd_image_c,prompt_array)
             capt.append(cap)
         else:
             sd_image_c = None
@@ -485,7 +487,7 @@ def generate_30(
             #sd_image_d.resize((height,width), Image.LANCZOS)
             sd_image_d.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_d))
-            prompt4, cap=captioning(sd_image_d)
+            prompt4, cap=captioning(sd_image_d,prompt_array)
             capt.append(cap)
         else:
             sd_image_d = None
@@ -494,7 +496,7 @@ def generate_30(
             #sd_image_e.resize((height,width), Image.LANCZOS)
             sd_image_e.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_e))
-            prompt5, cap=captioning(sd_image_e)
+            prompt5, cap=captioning(sd_image_e,prompt_array)
             capt.append(cap)
         else:
             sd_image_e = None
@@ -593,7 +595,8 @@ def generate_60(
     latent_file_3_scale: float = 1.0,
     latent_file_4_scale: float = 1.0,
     latent_file_5_scale: float = 1.0,
-    samples=1,
+    samples: int = 1,
+    prompt_array: list[str] = None,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     prompt1=None
@@ -611,14 +614,14 @@ def generate_60(
         #sd_image_a.resize((height,width), Image.LANCZOS)
         caption=[]
         #caption.append(captioning(sd_image_a))
-        prompt1, cap=captioning(sd_image_a)
+        prompt1, cap=captioning(sd_image_a,prompt_array)
         capt.append(cap)
         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))
-            prompt2, cap=captioning(sd_image_b)
+            prompt2, cap=captioning(sd_image_b,prompt_array)
             capt.append(cap)
         else:
             sd_image_b = None
@@ -627,7 +630,7 @@ def generate_60(
             #sd_image_c.resize((height,width), Image.LANCZOS)
             sd_image_c.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_c))
-            prompt3, cap=captioning(sd_image_c)
+            prompt3, cap=captioning(sd_image_c,prompt_array)
             capt.append(cap)
         else:
             sd_image_c = None
@@ -636,7 +639,7 @@ def generate_60(
             #sd_image_d.resize((height,width), Image.LANCZOS)
             sd_image_d.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_d))
-            prompt4, cap=captioning(sd_image_d)
+            prompt4, cap=captioning(sd_image_d,prompt_array)
             capt.append(cap)
         else:
             sd_image_d = None
@@ -645,7 +648,7 @@ def generate_60(
             #sd_image_e.resize((height,width), Image.LANCZOS)
             sd_image_e.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_e))
-            prompt5, cap=captioning(sd_image_e)
+            prompt5, cap=captioning(sd_image_e,prompt_array)
             capt.append(cap)
         else:
             sd_image_e = None
@@ -744,7 +747,8 @@ def generate_90(
     latent_file_3_scale: float = 1.0,
     latent_file_4_scale: float = 1.0,
     latent_file_5_scale: float = 1.0,
-    samples=1,
+    samples: int = 1,
+    prompt_array: list[str] = None,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     prompt1=None
@@ -762,14 +766,14 @@ def generate_90(
         #sd_image_a.resize((height,width), Image.LANCZOS)
         caption=[]
         #caption.append(captioning(sd_image_a))
-        prompt1, cap=captioning(sd_image_a)
+        prompt1, cap=captioning(sd_image_a,prompt_array)
         capt.append(cap)
         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))
-            prompt2, cap=captioning(sd_image_b)
+            prompt2, cap=captioning(sd_image_b,prompt_array)
             capt.append(cap)
         else:
             sd_image_b = None
@@ -778,7 +782,7 @@ def generate_90(
             #sd_image_c.resize((height,width), Image.LANCZOS)
             sd_image_c.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_c))
-            prompt3, cap=captioning(sd_image_c)
+            prompt3, cap=captioning(sd_image_c,prompt_array)
             capt.append(cap)
         else:
             sd_image_c = None
@@ -787,7 +791,7 @@ def generate_90(
             #sd_image_d.resize((height,width), Image.LANCZOS)
             sd_image_d.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_d))
-            prompt4, cap=captioning(sd_image_d)
+            prompt4, cap=captioning(sd_image_d,prompt_array)
             capt.append(cap)
         else:
             sd_image_d = None
@@ -796,7 +800,7 @@ def generate_90(
             #sd_image_e.resize((height,width), Image.LANCZOS)
             sd_image_e.resize((768,768), Image.LANCZOS)
             #caption.append(captioning(sd_image_e))
-            prompt5, cap=captioning(sd_image_e)
+            prompt5, cap=captioning(sd_image_e,prompt_array)
             capt.append(cap)
         else:
             sd_image_e = None
@@ -990,6 +994,9 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
                     value="('deformed', 'distorted', 'disfigured':1.3),'not photorealistic':1.5, 'poorly drawn', 'bad anatomy', 'wrong anatomy', 'extra limb', 'missing limb', 'floating limbs', 'poorly drawn hands', 'poorly drawn feet', 'poorly drawn face':1.3, 'out of frame', 'extra limbs', 'bad anatomy', 'bad art', 'beginner',  'distorted face','amateur'",
                     visible=True,
                 )
+                prompts_array = []
+                for i in range(5):  # Create 5 text fields for prompts
+                    prompts_array.append(gr.Textbox(label=f"Prompt {i+1}", lines=1))
         samples = gr.Slider(
             label="Samples",
             minimum=0,
@@ -1070,6 +1077,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             file_4_strength,
             file_5_strength,
             samples,
+            prompts_array,  # Pass prompts_array to the generate function
         ],
         outputs=[result],
     )
@@ -1102,6 +1110,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             file_4_strength,
             file_5_strength,
             samples,
+            prompts_array,  # Pass prompts_array to the generate function
         ],
         outputs=[result],
     )
@@ -1134,6 +1143,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             file_4_strength,
             file_5_strength,
             samples,
+            prompts_array,  # Pass prompts_array to the generate function
         ],
         outputs=[result],
     )