Update app.py

app.py

@@ -19,6 +19,10 @@ import gradio as gr
 import numpy as np
 import random
 
+import socket
+import threading # NEW IMPORT
+import queue     # NEW IMPORT
+
 import torch
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
@@ -58,19 +62,81 @@ FTP_HOST = 'noahcohn.com'
 FTP_USER = 'ford442'
 FTP_PASS = os.getenv("FTP_PASS")
 FTP_DIR = 'img.noahcohn.com/stablediff/'
+FTP_HOST_FALLBACK = '1ink.us'
+FTP_DIR_FALLBACK = 'img.1ink.us/stablediff/'
 
+# --- WORKER FUNCTION FOR THREADING ---
+# This function contains the logic to connect to a single host.
+# It will be executed by each of our threads.
+def connect_worker(host, result_queue):
+    """Tries to connect to a single host and puts the successful transport object into the queue."""
+    transport = None
+    try:
+        transport = paramiko.Transport((host, 22))
+        # We still use the 5-second timeout for the handshake
+        transport.start_client(timeout=5)
+        transport.auth_password(username=FTP_USER, password=FTP_PASS)
+        
+        # If we reach here, the connection was successful.
+        # Put the result in the queue for the main thread to use.
+        print(f"✅ Connection to {host} succeeded first.")
+        result_queue.put(transport)
+    except (paramiko.SSHException, socket.timeout, EOFError) as e:
+        # This is an expected failure, just print a note.
+        print(f"ℹ️ Connection to {host} failed or was too slow: {e}")
+        if transport:
+            transport.close()
+    except Exception as e:
+        # Handle any other unexpected errors.
+        print(f"❌ Unexpected error connecting to {host}: {e}")
+        if transport:
+            transport.close()
+            
 def upload_to_ftp(filename):
+    """
+    Attempts to connect to two FTP hosts simultaneously and uses the first one that responds.
+    It now uses a corresponding directory for the primary and fallback hosts.
+    """
+    hosts = [FTP_HOST]
+    if FTP_HOST_FALLBACK:
+        hosts.append(FTP_HOST_FALLBACK)
+    
+    result_queue = queue.Queue()
+    threads = []
+    
+    print(f"--> Racing connections to {hosts} for uploading {filename}...")
+    for host in hosts:
+        thread = threading.Thread(target=connect_worker, args=(host, result_queue))
+        thread.daemon = True
+        thread.start()
+        threads.append(thread)
     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)
+        winning_transport = result_queue.get(timeout=7)
+        
+        # --- THIS IS THE NEW LOGIC ---
+        # 1. Determine which host won the race.
+        winning_host = winning_transport.getpeername()[0]
+        
+        # 2. Select the correct destination directory based on the winning host.
+        #    If the fallback directory isn't specified, it safely defaults to the primary directory.
+        if winning_host == FTP_HOST:
+            destination_directory = FTP_DIR
+        else:
+            destination_directory = FTP_DIR_FALLBACK if FTP_DIR_FALLBACK else FTP_DIR
+        print(f"--> Proceeding with upload to {winning_host} in directory {destination_directory}...")
+        # 3. Construct the full destination path using the selected directory.
+        sftp = paramiko.SFTPClient.from_transport(winning_transport)
+        destination_path = os.path.join(destination_directory, os.path.basename(filename))
         sftp.put(filename, destination_path)
+        
+        print(f"✅ Successfully uploaded {filename}.")
+        
         sftp.close()
-        transport.close()
-        print(f"Uploaded {filename} to FTP server")
+        winning_transport.close()
+    except queue.Empty:
+        print("❌ Critical Error: Neither FTP host responded in time.")
     except Exception as e:
-        print(f"FTP upload error: {e}")
+        print(f"❌ An unexpected error occurred during SFTP operation: {e}")
 '''
 
 pyx = cyper.inline(code, fast_indexing=True, directives=dict(boundscheck=False, wraparound=False, language_level=3))