Create app.py

app.py

@@ -0,0 +1,35 @@
+from pandasai.llm.local_llm import LocalLLM
+from pandasai import Agent
+import streamlit as st
+import pandas as pd
+
+model = LocalLLM(
+    api_base='http://0.0.0.0:11434/v1',
+    model='llama3'
+)
+
+st.title('MBA data analysis with LLAMA3')
+
+upload_file = st.sidebar.file_uploader(
+    'upload a csv file',
+    type=["csv"]
+)
+
+if upload_file is not None:
+    data =pd.read_csv(upload_file)
+    st.write(data.head(7))
+    agent = Agent(data, config={"llm": model})
+    prompt = st.text_input('Prompt:')
+
+    if st.button("Generation"):
+        if prompt:
+            with st.spinner("Generating rep ...."):
+                st.write(agent.chat(prompt))
+
+
+
+
+
+
+
+