Update app.py

app.py

@@ -10,13 +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'
-PY_CHAIN = 'py_chain'
 llm=OpenAI(temperature=0)
 db = SQLDatabase.from_uri("sqlite:///FXTrades.db")
 
@@ -24,9 +20,6 @@ db = SQLDatabase.from_uri("sqlite:///FXTrades.db")
 if JSON_DATA_LABEL not in st.session_state:
     st.session_state[JSON_DATA_LABEL] = {}
 
-if PY_CHAIN not in st.session_state:
-    st.session_state[PY_CHAIN] = {}
-
 def get_sql_agent():
     toolkit = SQLDatabaseToolkit(llm=llm,db=db)
 
@@ -75,50 +68,31 @@ def get_json_chain():
         prompt=PromptTemplate.from_template(prompt_template)
     )
 
-def get_python_chain():
-    return create_python_agent(
-        llm=llm,
-        tool=PythonREPLTool(),
-        verbose=True
-    )
-
 def plot_chart():
     json = st.session_state[JSON_DATA_LABEL]
     # print(json)
     if not json:
         return "no data to plot"
+#   Add Open AI call to format outcome in a table
+    return ""
 
-    # dataframe = pd.DataFrame(json)
-    dataframe = alt.Data(values=json)
-
-    return alt.Chart(dataframe).mark_line()
-
-def python_exec():
-    ans = st.session_state[PY_CHAIN]
-    
-    if not ans:
-        return "no data to plot"
-    return python_chain.run("Plot as a bar chart this result:{ans}",ans)
-    
+   
 # sql_agent = get_sql_agent()
 db_chain = get_db_chain()
 json_chain = get_json_chain()
 python_chain = get_python_chain()
 
 def from_gpt(query: str, plot: bool):
-    ans=""
     try:
         chains = [db_chain, json_chain] if plot else [db_chain]
         main_chain = SimpleSequentialChain(chains=chains, verbose=True)
         ans = main_chain.run(query)
+       # Execute Python if plot
+        st.session_state[JSON_DATA_LABEL] = ans if plot else {}
+        return ans
     except Exception as e:
         return "Please ask a proper business question related to selected datasets" 
 
-    # Execute Python if plot
-    st.session_state[PY_CHAIN] = ans if plot else {}
-
-    return ans
-
 def circle_chart():
     df = pd.DataFrame(np.random.randn(200, 3), columns=['a', 'b', 'c'])
     return alt.Chart(df).mark_circle().encode(
@@ -143,7 +117,7 @@ def get_response(prompt: str, *kargs):
         response = from_gpt(p, plot=True)
     elif prompt_lower == 'plot':
         on_render = st.write
-        response = python_exec()
+        response = plot_chart()
     else:
         on_render = st.write
         response = from_gpt(prompt, plot=False)