Update app.py

app.py

@@ -10,12 +10,27 @@ DATA_FILE = "data/gwf_2017-2021_specter2_base.json"
 THEMES = {"cluster": "fall", "year": "mint", "source": "phase"}
 
 
+def decorated_to_string(list_of_things):
+    if len(list_of_things) > 2:
+        return ", ".join(list_of_things[:-1]) + ", and " + list_of_things[-1]
+    else:
+        return " and ".join(list_of_things)
+
+
 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["year"] = df["year"].replace("", 0)
     df["year"] = df["year"].astype(int)
+
+    df["authors_trimmed"] = df.authors.apply(
+        lambda row: [(x[x.index(",") + 1:].strip() + " " + x.split(",")[0].strip()) if "," in x else x for x in row]
+    )
+    df["authors_trimmed"] = df.authors_trimmed.apply(
+        lambda row: decorated_to_string(row[:5]) + (" et al." if len(row) > 5 else "")
+    )
+
     if "publication_type" in df.columns:
         df["type"] = df["publication_type"]
         df = df.drop(columns=["point2d", "publication_type"])
@@ -65,7 +80,7 @@ with st.sidebar:
     DF.loc[df_mask, "opacity"] = 1.0
     st.write(f"Number of points: {DF[df_mask].shape[0]}")
 
-    color = st.selectbox("Color", ("cluster", "source"))
+    color = st.selectbox("Color", ("cluster", "year", "source"))
 
 
 fig = px.scatter(
@@ -76,9 +91,12 @@ fig = px.scatter(
     color=color,
     width=1000,
     height=800,
-    hover_data=["title", "authors", "year", "source", "type"],
+    custom_data=("title", "authors_trimmed", "year", "source", "keywords"),
     color_continuous_scale=THEMES[color],
 )
+fig.update_traces(
+    hovertemplate="<b>%{customdata[0]}</b><br>%{customdata[1]}<br>%{customdata[2]}<br><i>%{customdata[3]}</i><br><i font-size='8'>Keywords: %{customdata[4]}</i>"
+)
 fig.update_layout(
     # margin=dict(l=10, r=10, t=10, b=10),
     showlegend=False,
@@ -86,8 +104,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)