Update script.py

script.py

@@ -5,7 +5,7 @@ import pandas as pd
 import sqlite3
 import streamlit as st
 
-API_KEY = os.getenv('OPENAI_API_KEY')
+OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
 
 TABLES = ['Album', 'Artist', 'Track']
 
@@ -20,7 +20,7 @@ def _get_metadata():
     cur = con.cursor()
 
     ts, cs = list(), list()
-    for table in tables:
+    for table in TABLES:
 
         rows = cur.execute("select * from %s limit 1" % table)
         cols = [k[0] for k in rows.description]
@@ -35,7 +35,7 @@ st.title("English to SQL via LangChain")
 st.subheader("table metadata")
 st.dataframe(_get_metadata())
 
-llm = OpenAI(temperature=0.0, openai_api_key=API_KEY)
+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)