Update app.py

app.py

@@ -225,9 +225,9 @@ def save_image(img):
     return unique_name
 
 def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
-    filename= f'IP_{timestamp}.txt'
+    filename= f'IPa_{timestamp}.txt'
     with open(filename, "w") as f:
-        f.write(f"Realvis 5.0 IP Adapter \n")
+        f.write(f"Realvis 5.0 IP Adapter Test A\n")
         f.write(f"Date/time: {timestamp} \n")
         f.write(f"Prompt: {prompt} \n")
         f.write(f"Steps: {num_inference_steps} \n")
@@ -419,13 +419,14 @@ def generate_30(
         else:
             sd_image_e = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename= f'rv_IP_{timestamp}.png'
+        filename= f'rv_IPa_{timestamp}.png'
         print("-- using image file --")
+        caption.append(prompt)
         captions = caption+caption_2
         captions = flatten_and_stringify(captions)
         captions = " ".join(captions)
         print(captions)
-        print("-- generating further caption --")
+        print("-- not generating further caption --")
         global model5
         global processor5
         del captioner2
@@ -483,7 +484,7 @@ def generate_30(
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
-        downscale_path = f"rvIP_upscale_{timestamp}.png"
+        downscale_path = f"rvIPa_upscale_{timestamp}.png"
         downscale1.save(downscale_path,optimize=False,compress_level=0)
         upload_to_ftp(downscale_path) 
         image_paths = [save_image(downscale1)]     
@@ -571,8 +572,9 @@ def generate_60(
         else:
             sd_image_e = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename= f'rv_IP_{timestamp}.png'
+        filename= f'rv_IPa_{timestamp}.png'
         print("-- using image file --")
+        caption.append(prompt)
         captions = caption+caption_2
         captions = flatten_and_stringify(captions)
         captions = " ".join(captions)
@@ -634,7 +636,7 @@ def generate_60(
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
-        downscale_path = f"rvIP_upscale_{timestamp}.png"
+        downscale_path = f"rvIPa_upscale_{timestamp}.png"
         downscale1.save(downscale_path,optimize=False,compress_level=0)
         upload_to_ftp(downscale_path) 
         image_paths = [save_image(downscale1)]     
@@ -723,8 +725,9 @@ def generate_90(
         else:
             sd_image_e = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename= f'rv_IP_{timestamp}.png'
+        filename= f'rv_IPa_{timestamp}.png'
         print("-- using image file --")
+        caption.append(prompt)
         captions = caption+caption_2
         captions = flatten_and_stringify(captions)
         captions = " ".join(captions)
@@ -786,7 +789,7 @@ def generate_90(
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
-        downscale_path = f"rvIP_upscale_{timestamp}.png"
+        downscale_path = f"rvIPa_upscale_{timestamp}.png"
         downscale1.save(downscale_path,optimize=False,compress_level=0)
         upload_to_ftp(downscale_path) 
         image_paths = [save_image(downscale1)]