Update script.py

script.py

@@ -1,24 +1,15 @@
 import os
 from langchain import OpenAI, SQLDatabase, SQLDatabaseChain
 
-api_key = os.getenv('OPENAI_API_KEY')
+import sqlite3
+import streamlit as st
 
-db = SQLDatabase.from_uri("sqlite:///Chinook.db"
-    , include_tables=['Album', 'Track']
-)
+st.title("English to SQL via LangChain")
 
-llm = OpenAI(temperature=0.0, openai_api_key=api_key)
+tables = ['Album', 'Artist', 'Track']
+db = SQLDatabase.from_uri("sqlite:///Chinook.db", include_tables=tables)
 
-db_chain = SQLDatabaseChain.from_llm(llm
-    , db
-    , verbose=True
-    , use_query_checker=True
-)
-
-queries = ("How many albums are there?"
-    , "How many tracks are there?"
-    , "Which album has the most tracks?"
-)
-
-for query in queries:
-    print(db_chain.run(query))
+con = sqlite3.connect("Chinook.db")
+for table in tables:
+    rows = cur.execute("select * from %s limit 5" %s table).fetchall()
+    st.table(rows)