add method to convert response into json

app.py

@@ -8,6 +8,7 @@ from langchain.sql_database import SQLDatabase
 from langchain.agents.agent_toolkits import SQLDatabaseToolkit
 from langchain.agents import create_sql_agent
 from langchain import OpenAI
+from langchain import PromptTemplate, OpenAI, LLMChain
 
 
 def tables_from_db():
@@ -50,6 +51,17 @@ def from_gpt(query: str):
     # Run the query using the agent executor
     return agent_executor.run(query)
 
+def parse_into_json(result: str):
+    prompt_template = "Reformat this {result} in JSON format"
+    llm=OpenAI(temperature=0)
+    llm_chain = LLMChain(
+        llm=llm,
+        prompt=PromptTemplate.from_template(prompt_template)
+    )
+    Final_Results=llm_chain(result)
+
+    Final_Results['text']
+
 def parse_into_chart(response: str):
     if ":" not in response:
         return response