Update app.py

app.py

@@ -10,6 +10,9 @@ from langchain import OpenAI
 from langchain import PromptTemplate, OpenAI, LLMChain
 from langchain.chains import SimpleSequentialChain
 from langchain import SQLDatabaseChain
+from langchain.agents.agent_toolkits import create_python_agent
+from langchain.tools.python.tool import PythonREPLTool
+from langchain.python import PythonREPL
 
 
 JSON_DATA_LABEL = 'json_data'
@@ -71,10 +74,17 @@ def get_json_chain():
 def plot_chart():
     json = st.session_state[JSON_DATA_LABEL]
     # print(json)
-    if not json:
-        return "no data to plot"
+#    if not json:
+#        return "no data to plot"
+
+    agent_executor = create_python_agent(
+    llm=llm,
+    tool=PythonREPLTool(),
+    verbose=True
+    )
+
 #   Add Open AI call to format outcome in a table
-    return ""
+    return agent_executor.run("Plot these results:{json}",json)
 
    
 # sql_agent = get_sql_agent()