first commit

narendra-bluebash · narendra-bluebash · commit 42f7c823e5b1 · 2024-05-22T18:31:58.000+05:30
diff --git a/.env.sample b/.env.sample
@@ -0,0 +1 @@
+GOOGLE_API_KEY="your_api_key"
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+.env
diff --git a/.ipynb_checkpoints/Untitled-checkpoint.ipynb b/.ipynb_checkpoints/Untitled-checkpoint.ipynb
@@ -0,0 +1,6 @@
+{
+ "cells": [],
+ "metadata": {},
+ "nbformat": 4,
+ "nbformat_minor": 5
+}
diff --git a/README.md b/README.md
@@ -13,6 +13,46 @@ The chatbot will be designed to understand and respond to layman language questi
 
 A user interface will be created using Streamlit where users can input their PostgreSQL credentials and ask questions.
 
+### Setup Instructions
+
+Follow these steps to set up the project on your local machine:
+
+
+**1. Clone the Repository**
+Begin by cloning the repository to your local machine:
+```
+https://github.com/langchain-tech/postgres-chatbot.git
+cd postgres-chatbot
+```
+
+**2. Create a Virtual Environment**
+It is recommended to create a virtual environment to manage dependencies:
+```
+python -m venv venv
+source venv/bin/activate   # On Windows, use `venv\Scripts\activate`
+```
+
+**3. Install Dependencies**
+Install the necessary packages listed in the requirements.txt file:
+```
+pip install -r requirements.txt
+```
+
+
+**4. Set Up Environment Variables**
+Create a .env file in the root directory of your project and add the required environment variables. For example:
+```
+LANGCHAIN_API_KEY=your_langchain_api_key
+```
+
+
+**5. Start the Application**
+
+Run the application using Streamlit:
+```
+streamlit run app.py
+```
+
 ### Notes
 
 - Questions will not be limited to one table, so it should be done to ask from database
