Update app.py

app.py

@@ -67,20 +67,28 @@ input=st.text_input("enter your input/question ")
 
 uploaded_file = st.file_uploader("Upload SQLite Database", type=["db"])
 
+connection = sqlite3.connect(":memory:")  # Use in-memory database
+cursor = connection.cursor()
+
 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()
+    # Read the contents of the uploaded database file in binary mode
+    db_content = uploaded_file.read()
+    
+    # Execute the script directly with binary data
+    cursor.executescript(db_content)
+
+    # Display a success message after uploading the database file
     st.success("Database uploaded successfully.")
 
+    # User input for SQL query
+    input_text = st.text_input("Enter the query")
+
 submit=st.button("submit")
 
 
 
 
-if submit and uploaded_file:
+if submit and uploaded_file and input:
     query=gemini_sql_query(prompt,input)
     response=read_sql_query(query,"uploaded.db")
     print(query)
@@ -90,7 +98,8 @@ if submit and uploaded_file:
     with col1:
         st.header("Response:")
         for row in response:
-            st.write(row)
+            values = [str(value) for value in row]
+            st.write(*values)
 
     with col2:
         st.header("Generated SQL Query:")
@@ -99,12 +108,12 @@ if submit and uploaded_file:
             st.write("SQL query copied to clipboard!")
             st.text_area("SQL Query:", value=query, height=100)
 
-if submit:
-    query=gemini_sql_query(prompt,input)
-    response=read_sql_query(query,"student.db")
-    print(query)
-    st.header("response")
-    for row in response:
-        values = [str(value) for value in row]
-        st.write(*values)
+# if submit:
+#     query=gemini_sql_query(prompt,input)
+#     response=read_sql_query(query,"student.db")
+#     print(query)
+#     st.header("response")
+#     for row in response:
+#         values = [str(value) for value in row]
+#         st.write(*values)