Update app.py

app.py

@@ -238,12 +238,10 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
-def uploadNote():
-      # write note txt
-    filename= f'tst_A_{seed}.txt'
-    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
+    filename= f'tst_A_{timestamp}.txt'
     with open(filename, "w") as f:
-        f.write(f"Realvis 5.0 (Tester A): {seed} png\n")
+        f.write(f"Realvis 5.0 (Tester A) \n")
         f.write(f"Date/time: {timestamp} \n")
         f.write(f"Prompt: {prompt} \n")
         f.write(f"Steps: {num_inference_steps} \n")
@@ -298,14 +296,15 @@ def generate_30(
         options["use_resolution_binning"] = True
     images = []
     pipe.scheduler.set_timesteps(num_inference_steps,device)
-    uploadNote()
+    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+    uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
         if "negative_prompt" in batch_options:
             batch_options["negative_prompt"] = options["negative_prompt"][i:i+BATCH_SIZE]
         images.extend(pipe(**batch_options).images)
-    sd_image_path = f"rv50_A_{seed}.png"
+    sd_image_path = f"rv50_A_{timestamp}.png"
     images[0].save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)    
     image_paths = [save_image(img) for img in images]
@@ -353,7 +352,8 @@ def generate_60(
         options["use_resolution_binning"] = True
     images = []
     pipe.scheduler.set_timesteps(num_inference_steps,device)
-    uploadNote()
+    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+    uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
@@ -408,7 +408,8 @@ def generate_90(
         options["use_resolution_binning"] = True
     images = []
     pipe.scheduler.set_timesteps(num_inference_steps,device)
-    uploadNote()
+    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+    uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]