Update app.py

app.py

@@ -84,6 +84,36 @@ def find_cuda_directories(search_paths=None):
 #import hidet
 #print(dir(hidet)) 
 #import torch_tensorrt
+import pythran
+
+with open("pyt.py", "w") as f:
+    f.write("""
+import paramiko
+import os
+
+FTP_HOST = os.getenv("FTP_HOST")
+FTP_USER = os.getenv("FTP_USER")
+FTP_PASS = os.getenv("FTP_PASS")
+FTP_DIR = os.getenv("FTP_DIR")
+
+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}")
+        
+    """)
+    
+os.environ['DISTUTILS_USE_SDK'] = '1'  # Force setuptools (distutils) to use the installed SDK.
+subprocess.run(['pythran', './pyt.py'])
+import pyt
 
 import random
 import uuid
@@ -435,14 +465,14 @@ def generate_30(
     rv_image = pipe(**batch_options).images[0]
     sd_image_path = f"rv_B_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
-    pyx.upload_to_ftp(sd_image_path)    
+    pyt.upload_to_ftp(sd_image_path)    
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)
     downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
     downscale_path = f"rv50_upscale_{timestamp}.png"
     downscale1.save(downscale_path,optimize=False,compress_level=0)
-    pyx.upload_to_ftp(downscale_path) 
+    pyt.upload_to_ftp(downscale_path) 
     unique_name = str(uuid.uuid4()) + ".png"  
     os.symlink(sd_image_path, unique_name)  
     return [unique_name]