Upload pandas_ai.py

pandas_ai.py

@@ -0,0 +1,23 @@
+import pandas as pd
+from pandasai import SmartDataframe
+from pandasai.llm import OpenAI
+import os
+
+def csv_file(file_upload,message):
+    df = pd.read_csv(file_upload,encoding='cp1252')
+    df = df.head(30)
+    os.environ['OPENAI_API_KEY'] = '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'
+    llm = OpenAI(temperature=0)
+    df_s = SmartDataframe(df, config={"llm": llm})
+    output = df_s.chat(message)
+    return output
+    
+def excel_file(file_upload,message):
+    df = pd.read_excel(file_upload,encoding='cp1252')
+    df = df.head(30)
+    os.environ['OPENAI_API_KEY'] = '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'
+    llm = OpenAI(temperature=0)
+    df_s = SmartDataframe(df, config={"llm": llm})
+    output = df_s.chat(message)
+    return output
+