Update app.py

app.py

@@ -418,7 +418,7 @@ def generate_30(
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
-        captions = caption+caption_2
+        captions = caption #+caption_2
         captions = flatten_and_stringify(captions)
         captions = " ".join(captions)
         print(captions)
@@ -568,7 +568,7 @@ def generate_60(
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
-        captions = caption+caption_2
+        captions = caption #+caption_2
         captions = flatten_and_stringify(captions)
         captions = " ".join(captions)
         print(captions)
@@ -672,9 +672,12 @@ def generate_90(
         caption=[]
         caption_2=[]
         #caption.append(captioner(sd_image_a))
-        caption.append(captioner2(sd_image_a))
+        cap=captioner2(sd_image_a)
+        caption.append(cap)
         #caption.append(captioner_3(sd_image_a))
         caption_2.append(captioning(sd_image_a))
+        global image_1_text
+        image_1_text=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)
@@ -718,7 +721,7 @@ def generate_90(
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
-        captions = caption+caption_2
+        captions = caption  #+caption_2
         captions = flatten_and_stringify(captions)
         captions = " ".join(captions)
         print(captions)
@@ -844,6 +847,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
     )
     with gr.Row():
         latent_file = gr.File(label="Image Prompt (Required)")
+        image_1_text = gr.Textbox(label="Image 1 Text", lines=5)  # Add this line
         file_1_strength =  gr.Slider(
             label="Img 1 %",
             minimum=0.0,