adding conn

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import pandas as pd
 import plotly.express as px
 from sqlalchemy import create_engine, text
+from sshtunnel import SSHTunnelForwarder
+import os
 
 # --- Language toggle ---
 LANG = st.radio("Language / Език", ["English", "Български"], horizontal=True)
@@ -33,14 +35,32 @@ T = {
     }
 }
 
-# --- MySQL config from Hugging Face secrets ---
-MYSQL_HOST = st.secrets["MYSQL_HOST"]
-MYSQL_PORT = int(st.secrets["MYSQL_PORT"])
+# --- SSH tunnel setup ---
+SSH_KEY = st.secrets["SSH_KEY"]
+SSH_USER = st.secrets["SSH_USER"]
+SSH_HOST = st.secrets["DB_HOST"]
+REMOTE_BIND_HOST = st.secrets["REMOTE_BIND_HOST"]
+LOCAL_BIND_HOST = st.secrets["LOCAL_BIND_HOST"]
 MYSQL_USER = st.secrets["MYSQL_USER"]
 MYSQL_PASSWORD = st.secrets["MYSQL_PASSWORD"]
 MYSQL_DB = st.secrets["DB_NAME"]
-
-MYSQL_URL = f"mysql+pymysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{MYSQL_HOST}:{MYSQL_PORT}/{MYSQL_DB}"
+MYSQL_PORT = st.secrets["MYSQL_PORT"]
+
+# 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()
+
+MYSQL_URL = f"mysql+pymysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{LOCAL_BIND_HOST}:{MYSQL_PORT}/{MYSQL_DB}"
 engine = create_engine(MYSQL_URL)
 
 @st.cache_data(ttl=300)

requirements.txt

@@ -2,4 +2,5 @@ streamlit
 pandas
 pymysql
 sqlalchemy
-plotly
+plotly
+sshtunnel