Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,10 +4,12 @@ from langchain.chat_models import ChatOpenAI
 from langchain.agents import create_sql_agent
 from langchain_groq import ChatGroq
 from langchain_community.agent_toolkits import SQLDatabaseToolkit
-
-import tempfile
+from dotenv import load_dotenv
 import sqlite3
 import pandas as pd
+import os
+
+load_dotenv()
 
 
 def is_valid_sqlite(file_path):
@@ -61,9 +63,14 @@ def show_tables_as_df(db_path):
 # Streamlit UI
 st.title('🗃️ Chat with SQLite Database')
 
-st.write("This app lets you interact with a SQLite database using natural language queries powered by LLMs.")
+st.write("Interact with your SQLite databases stored on server using LLM-powered queries.")
 
-uploaded_file = st.file_uploader("Upload SQLite Database (.db file)", type=["db"])
+# List of .db files in "data" folder
+db_folder = "data"
+os.makedirs(db_folder, exist_ok=True)
+db_files = [f for f in os.listdir(db_folder) if f.endswith('.db')]
+
+selected_db = st.selectbox("Select Database", options=db_files)
 
 llm_provider = st.radio("Choose LLM Provider", options=['OPEN_ROUTER', 'GROQ', 'OPENAI'])
 model_name = st.text_input("Enter the Model Name", value='nousresearch/deephermes-3-mistral-24b-preview:free')
@@ -71,17 +78,15 @@ api_key = st.text_input("Enter Your API Key", type="password")
 query = st.text_area("Enter Your Query")
 
 try:
-    if uploaded_file is not None:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".db") as tmpfile:
-            tmpfile.write(uploaded_file.read())
-            tmp_db_path = tmpfile.name
+    if selected_db:
+        tmp_db_path = os.path.join(db_folder, selected_db)
 
         if not is_valid_sqlite(tmp_db_path):
-            st.error("The uploaded file is not a valid SQLite database.")
+            st.error("The selected file is not a valid SQLite database.")
         else:
-            st.success("Valid SQLite database uploaded!")
+            st.success("Valid SQLite database selected!")
 
-            # Show tables as pandas DataFrames
+            # Show tables
             st.info("Displaying first 10 rows from each table:")
             show_tables_as_df(tmp_db_path)
 
@@ -91,10 +96,11 @@ try:
                 elif not query.strip():
                     st.error("Please enter a query.")
                 else:
-                    st.info(f"Running query on `{uploaded_file.name}`...")
+                    st.info(f"Running query on `{selected_db}`...")
                     result = text_to_sql(query, tmp_db_path, llm_provider, api_key, model_name)
                     st.success("Query Result:")
                     st.write(result)
 
 except Exception as e:
     st.error(f"Error: {str(e)}")
+