updated with user db

app.py

@@ -60,15 +60,37 @@ def gemini_sql_query(prompt,input):
     return response.text
 
 
-st.set_page_config("chat pdf")
-st.header("chat with pdf using gemini")
+st.set_page_config("chat database")
+st.header("chat with your database ")
+
+input=st.text_input("enter your input/question ")
+
+uploaded_file = st.file_uploader("Upload SQLite Database", type=["db"])
+
+if uploaded_file is not None:
+    # Read the contents of the uploaded database file
+    connection = sqlite3.connect("uploaded.db")
+    cursor = connection.cursor()
+    cursor.executescript(uploaded_file.read().decode())
+    connection.close()
+    st.success("Database uploaded successfully.")
 
-input=st.text_input("enter the query")
 submit=st.button("submit")
 
 
 
 
+if submit and uploaded_file:
+    query=gemini_sql_query(prompt,input)
+    response=read_sql_query(query,"uploaded.db")
+    print(query)
+    st.header("response")
+    for row in response:
+    # Convert integers to strings
+        values = [str(value) for value in row]
+    # Write the values to the Streamlit component
+        st.write(*values)
+
 if submit:
     query=gemini_sql_query(prompt,input)
     response=read_sql_query(query,"student.db")
@@ -80,4 +102,3 @@ if submit:
     # Write the values to the Streamlit component
         st.write(*values)
 
-