'credentials'

database.py

@@ -8,7 +8,7 @@ import os
 from langchain_community.utilities.sql_database import SQLDatabase
 
 
-db_params = {"host": st.secrets("DB_HOST"),
+db_params = {"host": os.getenv("DB_HOST"),
                 "user": os.getenv("DB_USER"),
                 "password": os.getenv("DB_PASSWORD"),
                 "port": 3306,
@@ -31,7 +31,7 @@ def initialize_database():
 def get_db():
     try:
         db = SQLDatabase.from_uri(
-            f"mysql+pymysql://{db_params['user']}:{db_params['password']}@{db_params['host']}/{db_params['database']}",
+            f"mysql+pymysql://{db_params['user']}:{db_params['password']}@{db_params["host"]}/{db_params['database']}",
             include_tables=['term_details_modified', 'veda_content_details', 'veda_content_modified']
         )
         #st.success("Database connection successful!")