Update script.py

script.py

@@ -42,9 +42,9 @@ db_chain = SQLDatabaseChain.from_llm(llm, db, use_query_checker=True, return_int
 for query in QUERIES:
 
     output = db_chain(query)
-    st.json(output)
-#    sql, result = output[1], output[5]
+#    st.json(output)
+    sql, result = output["intermediate_steps"][1], output["result"]
 
-#    st.text("Q: %s" % query)
-#    st.text("  (sql: %s)" % sql)
-#    st.text("A: %s" % result)
+    st.text("Q: %s" % query)
+    st.text("  (sql: %s)" % sql)
+    st.text("A: %s" % result)