Update app.py

app.py

@@ -8,6 +8,8 @@ import spacy
 from spacy import displacy
 import re
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+import pyodbc
+from azure.identity import DefaultAzureCredential
 
 # Configure the page to use the full width
 st.set_page_config(
@@ -21,6 +23,22 @@ from modules.morpho_analysis import get_repeated_words_colors, highlight_repeate
 from modules.syntax_analysis import visualize_syntax
 
 @st.cache_resource
+def get_db_connection():
+    server = 'aideatexdb.database.windows.net'
+    database = 'textdb'
+    username = 'manuel.var.ale_aideatext.ai#EXT#@manuelvaraleaideatext.onmicrosoft.com'
+    connection_string = f'Driver={{ODBC Driver 18 for SQL Server}};Server={server};Database={database}'
+    
+    credential = DefaultAzureCredential()
+    token = credential.get_token("https://database.windows.net/.default")
+    
+    conn = pyodbc.connect(connection_string, 
+                          attrs_before = {1256: f'Bearer {token.token}'})
+    return conn
+
+
+
+
 def load_chatbot_model():
     tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
     model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")