Update app.py

app.py

@@ -6,7 +6,7 @@ import streamlit as st
 
 st.set_page_config(layout="wide")
 DATA_FILE = "data/anthology-2020-23_specter2_base.json"
-
+THEMES = {"cluster": "fall", "year": "mint", "source": "phase"}
 
 def load_df(data_file: os.PathLike):
     df = pd.read_json(data_file, orient="records")
@@ -39,6 +39,8 @@ with st.sidebar:
     )
     author_names = st.text_input("Author names (separated by comma)")
 
+    title = st.text_input("Title")
+
     start_year = int(start_year)
     end_year = int(end_year)
     df = DF[(DF["year"] >= start_year) & (DF["year"] <= end_year)]
@@ -55,18 +57,23 @@ with st.sidebar:
         )
         df = df[author_mask]
 
+    if title:
+        df = df[df.title.apply(lambda x: title.lower() in x.lower())]
+
     st.write(f"Number of points: {df.shape[0]}")
 
+    color = st.selectbox("Color", ("cluster", "year", "source"))
+
 
 fig = px.scatter(
     df,
     x="x",
     y="y",
-    color="cluster",
+    color=color,
     width=1000,
     height=800,
     hover_data=["title", "authors", "year", "source", "type"],
-    color_continuous_scale="fall",
+    color_continuous_scale=THEMES[color],
 )
 fig.update_layout(
     # margin=dict(l=10, r=10, t=10, b=10),
@@ -79,5 +86,4 @@ fig.update_layout(
 fig.update_xaxes(title="")
 fig.update_yaxes(title="")
 
-
-st.plotly_chart(fig, use_container_width=True)
+st.plotly_chart(fig, use_container_width=True)