Update main.py

main.py

@@ -4,17 +4,9 @@ import os
 import io
 from flask import Flask, request, jsonify
 from flask_cors import CORS, cross_origin
-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
-from langchain.chat_models import ChatOpenAI
-from langchain.agents.agent_types import AgentType
-from langchain.chat_models import ChatOpenAI
 import pandas as pd
-from langchain.llms import OpenAI
+
 from dotenv import load_dotenv
-import google.generativeai as palm
-from langchain.llms import GooglePalm
 import json
 
 
@@ -48,15 +40,6 @@ def bot():
     data = eval(str(json_table))
     df = pd.DataFrame(data)
     print(list(df))
-    #df = df.rename(columns={'dateupdated': 'date'})
-    #df['date'] = pd.to_datetime(df['date'], format="%b %d", errors='coerce')
-    #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'])
-    
-    agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)
-    response = agent.run(user_question)
-    #response.headers.add('Access-Control-Allow-Origin', '*')
     return jsonify(response)