Update main.py

main.py

@@ -37,15 +37,15 @@ def bot():
     user_question = request.json.get("user_question")
     print(json_table)
     print(user_question)
-    df = pd.DataFrame(json_table)
+    #df = pd.DataFrame(json_table)
     #df['Profit'] = df['Profit'].apply(lambda x: "R{:.1f}".format((x)))
     #df['Revenue'] = df['Revenue'].apply(lambda x: "R{:.1f}".format((x)))
-    #llm = ChatOpenAI(model_name='gpt-3.5-turbo-0613', temperature=0, openai_api_key=os.getenv('OPENAI_API_KEY'))
-    llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
-    #agent = create_pandas_dataframe_agent(llm, df, verbose=True, agent_type=AgentType.OPENAI_FUNCTIONS)
-    agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)
-    response = agent.run(user_question)
-    return jsonify(response)
+    
+    #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)
+    return jsonify(json_table)
 
 
 if __name__ == "__main__":