rairo commited on
Commit
95ea877
·
verified ·
1 Parent(s): cc13f62

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -45,8 +45,8 @@ def bot():
45
  #user_question = data["user_question"]
46
  #print(json_table)
47
  print(user_question)
48
- data = eval(str(json_table))
49
- df = pd.DataFrame(data)
50
  print(list(df))
51
  #df = df.rename(columns={'dateupdated': 'date'})
52
  #df['date'] = pd.to_datetime(df['date'], format="%b %d", errors='coerce')
@@ -57,7 +57,7 @@ def bot():
57
  agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)
58
  response = agent.run(user_question)
59
  #response.headers.add('Access-Control-Allow-Origin', '*')
60
- return jsonify(response+' and the columns in your data are: ' + str(list(df)))
61
 
62
 
63
  if __name__ == "__main__":
 
45
  #user_question = data["user_question"]
46
  #print(json_table)
47
  print(user_question)
48
+ #data = eval(str(json_table))
49
+ df = pd.DataFrame(json_table)
50
  print(list(df))
51
  #df = df.rename(columns={'dateupdated': 'date'})
52
  #df['date'] = pd.to_datetime(df['date'], format="%b %d", errors='coerce')
 
57
  agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)
58
  response = agent.run(user_question)
59
  #response.headers.add('Access-Control-Allow-Origin', '*')
60
+ return jsonify(response)
61
 
62
 
63
  if __name__ == "__main__":