Update app.py

app.py

@@ -5,7 +5,6 @@ import streamlit as st
 import google.generativeai as genai
 import sqlite3
 import os
-import pyperclip
 
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
@@ -51,15 +50,6 @@ def read_sql_query(sql,db):
         print(row)
     return rows
 
-st.set_page_config("DataChat: Explore Your Database")
-st.header("chat with your sql database")
-
-table_name = st.text_input("Enter the correct table name")
-
-question=st.text_input("enter your input/question")
-
-input=f"{question} in {table_name} table"
-
 
 #save uploaded file 
 def save_uploaded_file(uploaded_file):
@@ -75,13 +65,25 @@ uploaded_file = st.sidebar.file_uploader("Upload SQLite Database", type=["db"])
 if uploaded_file is not None:
     # Save the uploaded file
     db_path = save_uploaded_file(uploaded_file)
-    st.success("Database uploaded successfully.")
+    st.sidebar.success("Database uploaded successfully.")
+
+
+
+st.set_page_config("DataChat: Explore Your Database")
+st.header("chat with your sql database")
+
+table_name = st.text_input("Enter the correct table name")
+
+question=st.text_input("enter your input/question")
+
+input=f"{question} in {table_name} table"
 
 submit=st.button("submit")
 
 
 
 
+
 if submit and uploaded_file and input:
     query=gemini_sql_query(prompt,input)
     response=read_sql_query(query,db_path)
@@ -97,7 +99,7 @@ if submit and uploaded_file and input:
 
     with col2:
         st.header("Generated SQL Query:")
-        st.code(query)
+        st.code(query,height=300)