Update app.py

app.py

@@ -5,13 +5,6 @@ import networkx as nx
 from pyvis.network import Network
 import tempfile
 import os
-import jinja2
-
-# Fix pyvis missing template issue
-if Network.template is None:
-    template_loader = jinja2.PackageLoader("pyvis", "templates")
-    Network.template = jinja2.Environment(loader=template_loader).get_template("template.html")
-
 
 def process_file(file):
     file_ext = os.path.splitext(file.name)[1].lower()
@@ -28,7 +21,7 @@ def process_file(file):
         return f"Missing columns: {', '.join(missing)}", None, None, None, None
 
     df['date'] = pd.to_datetime(df['timestamp'], unit='s', errors='coerce')
-    df = df.dropna(subset=['date'])  # drop rows with invalid timestamps
+    df = df.dropna(subset=['date'])
     df['date_only'] = df['date'].dt.date
 
     # Messages per Day
@@ -54,7 +47,6 @@ def process_file(file):
     os.remove(file.name)
     return "Success", fig1, fig2, fig3, build_network_html(df)
 
-
 def build_network_html(df):
     G = nx.DiGraph()
 
@@ -85,7 +77,6 @@ def build_network_html(df):
 
     return html_content
 
-
 def search_keyword(keyword):
     if not os.path.exists("latest_data.csv"):
         return pd.DataFrame(columns=['date', 'author', 'like_count', 'text'])
@@ -96,11 +87,14 @@ def search_keyword(keyword):
         return pd.DataFrame(columns=['date', 'author', 'like_count', 'text'])
 
     mask = df['text'].astype(str).str.contains(keyword, case=False, na=False)
-    result = df.loc[mask, ['timestamp', 'author', 'text', 'like_count']] if 'like_count' in df.columns else df.loc[mask, ['timestamp', 'author', 'text']]
+    result = df.loc[mask, ['timestamp', 'author', 'text']]
+    if 'like_count' in df.columns:
+        result['like_count'] = df.loc[mask, 'like_count']
+    else:
+        result['like_count'] = None
     result['date'] = pd.to_datetime(result['timestamp'], unit='s', errors='coerce')
     result = result.dropna(subset=['date'])
-    return result[['date', 'author', 'like_count', 'text']] if 'like_count' in result.columns else result[['date', 'author', 'text']]
-
+    return result[['date', 'author', 'like_count', 'text']].head(100)
 
 # Gradio interface
 with gr.Blocks() as demo: