Update app.py

app.py

@@ -1,8 +1,10 @@
 import os
+import re
 import pandas as pd
 import plotly.express as px
 import streamlit as st
 
+st.set_page_config(layout="wide")
 DATA_FILE = "data/anthology-2020-23_specter2_base.json"
 
 
@@ -22,6 +24,7 @@ def load_df(data_file: os.PathLike):
 def load_dataframe():
     return load_df(DATA_FILE)
 
+
 DF = load_dataframe()
 
 with st.sidebar:
@@ -31,32 +34,37 @@ with st.sidebar:
         ["ACL", "EMNLP", "NAACL", "TACL"],
     )
 
-    start_year, end_year = st.select_slider("Publication year", options=("2020", "2021", "2022", "2023"), value=("2020", "2023"))
-    author_names = st.text_input('Author names (separated by comma)')
+    start_year, end_year = st.select_slider(
+        "Publication year", options=("2020", "2021", "2022", "2023"), value=("2020", "2023")
+    )
+    author_names = st.text_input("Author names (separated by comma)")
 
     start_year = int(start_year)
     end_year = int(end_year)
     df = DF[(DF["year"] >= start_year) & (DF["year"] <= end_year)]
-    if len(venues) < 4:
+    if 0 < len(venues) < 4:
         selected_venues = [v.lower() for v in venues]
         df = df[df["source"].isin(selected_venues)]
+    elif not venues:
+        st.write(":red[Please select a venue]")
 
     if author_names:
-        authors = [a.strip().lower() for a in author_names.split(",")]
-        author_mask = df.authors.apply(lambda x: all(a in x for a in authors))
+        authors = [a.strip() for a in author_names.split(",")]
+        author_mask = df.authors.apply(
+            lambda row: all(any(re.match(rf".*{a}.*", x, re.IGNORECASE) for x in row) for a in authors)
+        )
         df = df[author_mask]
 
     st.write(f"Number of points: {df.shape[0]}")
 
 
-
 fig = px.scatter(
     df,
     x="x",
     y="y",
     color="cluster",
-    width=1200,
-    height=750,
+    width=1000,
+    height=800,
     hover_data=["title", "authors", "year", "source", "type"],
     color_continuous_scale="fall",
 )
@@ -71,4 +79,5 @@ fig.update_layout(
 fig.update_xaxes(title="")
 fig.update_yaxes(title="")
 
+
 st.plotly_chart(fig, use_container_width=True)