Update app.py

app.py

@@ -35,9 +35,12 @@ data.to_sql(table_name, conn, index=False, if_exists='replace')
 valid_columns = list(data.columns)
 
 # Display the conversation thread
-st.write("### Conversation Thread:")
+st.markdown("### Conversation Thread:")
 for message in st.session_state.conversation:
-    st.write(message)
+    if message.startswith("User:"):
+        st.markdown(f"<p style='color:blue'><strong>{message}</strong></p>", unsafe_allow_html=True)
+    else:
+        st.markdown(f"<p style='color:green'><strong>{message}</strong></p>", unsafe_allow_html=True)
 
 # Step 3: Define SQL validation helpers
 def validate_sql(query, valid_columns):
@@ -82,7 +85,6 @@ if user_prompt:
             # Custom logic to return columns
             columns_response = f"The columns are: {', '.join(valid_columns)}"
             st.session_state.conversation.append(f"Bot: {columns_response}")
-            st.write(f"The columns are: {', '.join(valid_columns)}")
         else:
             # Generate SQL query based on user input
             columns = ', '.join(valid_columns)
@@ -93,27 +95,24 @@ if user_prompt:
             })
 
             # Debug: Display generated SQL query for inspection
-            st.write(f"Generated SQL Query:\n{generated_sql}")
+            st.session_state.conversation.append(f"Bot: Generated SQL Query:\n{generated_sql}")
 
             # Step 7: Validate SQL query
             if not validate_sql_with_sqlparse(generated_sql):
                 error_message = "Generated SQL is not valid."
                 st.session_state.conversation.append(f"Bot: {error_message}")
-                st.write(error_message)
             elif not validate_sql(generated_sql, valid_columns)[0]:
                 invalid_column_message = "Generated SQL references invalid columns."
                 st.session_state.conversation.append(f"Bot: {invalid_column_message}")
-                st.write(invalid_column_message)
             else:
                 # Step 8: Execute SQL query
                 result = pd.read_sql_query(generated_sql, conn)
-                st.write("Query Results:")
-                st.dataframe(result)
-                st.session_state.conversation.append("Bot: Here are the results of your query.")
+                st.session_state.conversation.append("Bot: Here are the results of your query:")
+                st.session_state.conversation.append(result.to_string(index=False))  # Add query result as string
+                
     except Exception as e:
         logging.error(f"An error occurred: {e}")
         error_message = f"Error: {e}"
         st.session_state.conversation.append(f"Bot: {error_message}")
-        st.write(f"Error: {e}")
 
 # Persist the conversation after each message