added files

app.py

@@ -1,17 +1,14 @@
 import streamlit as st 
 import pandas as pd
-from pandasai import PandasAI
+from pandasai import SmartDataframe
 from pandasai.llm.openai import OpenAI
 
 openai_api_key = st.secrets["openai_api_key"]
 
 # create an LLM by instantiating OpenAI object, and passing API token
-llm = OpenAI(api_key = openai_api_key)
+llm = OpenAI(api_token = openai_api_key)
 
-# create PandasAI object, passing the LLM
-pandas_ai = PandasAI(llm)
-
-st.title("Prompt-driven data analysis with PandasAI")
+st.title("Data analysis with PandasAI")
 
 uploaded_file = st.file_uploader("Upload a CSV file for analysis", type=['csv'])
 
@@ -19,29 +16,29 @@ if uploaded_file is not None:
     df = pd.read_csv(uploaded_file)
     st.write(df.head(3))
 
-if uploaded_file is not None:
-    df = pd.read_csv(uploaded_file)
-    st.write(df.head(3))
+    # create PandasAI object, passing the LLM   
+    sdf = SmartDataframe(df, config={"llm": llm})
 
-    prompt = st.text_area("Enter your prompt:")
+    prompt = st.text_area("Enter your question:")
 
-    # Generate output
-    if st.button("Generate"):
-        if prompt:
-            st.write("Generating response...")
-        else:
-            st.warning("Please enter a prompt.")
-
-if uploaded_file is not None:
-    df = pd.read_csv(uploaded_file)
-    st.write(df.head(3))
-    prompt = st.text_area("Enter your prompt:")
-
-    # Generate output
-    if st.button("Generate"):
+    # Generate the answer
+    if st.button("Find the answer"):
         if prompt:
             # call pandas_ai.run(), passing dataframe and prompt
             with st.spinner("Generating response..."):
-                st.write(pandas_ai.run(df, prompt))
+                
+                st.write("The answer is: ")
+                st.write(sdf.chat(prompt))
+
+                st.write("Here is the code to get the answer: ")
+                code_string = sdf.last_code_generated
+
+                # Split the string into separate lines
+                code_lines = code_string.strip().split('\n')
+
+                # Print the lines
+                for line in code_lines:
+                    st.write(line)
+
         else:
-            st.warning("Please enter a prompt.")
+            st.warning("Please enter a question.")

requirements.txt

@@ -1,3 +1,3 @@
 streamlit
 pandasai
-pydantic
+scikit-learn