Upload 2 files

app.py

@@ -0,0 +1,45 @@
+import streamlit as st
+import pandas as pd
+import numpy as np
+from dotenv import load_dotenv
+from pandasai.llm.google_gemini import GoogleGemini
+from pandasai import  SmartDataframe
+import os
+
+load_dotenv()
+
+llm=GoogleGemini(api_key=os.getenv('GOOGLE_API_KEY'))
+
+def chat_with_csv(df,prompt):
+    pandas_ai = SmartDataframe(df,config={'llm':llm})
+    result=pandas_ai.chat(prompt)
+    return result
+      
+st.set_page_config(layout='wide')
+
+st.title('ChatCSV powered by LLM')
+
+input_csv=st.file_uploader('upload your csv file',type=['csv'])
+
+if input_csv is not None:
+    
+    col1,col2=st.columns([1,1])
+
+    with col1:
+        st.info('CSV Uploaded successfully')
+
+        data=pd.read_csv(input_csv)
+        
+        st.dataframe(data)
+
+    with col2:
+       st.info('Chat with CSV')
+
+       input_text=st.text_area('Enter your query')
+
+       if input_text is not None:
+           if st.button('Chat with CSV'):
+              st.info(f'Your query: {input_text}')
+              result=chat_with_csv(data,input_text)
+              st.success(result)
+

requirements.txt

@@ -0,0 +1,5 @@
+pandas==1.5.3
+numpy==1.23.4
+streamlit==1.36.0
+pandasai
+python-dotenv