updating tunnel

app.py

@@ -35,7 +35,7 @@ T = {
     }
 }
 
-# --- SSH tunnel setup ---
+# --- Secrets ---
 SSH_KEY = st.secrets["SSH_KEY"]
 SSH_USER = st.secrets["SSH_USER"]
 SSH_HOST = st.secrets["DB_HOST"]
@@ -46,20 +46,30 @@ MYSQL_PASSWORD = st.secrets["MYSQL_PASSWORD"]
 MYSQL_DB = st.secrets["DB_NAME"]
 MYSQL_PORT = int(st.secrets["MYSQL_PORT"])
 
-# Write SSH key to temp file
+# --- Write SSH key to temp file ---
 with open("/tmp/ssh_key.pem", "w") as f:
     f.write(SSH_KEY)
 os.chmod("/tmp/ssh_key.pem", 0o600)
 
-server = SSHTunnelForwarder(
-    (SSH_HOST, 22),
-    ssh_username=SSH_USER,
-    ssh_pkey="/tmp/ssh_key.pem",
-    remote_bind_address=(REMOTE_BIND_HOST, MYSQL_PORT),
-    local_bind_address=(LOCAL_BIND_HOST, MYSQL_PORT)
-)
-server.start()
+# --- SSH Tunnel Setup (cached with check) ---
+@st.cache_resource
+def start_ssh_tunnel():
+    tunnel = SSHTunnelForwarder(
+        (SSH_HOST, 22),
+        ssh_username=SSH_USER,
+        ssh_pkey="/tmp/ssh_key.pem",
+        remote_bind_address=(REMOTE_BIND_HOST, MYSQL_PORT),
+        local_bind_address=(LOCAL_BIND_HOST, MYSQL_PORT)
+    )
+    if not tunnel.is_active:
+        tunnel.start()
+    return tunnel
+
+server = start_ssh_tunnel()
+if not server.is_active:
+    server.start()
 
+# --- SQLAlchemy engine ---
 MYSQL_URL = f"mysql+pymysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{LOCAL_BIND_HOST}:{MYSQL_PORT}/{MYSQL_DB}"
 engine = create_engine(MYSQL_URL)