devin-ai commited on
Commit
2576d6b
·
verified ·
1 Parent(s): c234965

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -17
app.py CHANGED
@@ -65,26 +65,13 @@ st.header("chat with your sql database")
65
 
66
  input=st.text_input("enter your input/question ")
67
 
68
- uploaded_file = st.file_uploader("Upload SQLite Database", type=["db"])
69
 
70
  connection = sqlite3.connect(":memory:") # Use in-memory database
71
  cursor = connection.cursor()
72
 
73
- if uploaded_file is not None:
74
- # Read the contents of the uploaded database file in binary mode
75
- db_content = uploaded_file.read()
76
-
77
- # Decode binary data into a Unicode string
78
- db_content_str = db_content.decode("utf-8")
79
-
80
- # Execute the script
81
- cursor.executescript(db_content_str)
82
-
83
- # Display a success message after uploading the database file
84
- st.success("Database uploaded successfully.")
85
-
86
- # User input for SQL query
87
- input_text = st.text_input("Enter the query")
88
 
89
  submit=st.button("submit")
90
 
@@ -93,7 +80,7 @@ submit=st.button("submit")
93
 
94
  if submit and uploaded_file and input:
95
  query=gemini_sql_query(prompt,input)
96
- response=read_sql_query(query,"uploaded.db")
97
  print(query)
98
  st.header("response")
99
  col1, col2 = st.columns(2)
 
65
 
66
  input=st.text_input("enter your input/question ")
67
 
68
+ uploaded = st.file_uploader("Upload SQLite Database", type=["db"])
69
 
70
  connection = sqlite3.connect(":memory:") # Use in-memory database
71
  cursor = connection.cursor()
72
 
73
+ if uploaded is not None:
74
+ st.success("Database uploaded successfully
 
 
 
 
 
 
 
 
 
 
 
 
 
75
 
76
  submit=st.button("submit")
77
 
 
80
 
81
  if submit and uploaded_file and input:
82
  query=gemini_sql_query(prompt,input)
83
+ response=read_sql_query(query,uploaded)
84
  print(query)
85
  st.header("response")
86
  col1, col2 = st.columns(2)