Update app.py

app.py

@@ -7,6 +7,9 @@ import gradio as gr
 import numpy as np
 import time
 import os
+
+import pyodbc
+
 #import pkg_resources
 
 '''
@@ -18,6 +21,38 @@ for package, version in installed_packages.items():
     print(f"{package}=={version}")
 '''
 
+'''
+# Replace the connection parameters with your SQL Server information
+server = 'your_server'
+database = 'your_database'
+username = 'your_username'
+password = 'your_password'
+driver = 'SQL Server'  # This depends on the ODBC driver installed on your system
+
+# Create the connection string
+connection_string = f'DRIVER={{{driver}}};SERVER={server};DATABASE={database};UID={username};PWD={password}'
+
+# Connect to the SQL Server
+conn = pyodbc.connect(connection_string)
+
+#============================================================================
+# Replace "your_query" with your SQL query to fetch data from the database
+query = 'SELECT * FROM your_table_name'
+
+# Use pandas to read data from the SQL Server and store it in a DataFrame
+df = pd.read_sql_query(query, conn)
+
+# Close the SQL connection
+conn.close()
+'''
+
+data = {
+    "year": [1896, 1900, 1904, 2004, 2008, 2012],
+    "city": ["athens", "paris", "st. louis", "athens", "beijing", "london"]
+}
+table = pd.DataFrame.from_dict(data)
+
+
 # Load the chatbot model
 chatbot_model_name = "microsoft/DialoGPT-medium" 
 tokenizer = AutoTokenizer.from_pretrained(chatbot_model_name)
@@ -28,19 +63,13 @@ sql_model_name = "microsoft/tapex-large-finetuned-wtq"
 sql_tokenizer = TapexTokenizer.from_pretrained(sql_model_name)
 sql_model = BartForConditionalGeneration.from_pretrained(sql_model_name)
 
-data = {
-    "year": [1896, 1900, 1904, 2004, 2008, 2012],
-    "city": ["athens", "paris", "st. louis", "athens", "beijing", "london"]
-}
-table = pd.DataFrame.from_dict(data)
-
-sql_response = None
+#sql_response = None
 
 def predict(input, history=[]):
 
-    global sql_response
+    #global sql_response
     # Check if the user input is a question
-    is_question = "?" in input
+    #is_question = "?" in input
 
     '''
     if is_question: