Update script.py

script.py

@@ -39,16 +39,11 @@ llm = OpenAI(temperature=0.0, openai_api_key=OPENAI_API_KEY)
 db = SQLDatabase.from_uri("sqlite:///Chinook.db", include_tables=TABLES)
 db_chain = SQLDatabaseChain.from_llm(llm, db, use_query_checker=True, return_intermediate_steps=True)
 
-# db_chain = SQLDatabaseChain.from_llm(llm, db, use_query_checker=True)
-
 for query in QUERIES:
 
-    # result = db_chain.run(query)
-    # print(result)
-    
-    result = db_chain(query)
+    output = db_chain(query)
+    sql, result = output[1], output[5]
 
     st.text("Q: %s" % query)
-    st.text(result["intermediate_steps"])
-    st.text("A: %s" % result)
-
+    st.text("  (sql: %s)" % sql)
+    st.text("A: %s" % result)