Update app.py

app.py

@@ -9,10 +9,26 @@ DATA_FILE = "data/aclanthology2016-23_specter2_base.json"
 THEMES = {"cluster": "fall", "year": "mint", "source": "phase"}
 
 
+def to_string_authors(list_of_authors):
+    if len(list_of_authors) > 5:
+        return ", ".join(list_of_authors[:5]) + ", et al."
+    elif len(list_of_authors) > 2:
+        return ", ".join(list_of_authors[:-1]) + ", and " + list_of_authors[-1]
+    else:
+        return " and ".join(list_of_authors)
+
+
 def load_df(data_file: os.PathLike):
     df = pd.read_json(data_file, orient="records")
     df["x"] = df["point2d"].apply(lambda x: x[0])
     df["y"] = df["point2d"].apply(lambda x: x[1])
+
+    df["authors_trimmed"] = df.authors.apply(
+        lambda row: to_string_authors(
+            [(x[x.index(",") + 1 :].strip() + " " + x.split(",")[0].strip()) if "," in x else x for x in row]
+        )
+    )
+
     if "publication_type" in df.columns:
         df["type"] = df["publication_type"]
         df = df.drop(columns=["point2d", "publication_type"])
@@ -79,9 +95,12 @@ fig = px.scatter(
     color=color,
     width=1000,
     height=800,
-    hover_data=["title", "authors", "year", "source", "type"],
+    custom_data=("title", "authors_trimmed", "year", "source", "type"),
     color_continuous_scale=THEMES[color],
 )
+fig.update_traces(
+    hovertemplate="<b>%{customdata[0]}</b><br>%{customdata[1]}<br>%{customdata[2]}<br><i>%{customdata[3]}</i>"
+)
 fig.update_layout(
     # margin=dict(l=10, r=10, t=10, b=10),
     showlegend=False,
@@ -89,8 +108,14 @@ fig.update_layout(
         family="Times New Roman",
         size=30,
     ),
+    hoverlabel=dict(
+        align="left",
+        font_size=14,
+        font_family="Rockwell",
+        namelength=-1,
+    ),
 )
 fig.update_xaxes(title="")
 fig.update_yaxes(title="")
 
-st.plotly_chart(fig, use_container_width=True)
+st.plotly_chart(fig, use_container_width=True)