checking db conn

app.py

@@ -1,12 +1,75 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
+import pyodbc
+# from sqlalchemy import create_engine
 
 # Clients for each model provider
 llama_client = InferenceClient(provider="sambanova", api_key=os.environ["HF_TOKEN"])
 minimax_client = InferenceClient(provider="novita", api_key=os.environ["HF_TOKEN"])
 mistral_client = InferenceClient(provider="together", api_key=os.environ["HF_TOKEN"])
 
+# Global objects
+db_connection = None  
+
+def connect_sql_pyodbc():
+    server = os.getenv("SQL_SERVER")
+    database = os.getenv("SQL_DATABASE")
+    username = os.getenv("SQL_USERNAME")
+    password = os.getenv("SQL_PASSWORD")
+
+    conn_str = (
+        f"Driver={{ODBC Driver 17 for SQL Server}};"
+        f"Server={server};"
+        f"Database={database};"
+        f"UID={username};"
+        f"PWD={password};"
+    )
+
+    try:
+        conn = pyodbc.connect(conn_str)
+        print("✅ pyodbc connection successful")
+        return conn
+    except Exception as e:
+        print(f"❌ pyodbc connection failed: {e}")
+        return None
+
+def get_sql_connection():
+    global db_connection
+
+    if db_connection is not None:
+        try:
+            db_connection.cursor()  # test if still open
+            return db_connection
+        except Exception as e:
+            print(f"❌ SQL connection failed: {e}")
+            db_connection = None  # reset if broken
+
+    # Reconnect if needed
+    db_connection = connect_sql_pyodbc()
+    return db_connection
+    
+# def connect_sql_sqlalchemy():
+#     server = os.getenv("SQL_SERVER")
+#     database = os.getenv("SQL_DATABASE")
+#     username = os.getenv("SQL_USERNAME")
+#     password = os.getenv("SQL_PASSWORD")
+#     driver = "ODBC+Driver+17+for+SQL+Server"  # URL encoded version
+
+#     conn_url = (
+#         f"mssql+pyodbc://{username}:{password}@{server}/{database}"
+#         f"?driver={driver.replace(' ', '+')}"
+#     )
+
+#     try:
+#         engine = create_engine(conn_url)
+#         conn = engine.connect()
+#         print("✅ SQLAlchemy connection successful")
+#         return conn
+#     except Exception as e:
+#         print(f"❌ SQLAlchemy connection failed: {e}")
+#         return None
+
 # Format chat history for Markdown display
 def format_chat_history(chat_history):
     formatted = ""
@@ -26,10 +89,14 @@ def format_chat_history(chat_history):
 # Main chat handler
 def chat_with_model(model_choice, prompt, image_url, chat_history):
     if not prompt:
-        return "Please enter a text prompt.", chat_history, "", ""
+        return "❌ Please enter a text prompt.", chat_history, "", ""
 
     if chat_history is None:
         chat_history = []
+        
+    conn = get_sql_connection()
+    if conn is None:
+        return "❌ Failed to connect to database.", chat_history, "", ""
 
     try:
         # === LLaMA 4 ===
@@ -67,12 +134,12 @@ def chat_with_model(model_choice, prompt, image_url, chat_history):
             chat_history.append({"role": "assistant", "content": bot_msg})
 
         else:
-            return "Unsupported model selected.", chat_history, "", ""
+            return "❌ Unsupported model selected.", chat_history, "", ""
 
         return format_chat_history(chat_history), chat_history, "", ""
 
     except Exception as e:
-        return f"Error: {e}", chat_history
+        return f"❌ Error: {e}", chat_history, "", ""
 
 # Gradio interface
 with gr.Blocks() as demo: