QA module added

.gitignore

@@ -7,6 +7,8 @@ __pycache__/
 # C extensions
 *.so
 
+#Env variables
+.env
 # Distribution / packaging
 .Python
 build/

.gitpod.yml

@@ -0,0 +1,11 @@
+# This configuration file was automatically generated by Gitpod.
+# Please adjust to your needs (see https://www.gitpod.io/docs/introduction/learn-gitpod/gitpod-yaml)
+# and commit this file to your remote git repository to share the goodness with others.
+
+# Learn more from ready-to-use templates: https://www.gitpod.io/docs/introduction/getting-started/quickstart
+
+tasks:
+  - init: pip install -r requirements.txt
+    command: python app.py
+
+

data_QA.py

@@ -0,0 +1,17 @@
+import streamlit as st
+from langchain_google_genai import GoogleGenerativeAI
+from langchain_experimental.agents import create_pandas_agent
+import pandas as pd
+from dotenv import load_dotenv
+
+load_dotenv()  # take environment variables from .env.
+
+class DataQA:
+    def __init__(self, data):
+        self.data = data
+    def ask_csv(self):
+        llm = GoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_API_KEY)
+        csv_agent = create_pandas_agent(llm,self.data, verbose=True)
+        question = st.text_input("Ask your question:")
+        if question:
+            csv_agent.run(question)

requirements.txt

@@ -2,4 +2,7 @@ streamlit
 pandas
 numpy
 matplotlib
-seaborn
+seaborn
+langchain-google-genai
+langchain-experimental
+python-dotenv

test.py

@@ -0,0 +1,14 @@
+import streamlit as st
+from langchain_google_genai import GoogleGenerativeAI
+from langchain_experimental.agents import create_pandas_agent
+import pandas as pd
+from dotenv import load_dotenv
+
+load_dotenv()  # take environment variables from .env.
+
+data = pd.read_csv("")
+llm = GoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_API_KEY)
+csv_agent = create_pandas_agent(llm,self.data, verbose=True)
+question = st.text_input("Ask your question:")
+if question:
+    csv_agent.run(question)