feat: add database connection

requirements.txt

@@ -0,0 +1,6 @@
+streamlit==1.31.1
+langchain==0.1.8
+langchain-community==0.0.21
+langchain-core==0.1.24
+langchain-openai==0.0.6
+mysql-connector-python==8.3.0

src/app.py

@@ -1,4 +1,10 @@
 import streamlit as st
+from langchain_community.utilities import SQLDatabase
+
+def initialize_database(host, port, username, password, database):
+  db_uri = f"mysql+mysqlconnector://{username}:{password}@{host}:{port}/{database}"
+  return SQLDatabase.from_uri(db_uri)
+
 
 st.set_page_config(initial_sidebar_state="expanded", page_title="Chat with a MySQL Database", page_icon=":speech_balloon:")
 
@@ -12,6 +18,17 @@ with st.sidebar:
     st.text_input("Password", key="password")
     st.text_input("Database", key="database")
     
+    if st.button("Connect"):
+        with st.spinner("Connecting to the database..."):
+          db = initialize_database(
+            username=st.session_state.username,
+            password=st.session_state.password,
+            host=st.session_state.name,
+            port=st.session_state.port,
+            database=st.session_state.database
+          )
+          st.success("Connected to the database!")
+    
 user_query = st.chat_input("Type a message...")
 
 if user_query != "":