Update app.py

app.py

@@ -1,6 +1,7 @@
 import logging
+import sqlite3
+import requests
 import streamlit as st
-from dotenv import load_dotenv, find_dotenv
 from langgraph.errors import GraphRecursionError
 from langchain_groq import ChatGroq
 from agent import SQLAgentRAG
@@ -11,8 +12,14 @@ from constant import GROQ_API_KEY, CONFIG
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Load environment variables
-load_dotenv(find_dotenv())
+def download_sqlite_db(url, local_filename='travel.sqlite'):
+    # Download the file
+    with requests.get(url, stream=True) as r:
+        r.raise_for_status()
+        with open(local_filename, 'wb') as f:
+            for chunk in r.iter_content(chunk_size=8192):
+                f.write(chunk)
+    return local_filename
 
 # Initialize the language model
 llm = ChatGroq(
@@ -50,6 +57,9 @@ def query_rag_agent(query: str):
         logger.error("Graph recursion limit reached; query processing failed.")
         return "Graph recursion limit reached. No SQL result generated.", ""
 
+URL = "https://storage.googleapis.com/benchmarks-artifacts/travel-db/travel2.sqlite"
+download_sqlite_db(URL)
+
 with st.sidebar:
     st.header("About Project")
     st.markdown(