Update main.py

main.py

@@ -2,7 +2,7 @@ import google.generativeai as palm
 import pandas as pd
 
 import io
-from flask import Flask, request
+from flask import Flask, request, jsonify
 from langchain.chains.question_answering import load_qa_chain
 from langchain import PromptTemplate, LLMChain
 from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent
@@ -33,6 +33,8 @@ def home():
 def bot(json_table, user_question):
     load_dotenv()
     #
+    json_table = request.args.get("json_table")
+    user_question = request.args.get("user_question")
     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)))
@@ -41,7 +43,7 @@ def bot(json_table, user_question):
     #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 response
+    return jsonify(response)
 
 
 if __name__ == "__main__":