Update app.py

app.py

@@ -112,6 +112,19 @@ models = {key: load_and_prepare_model(value) for key, value in MODEL_OPTIONS.ite
 
 MAX_SEED = np.iinfo(np.int32).max
 
+def upload_to_ftp(filename):
+    try:
+        transport = paramiko.Transport((FTP_HOST, 22))
+        destination_path=FTP_DIR+filename
+        transport.connect(username = FTP_USER, password = FTP_PASS)
+        sftp = paramiko.SFTPClient.from_transport(transport)
+        sftp.put(filename, destination_path)
+        sftp.close()
+        transport.close()
+        print(f"Uploaded {filename} to FTP server")
+    except Exception as e:
+        print(f"FTP upload error: {e}")
+        
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"
     img.save(unique_name)
@@ -167,7 +180,9 @@ def generate(
         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_{seed}.png"
+    images[0].save(sd_image_path)
+    upload_to_ftp(sd_image_path)    
     image_paths = [save_image(img) for img in images]
     return image_paths, seed