Update app.py

app.py

@@ -53,26 +53,26 @@ st.subheader('SHOW TABLE')
 input1=st.text_input("Enter table name")
 submit1=st.button("Show")
 if input1 is not None and submit1:
-conn = sqlite3.connect('data.sqlite')
-cur = conn.cursor()
-query = f"select * from {input1} limit 5"
-cur.execute(query)
-records = cur.fetchall()
-df1 = pd.read_sql_query(query, con=conn)
-conn.close()
-st.dataframe(df1)
+    conn = sqlite3.connect('data.sqlite')
+    cur = conn.cursor()
+    query = f"select * from {input1} limit 5"
+    cur.execute(query)
+    records = cur.fetchall()
+    df1 = pd.read_sql_query(query, con=conn)
+    conn.close()
+    st.dataframe(df1)
 
 st.subheader("GENERATE SQL RESULT")
 question=st.text_input("Enter question related to the database")
 submit2=st.button("Run")
 if question is not None and submit2:
-prompt_parts = [prompt_parts_1[0], question]
-response = model.generate_content(prompt_parts)
-query1 = response.text
-conn1 = sqlite3.connect('data.sqlite')
-cur1 = conn1.cursor()
-cur1.execute(query1)
-records = cur1.fetchall()
-df2 = pd.read_sql_query(query1, con=conn1)
-conn1.close()
-st.dataframe(df2)
+    prompt_parts = [prompt_parts_1[0], question]
+    response = model.generate_content(prompt_parts)
+    query1 = response.text
+    conn1 = sqlite3.connect('data.sqlite')
+    cur1 = conn1.cursor()
+    cur1.execute(query1)
+    records = cur1.fetchall()
+    df2 = pd.read_sql_query(query1, con=conn1)
+    conn1.close()
+    st.dataframe(df2)