Update app.py

app.py

@@ -1,32 +1,26 @@
 import gradio as gr
-import pyodbc
 import pandas as pd
 from datetime import date
+import pymssql
 
-# Function to connect to SQL Server and run stored procedure
 def run_procedure(server, database, username, password, start_date, end_date):
     try:
-        # Create connection string
-        conn_str = (
-            f"DRIVER={{SQL Server Native Client 11.0}};"
-            f"SERVER={server};"
-            f"DATABASE={database};"
-            f"UID={username};"
-            f"PWD={password}"
-        )
-        with pyodbc.connect(conn_str) as conn:
-            cursor = conn.cursor()
-
-            # Execute stored procedure
-            cursor.execute("EXEC dbo.Temp_Get_EQUITY_DocumentProcurement ?, ?", start_date, end_date)
-            conn.commit()
-
-            # Fetch the output data from the timeline table
-            df = pd.read_sql(
-                "SELECT * FROM [DocumentAcquisition].dbo.Temp_Document_Procurement_Timeline "
-                "WHERE Document_Procurement_Name = 'EQUITY'",
-                conn
-            )
+        # Connect to SQL Server using pymssql (no ODBC required)
+        conn = pymssql.connect(server=server, user=username, password=password, database=database)
+        cursor = conn.cursor()
+
+        # Run the stored procedure with parameters
+        cursor.execute("EXEC dbo.Temp_Get_EQUITY_DocumentProcurement %s, %s", (start_date, end_date))
+        conn.commit()
+
+        # Now read the output table using pandas
+        query = """
+            SELECT * FROM [DocumentAcquisition].dbo.Temp_Document_Procurement_Timeline
+            WHERE Document_Procurement_Name = 'EQUITY'
+        """
+        df = pd.read_sql(query, conn)
+
+        conn.close()
 
         if df.empty:
             return "No data found for the given dates.", None