diff --git a/app.py b/app.py
@@ -0,0 +1,71 @@
+import os
+import asyncio
+import streamlit as st
+from langchain_community.utilities import SQLDatabase
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_community.agent_toolkits import create_sql_agent
+
+from dotenv import load_dotenv
+load_dotenv()
+
+os.environ["GOOGLE_API_KEY"]=os.getenv("GOOGLE_API_KEY")
+groq_api_key=os.getenv('GROQ_API_KEY')
+
+async def get_answer_by_sql_query(db,input):
+    llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro-latest",temperature=0)
+    agent_executor = create_sql_agent(llm, db=db, verbose=True)
+    
+    response= agent_executor.invoke({"input": input})
+    return response
+
+
+def main():
+    st.header('Chat with your Postgres Sql Database...')
+    
+    if "conversation" not in st.session_state:
+        st.session_state.conversation = None
+
+    if "activate_chat" not in st.session_state:
+        st.session_state.activate_chat = False
+
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+        st.session_state.chat_history=[]
+
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"], avatar = message['avatar']):
+            st.markdown(message["content"])
+
+    with st.sidebar:
+        st.subheader('Please Enter your credentials')
+        username = st.text_input("enter your username")
+        password=  st.text_input("enter your password")
+        db_name=  st.text_input("enter your database name")
+        if st.button('Process'):
+            if username is not None and password is not None:
+                postgres_uri = f"postgresql+psycopg2://{username}:{password}@localhost:5432/{db_name}"
+                st.write(f'Processed database: {postgres_uri}')
+                st.write(f'Connection to database successfully..')
+
+                db = SQLDatabase.from_uri(postgres_uri)
+                if "db" not in st.session_state:
+                    st.session_state.db = db
+                st.session_state.activate_chat = True
+
+    if st.session_state.activate_chat == True:
+        if prompt := st.chat_input("Ask your question from the Database"):
+            with st.chat_message("user", avatar = '👨🏻'):
+                st.markdown(prompt)
+            st.session_state.messages.append({"role": "user",  "avatar" :'👨🏻', "content": prompt})
+            db = st.session_state.db
+            response = asyncio.run(get_answer_by_sql_query(db,prompt))
+            cleaned_response=response["output"]
+            with st.chat_message("assistant", avatar='🤖'):
+                st.markdown(cleaned_response)
+            st.session_state.messages.append({"role": "assistant",  "avatar" :'🤖', "content": cleaned_response})
+        else:
+            st.markdown('Please Enter your credentials to chat')
+
+
+if __name__ == '__main__':
+    main()
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,119 @@
+aiohttp==3.9.5
+aiosignal==1.3.1
+altair==5.3.0
+annotated-types==0.7.0
+anyio==4.3.0
+asttokens @ file:///home/conda/feedstock_root/build_artifacts/asttokens_1698341106958/work
+async-timeout==4.0.3
+attrs==23.2.0
+blinker==1.8.2
+cachetools==5.3.3
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+comm @ file:///home/conda/feedstock_root/build_artifacts/comm_1710320294760/work
+dataclasses-json==0.6.6
+debugpy @ file:///croot/debugpy_1690905042057/work
+decorator @ file:///home/conda/feedstock_root/build_artifacts/decorator_1641555617451/work
+distro==1.9.0
+exceptiongroup @ file:///home/conda/feedstock_root/build_artifacts/exceptiongroup_1704921103267/work
+executing @ file:///home/conda/feedstock_root/build_artifacts/executing_1698579936712/work
+frozenlist==1.4.1
+gitdb==4.0.11
+GitPython==3.1.43
+google-ai-generativelanguage==0.6.4
+google-api-core==2.19.0
+google-api-python-client==2.129.0
+google-auth==2.29.0
+google-auth-httplib2==0.2.0
+google-generativeai==0.5.4
+googleapis-common-protos==1.63.0
+greenlet==3.0.3
+groq==0.7.0
+grpcio==1.64.0
+grpcio-status==1.62.2
+h11==0.14.0
+httpcore==1.0.5
+httplib2==0.22.0
+httpx==0.27.0
+idna==3.7
+importlib_metadata @ file:///home/conda/feedstock_root/build_artifacts/importlib-metadata_1710971335535/work
+ipykernel @ file:///home/conda/feedstock_root/build_artifacts/ipykernel_1708996548741/work
+ipython @ file:///home/conda/feedstock_root/build_artifacts/ipython_1701831663892/work
+jedi @ file:///home/conda/feedstock_root/build_artifacts/jedi_1696326070614/work
+Jinja2==3.1.4
+jsonpatch==1.33
+jsonpointer==2.4
+jsonschema==4.22.0
+jsonschema-specifications==2023.12.1
+jupyter_client @ file:///home/conda/feedstock_root/build_artifacts/jupyter_client_1710255804825/work
+jupyter_core @ file:///home/conda/feedstock_root/build_artifacts/jupyter_core_1710257447442/work
+langchain==0.2.0
+langchain-community==0.2.0
+langchain-core==0.2.0
+langchain-google-genai==1.0.4
+langchain-text-splitters==0.2.0
+langsmith==0.1.60
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+marshmallow==3.21.2
+matplotlib-inline @ file:///home/conda/feedstock_root/build_artifacts/matplotlib-inline_1713250518406/work
+mdurl==0.1.2
+multidict==6.0.5
+mypy-extensions==1.0.0
+nest_asyncio @ file:///home/conda/feedstock_root/build_artifacts/nest-asyncio_1705850609492/work
+numpy==1.26.4
+orjson==3.10.3
+packaging==23.2
+pandas==2.2.2
+parso @ file:///home/conda/feedstock_root/build_artifacts/parso_1712320355065/work
+pexpect @ file:///home/conda/feedstock_root/build_artifacts/pexpect_1706113125309/work
+pickleshare @ file:///home/conda/feedstock_root/build_artifacts/pickleshare_1602536217715/work
+pillow==10.3.0
+platformdirs @ file:///home/conda/feedstock_root/build_artifacts/platformdirs_1715777629804/work
+prompt-toolkit @ file:///home/conda/feedstock_root/build_artifacts/prompt-toolkit_1702399386289/work
+proto-plus==1.23.0
+protobuf==4.25.3
+psutil @ file:///home/conda/feedstock_root/build_artifacts/psutil_1705722404069/work
+psycopg2-binary==2.9.9
+ptyprocess @ file:///home/conda/feedstock_root/build_artifacts/ptyprocess_1609419310487/work/dist/ptyprocess-0.7.0-py2.py3-none-any.whl
+pure-eval @ file:///home/conda/feedstock_root/build_artifacts/pure_eval_1642875951954/work
+pyarrow==16.1.0
+pyasn1==0.6.0
+pyasn1_modules==0.4.0
+pydantic==2.7.1
+pydantic_core==2.18.2
+pydeck==0.9.1
+Pygments @ file:///home/conda/feedstock_root/build_artifacts/pygments_1714846767233/work
+pyparsing==3.1.2
+python-dateutil @ file:///home/conda/feedstock_root/build_artifacts/python-dateutil_1709299778482/work
+python-dotenv==1.0.1
+pytz==2024.1
+PyYAML==6.0.1
+pyzmq @ file:///croot/pyzmq_1705605076900/work
+referencing==0.35.1
+requests==2.32.1
+rich==13.7.1
+rpds-py==0.18.1
+rsa==4.9
+six @ file:///home/conda/feedstock_root/build_artifacts/six_1620240208055/work
+smmap==5.0.1
+sniffio==1.3.1
+SQLAlchemy==2.0.30
+stack-data @ file:///home/conda/feedstock_root/build_artifacts/stack_data_1669632077133/work
+streamlit==1.34.0
+tenacity==8.3.0
+toml==0.10.2
+toolz==0.12.1
+tornado @ file:///home/conda/feedstock_root/build_artifacts/tornado_1708363103305/work
+tqdm==4.66.4
+traitlets @ file:///home/conda/feedstock_root/build_artifacts/traitlets_1713535121073/work
+typing-inspect==0.9.0
+typing_extensions @ file:///home/conda/feedstock_root/build_artifacts/typing_extensions_1712329955671/work
+tzdata==2024.1
+uritemplate==4.1.1
+urllib3==2.2.1
+watchdog==4.0.0
+wcwidth @ file:///home/conda/feedstock_root/build_artifacts/wcwidth_1704731205417/work
+yarl==1.9.4
+zipp @ file:///home/conda/feedstock_root/build_artifacts/zipp_1695255097490/work
