Update app.py

app.py

@@ -66,13 +66,19 @@ st.header("chat with your sql database")
 input=st.text_input("enter your input/question ")
 
 #uploaded file
-uploaded = st.file_uploader("Upload SQLite Database", type=["db"])
+def save_uploaded_file(uploaded_file):
+    file_path = os.path.join(os.getcwd(), "uploaded.db")
+    with open(file_path, "wb") as f:
+        f.write(uploaded_file.getbuffer())
+    return file_path
 
-connection = sqlite3.connect(":memory:")  # Use in-memory database
-cursor = connection.cursor()
+# File uploader component
+uploaded_file = st.file_uploader("Upload SQLite Database", type=["db"])
 
-if uploaded is not None:
-    st.success("Database uploaded successfully")
+if uploaded_file is not None:
+    # Save the uploaded file
+    db_path = save_uploaded_file(uploaded_file)
+    st.success("Database uploaded successfully.")
 
 submit=st.button("submit")
 
@@ -81,7 +87,7 @@ submit=st.button("submit")
 
 if submit and uploaded and input:
     query=gemini_sql_query(prompt,input)
-    response=read_sql_query(query,uploaded)
+    response=read_sql_query(query,db_path)
     print(query)
     st.header("response")
     col1, col2 = st.columns(2)