first commit

.gitignore

@@ -0,0 +1,6 @@
+Chinook_Sqlite.sql.txt
+test_notebook.ipynb
+app_class.py
+.env
+SQL_chat/
+__pycache__/

README.md

@@ -1,8 +1,8 @@
 ---
 title: SQL Chatbot
 emoji: 💻
-colorFrom: yellow
-colorTo: pink
+colorFrom: white
+colorTo: black
 sdk: streamlit
 sdk_version: 1.33.0
 app_file: app.py

app.py

@@ -0,0 +1,76 @@
+import streamlit as st
+from openai import OpenAI
+from langchain_community.utilities import SQLDatabase
+from langchain_google_cloud_sql_mysql import MySQLEngine
+from langchain_openai import ChatOpenAI
+from langchain_community.agent_toolkits import create_sql_agent
+from dotenv import load_dotenv
+load_dotenv()
+import streamlit as st
+import random
+import time
+
+# Create a SQL chatbot instance
+username = "root" 
+host = "35.204.50.120" 
+port = "3306"
+mydatabase = "Chinook"
+pg_uri = f"mysql+pymysql://{username}:{password}@{host}:{port}/{mydatabase}"
+db = SQLDatabase.from_uri(pg_uri)
+llm = ChatOpenAI(model="gpt-4-turbo", temperature=0)
+agent_executor = create_sql_agent(llm, db=db, agent_type="openai-tools", verbose=False)
+
+#sidebar 
+st.sidebar.title("SQL Chatbot")
+st.sidebar.write("This is a simple chatbot that can answer your SQL queries.")
+#list of the tables in database
+tables = db.get_table_names()
+st.sidebar.write("Tables in the database:")
+st.sidebar.write(tables)
+
+yes_list = ["yes","Yes", "yes", "YES", "Y", "y","yues", "yeah", "sure", "yea", "ok", "okay", "fine", "cool", "alright", "yup", "yep", "ya", "ye"]
+def add_spaces_before_newline(input_string):
+    modified_chars = []
+    for char in input_string:
+        if char == '\n':
+            modified_chars.append('  \n')
+        else:
+            modified_chars.append(char)
+    return ''.join(modified_chars)
+
+def run_query(query):
+    if query: 
+        if query in yes_list:
+            query = "Give a list of all the tables and their column in bullet points"
+            answer = agent_executor.invoke(query)
+        else:
+            answer = agent_executor.invoke(query)
+        return answer["output"]
+    else:
+        return """Welcome, I am a SQL chatbot. Please type a query in the chatbox to get started.
+                To get started do you want me to provide you will the tables and their respective columns? Simply answer yes."""
+
+st.title("SQL Chatbot")
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+if prompt := st.chat_input("Type a query..."):
+    with st.chat_message("user"):
+        st.markdown(prompt)
+    st.session_state.messages.append({"role": "user", "content": prompt})
+
+def response_generator():
+    response = run_query(prompt)
+    print(response) 
+    for word in response.splitlines(keepends=True):
+        yield word + " "
+        time.sleep(0.05)
+
+with st.chat_message("assistant"):
+    response = st.write_stream(response_generator())
+st.session_state.messages.append({"role": "assistant", "content": response})

requirements.txt

@@ -0,0 +1,82 @@
+aiohttp==3.9.5
+aiosignal==1.3.1
+annotated-types==0.6.0
+anyio==4.3.0
+asttokens==2.4.1
+attrs==23.2.0
+cachetools==5.3.3
+certifi==2024.2.2
+cffi==1.16.0
+charset-normalizer==3.3.2
+cloud-sql-python-connector==1.9.1
+colorama==0.4.6
+comm==0.2.2
+cryptography==42.0.5
+dataclasses-json==0.6.4
+debugpy==1.8.1
+decorator==5.1.1
+distro==1.9.0
+executing==2.0.1
+frozenlist==1.4.1
+google-auth==2.29.0
+greenlet==3.0.3
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.0
+idna==3.7
+ipykernel==6.29.4
+ipython==8.23.0
+jedi==0.19.1
+jsonpatch==1.33
+jsonpointer==2.4
+jupyter_client==8.6.1
+jupyter_core==5.7.2
+langchain==0.1.16
+langchain-community==0.0.34
+langchain-core==0.1.45
+langchain-google-cloud-sql-mysql==0.2.1
+langchain-openai==0.1.3
+langchain-text-splitters==0.0.1
+langsmith==0.1.49
+marshmallow==3.21.1
+matplotlib-inline==0.1.7
+multidict==6.0.5
+mypy-extensions==1.0.0
+nest-asyncio==1.6.0
+numpy==1.26.4
+openai==1.23.2
+orjson==3.10.1
+packaging==23.2
+parso==0.8.4
+platformdirs==4.2.0
+prompt-toolkit==3.0.43
+psutil==5.9.8
+pure-eval==0.2.2
+pyasn1==0.6.0
+pyasn1_modules==0.4.0
+pycparser==2.22
+pydantic==2.7.0
+pydantic_core==2.18.1
+Pygments==2.17.2
+PyMySQL==1.1.0
+python-dateutil==2.9.0.post0
+pywin32==306
+PyYAML==6.0.1
+pyzmq==26.0.2
+regex==2024.4.16
+requests==2.31.0
+rsa==4.9
+six==1.16.0
+sniffio==1.3.1
+SQLAlchemy==2.0.29
+stack-data==0.6.3
+tenacity==8.2.3
+tiktoken==0.6.0
+tornado==6.4
+tqdm==4.66.2
+traitlets==5.14.3
+typing-inspect==0.9.0
+typing_extensions==4.11.0
+urllib3==2.2.1
+wcwidth==0.2.13
+yarl==1.9.4