returns query

app.py

@@ -46,7 +46,7 @@ prompt=[
     \nExample 5-List the classes with the highest average marks.
     SQL Command Example: SELECT CLASS FROM STUDENT GROUP BY CLASS HAVING AVG(MARKS) = (SELECT MAX(AVG(MARKS)) FROM STUDENT GROUP BY CLASS);
     
-    also the sql code should not have ``` in beginning or end and sql word in output and table name is "student" u want to retrieve from
+    also the sql code should not have ``` in beginning or end and sql word in output and 
      
     """
 
@@ -60,8 +60,8 @@ def gemini_sql_query(prompt,input):
     return response.text
 
 
-st.set_page_config("chat database")
-st.header("chat with your database ")
+st.set_page_config(""DataChat: Explore Your Database"")
+st.header("chat with your sql database ")
 
 input=st.text_input("enter your input/question ")
 
@@ -85,11 +85,19 @@ if submit and uploaded_file:
     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)
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        st.header("Response:")
+        for row in response:
+            st.write(row)
+
+    with col2:
+        st.header("Generated SQL Query:")
+        st.code(query)
+        if st.button("Copy SQL Query"):
+            st.write("SQL query copied to clipboard!")
+            st.text_area("SQL Query:", value=query, height=100)
 
 if submit:
     query=gemini_sql_query(prompt,input)
@@ -97,8 +105,6 @@ if submit:
     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)
+         st.write(*values)