Refactor app

app.py

@@ -45,7 +45,8 @@ def show_selectbox(demonstration_dir):
 
 
 def show_overview(data, recording_name, basedir):
-    st.title(f"Recording {recording_name}")
+    st.title('[WebLINX](https://mcgill-nlp.github.io/weblinx) Explorer')
+    st.header(f"Recording: `{recording_name}`")
 
     screenshot_size = st.session_state.get("screenshot_size_view_mode", "regular")
     show_advanced_info = st.session_state.get("show_advanced_information", False)
@@ -111,78 +112,17 @@ def show_overview(data, recording_name, basedir):
         col_act.markdown(action_str)
 
         if show_advanced_info:
-            colors = {
-                "good": "green",
-                "broken": "red",
-                "delayed": "orange"
-            }
             status = d["state"].get("screenshot_status", "unknown")
-            color = colors.get(status, "blue")
 
             text = ""
-            text += f'Status: **:{color}[{status.upper()}]**\n\n'
+            if status == "good":
+                text += f'**:green[Used in demo]**\n\n'
             text += f'Screenshot: `{d["state"]["screenshot"]}`\\\n'
             text += f'Page: `{d["state"]["page"]}`\n'
 
             col_act.markdown(text)
 
 
-def show_slide(data, example_index, recording_name, basedir):
-    columns = st.columns([1, 5], gap="large")
-
-    chat = gather_chat_history(data, example_index)
-    example = data[example_index]
-
-    # display current message
-    if example["type"] == "chat":
-        columns[0].markdown(format_chat_message(example), unsafe_allow_html=True)
-
-    # display chat history
-    with columns[0]:
-        for d in chat:
-            columns[0].markdown(format_chat_message(d), unsafe_allow_html=True)
-
-    img = find_screenshot(data, example_index, basedir)
-
-    if example["type"] == "browser":
-        event_type = example["action"]["intent"]
-        arguments = parse_arguments(example["action"])
-        event_type = example["action"]["intent"]
-
-        action = f"**{event_type}**({arguments})"
-
-        with columns[1]:
-            st.write(action)
-
-    with columns[1]:
-        st.image(img)
-
-
-def show_presentation(data, recording_name, basedir):
-    example_index = st.slider(
-        "example",
-        min_value=0,
-        max_value=len(data) - 1,
-        step=1,
-        label_visibility="hidden",
-    )
-
-    show_slide(data, example_index, recording_name=recording_name, basedir=basedir)
-
-
-def show_video(basedir):
-    # find the mp4 file in the basedir
-    video_filename = None
-    for filename in os.listdir(f"{basedir}"):
-        if filename.endswith(".mp4"):
-            video_filename = filename
-            video_path = f"{basedir}/{video_filename}"
-            st.video(video_path)
-
-    if not video_filename:
-        st.error("No video found")
-
-
 def load_recording(basedir):
     # Before loading replay, we need a dropdown that allows us to select replay.json or replay_orig.json
     # Find all files in basedir starting with "replay" and ending with ".json"
@@ -198,15 +138,12 @@ def load_recording(basedir):
         "Advanced Screenshot Info", False, key="show_advanced_information"
     )
     st.sidebar.checkbox(
-        "Enable HTML download", True, key="enable_html_download"
+        "Enable HTML download", False, key="enable_html_download"
     )
     replay_file = replay_file.replace(".json", "")
 
     metadata = load_json(basedir, "metadata")
 
-    extension_version = metadata["version"]
-    st.sidebar.markdown(f"**extension version**: {extension_version}")
-
     # convert timestamp to readable date string
     recording_start_timestamp = metadata["recordingStart"]
     recording_start_date = datetime.fromtimestamp(
@@ -239,10 +176,13 @@ def load_recording(basedir):
         "uses_ai_generated_output",
     ]:
         if form and key in form:
+            # Normalize the key to be more human-readable
+            key_name = key.replace("_", " ").title()
+
             if type(form[key]) == list:
-                st.sidebar.markdown(f"**{key}**: {', '.join(form[key])}")
+                st.sidebar.markdown(f"**{key_name}**: {', '.join(form[key])}")
             else:
-                st.sidebar.markdown(f"**{key}**: {form[key]}")
+                st.sidebar.markdown(f"**{key_name}**: {form[key]}")
 
     st.sidebar.markdown("---")
     if replay_dict and "status" in replay_dict:
@@ -266,7 +206,7 @@ def load_recording(basedir):
 
 
 def run():
-    mode = st.sidebar.radio("Mode", ["Overview"])
+    # mode = st.sidebar.radio("Mode", ["Overview"])
     demonstration_dir = "./demonstrations"
 
     # params = st.experimental_get_query_params()
@@ -296,15 +236,14 @@ def run():
         # use st.query_params as a dict instead
         st.query_params['recording'] = recording_name
 
-    if mode == "Overview":
-        with st.sidebar:
-            # Want a dropdown
-            st.selectbox(
-                "Screenshot size",
-                ["small", "regular", "large"],
-                index=1,
-                key="screenshot_size_view_mode",
-            )
+    with st.sidebar:
+        # Want a dropdown
+        st.selectbox(
+            "Screenshot size",
+            ["small", "regular", "large"],
+            index=1,
+            key="screenshot_size_view_mode",
+        )
 
     if recording_name is not None:
         basedir = f"{demonstration_dir}/{recording_name}"
@@ -313,12 +252,7 @@ def run():
         if not data:
             st.stop()
 
-        if mode == "Overview":
-            show_overview(data, recording_name=recording_name, basedir=basedir)
-        elif mode == "Presentation":
-            show_presentation(data, recording_name=recording_name, basedir=basedir)
-        elif mode == "Video":
-            show_video(basedir=basedir)
+        show_overview(data, recording_name=recording_name, basedir=basedir)
 
 
 if __name__ == "__main__":