Update main.py

main.py

@@ -37,13 +37,14 @@ def bot():
     user_question = request.json.get("user_question")
     print(json_table)
     print(user_question)
-    #df = pd.DataFrame(json_table)
+    data = eval(json_table)
+    df = pd.DataFrame(data)
     #df['Profit'] = df['Profit'].apply(lambda x: "R{:.1f}".format((x)))
     #df['Revenue'] = df['Revenue'].apply(lambda x: "R{:.1f}".format((x)))
-    #llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
+    llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
     
-    #agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)
-    #response = agent.run(user_question)
+    agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)
+    response = agent.run(user_question)
     return json_table