try to parse into chart

app.py

@@ -50,6 +50,13 @@ def from_gpt(query: str):
     # Run the query using the agent executor
     return agent_executor.run(query)
 
+def parse_into_chart(response: str):
+    if ":" not in response:
+        return response
+
+    data = response.split(":")[1]
+    return alt.Chart(data)
+
 
 
 def circle_chart():
@@ -79,7 +86,8 @@ def get_response(prompt: str, *kargs):
         response = from_db(table)
     else:
         on_render = st.write
-        response = from_gpt(prompt)
+        ans = from_gpt(prompt)
+        response = parse_into_chart(ans)
 
     return (response, on_render)