Update app.py

app.py

@@ -19,7 +19,9 @@ def read_sql_query(sql, db):
         #st.write(row)
     conn.close()
     output_df = pd.DataFrame(rows)
-    return output_df
+    st.tablw(output_df)
+    
+    #return output_df
 
 
 
@@ -51,7 +53,7 @@ if i_pwd=='password':
     # Step 1: User input
     with st.chat_message(name= 'assistant'):
         st.markdown('Ask question and press Enter')
-    st.session_state.messages.append({"role":"assistant", "content": 'Ask question and press Enter' })
+        #st.session_state.messages.append({"role":"assistant", "content": 'Ask question and press Enter' })
 
     user_query= st.chat_input('')
     if user_query:
@@ -67,10 +69,9 @@ if i_pwd=='password':
         # Step 2: Backend processing on Enter
     #if  user_query:
         text='question:{}. table:{}'.format(user_query,create_table_statement.replace("text", 'VARCHAR') )
-        generated_sql= translator([text])
+        with st.spinner('Loading model. Please wait...'):
+            generated_sql= translator([text])
         generated_sql= generated_sql[0]['generated_text']
-        
-        
         i_gen_query = st.text_area("AI Generated SQL (edit if required)", generated_sql)
 
         
@@ -78,7 +79,7 @@ if i_pwd=='password':
         # Step 4: Process edited text on Enter
         if st.button("Show results") and i_gen_query:
             st.session_state.messages.append({"role":"assistant", "content": i_gen_query })
-            st.table(read_sql_query(i_gen_query, "Cars_db.sqlite"))
+            read_sql_query(i_gen_query, "Cars_db.sqlite"))
 
             # Step 5: Display final output
             #st.write("Final Output:", final_output)