Update views.py

views.py

@@ -3,6 +3,7 @@ import pandas as pd
 import tempfile
 from bertopic import BERTopic
 from src.reddit import RedditBot
+from pretty_html_table import build_table
 from flask import Blueprint, render_template, request, send_file, redirect, url_for, send_from_directory
 
 # DOWNLOADS_PATH = os.path.join(os.getcwd(), 'downloads')
@@ -51,7 +52,7 @@ def home():
             docs_df.to_csv('docs_with_topics_info.csv', index=False)
             send_file('docs_with_topics_info.csv', as_attachment=True)
             return render_template('success.html', 
-                                   topics = [topics_df.to_html(classes='data')], 
+                                   topics = [build_table(topics_df, 'blue_light')], 
                                    titles=topics_df.columns.values,
                                    docs = [docs_df.to_html(classes='data')],
                                    docs_title=docs_df.columns.values