updated app, removed keys

app.py

@@ -1,24 +1,44 @@
+import os
 import streamlit as st
 
+from pathlib import Path
+from landingai.predict import Predictor
+from landingai.vision_pipeline import NetworkedCamera, FrameSet
+
+
+VIDEO_CACHE_PATH = Path("cached_data")
+VIDEO_CACHE_PATH.mkdir(exist_ok=True, parents=True)
+VIDEO_CACHE_PATH = VIDEO_CACHE_PATH / "latest.mp4"
+VIDEO_LEN_SEC = 10
+FPS = 2
+PLAYLIST_URL = (
+    "https://live.hdontap.com/hls/hosb1/topanga_swellmagnet.stream/playlist.m3u8"
+)
+API_KEY = os.environ["API_KEY"]
+ENDPOINT_ID = os.environ["ENDPOINT_ID"]
+
 
 st.title("Topanga Beach Surfer Counter")
 st.write(
     "This application will grab the latest 10s clip of surfers from the Topanga Beach surf cam"
     "and count the number of surfers there."
 )
-st.write("Please enter your LandingLens API key and Cloud Inference Endpoint ID.")
-api_key = st.text_input(
-    "LandingLens API Key", value=st.session_state.get("api_key", "")
-)
-endpoint_id = st.text_input(
-    "Cloud Inference Endpoint ID",
-    value=st.session_state.get("endpoint_id", ""),
-)
 
 
-def save(api_key: str, endpoint_id: str):
-    st.session_state["api_key"] = api_key
-    st.session_state["endpoint_id"] = endpoint_id
+def get_latest_surfer_count():
+    vid_src = NetworkedCamera(PLAYLIST_URL, fps=FPS)
+    surfer_model = Predictor(ENDPOINT_ID, api_key=API_KEY)
+
+    frs = FrameSet()
+    for i, frame in enumerate(vid_src):
+        if i >= VIDEO_LEN_SEC * FPS:
+            break
+        frs.extend(frame.run_predict(predictor=surfer_model).overlay_predictions())
+    frs.save_video(str(VIDEO_CACHE_PATH), video_fps=FPS, image_src="overlay")
+    surfers = frs.get_class_counts()["surfer"] / (VIDEO_LEN_SEC * FPS)
+    st.video(open(VIDEO_CACHE_PATH, "rb").read())
+    st.write(f"Surfer count: **{surfers}**")
 
 
-st.button("Save", on_click=save(api_key, endpoint_id))
+st.title("Surfer Counter")
+button = st.button("Get Topanga Beach Surfer Count", on_click=get_latest_surfer_count)

pages/run_inference.py

@@ -1,33 +0,0 @@
-import streamlit as st
-
-from pathlib import Path
-from landingai.predict import Predictor
-from landingai.vision_pipeline import NetworkedCamera, FrameSet
-
-
-VIDEO_CACHE_PATH = Path("cached_data")
-VIDEO_CACHE_PATH.mkdir(exist_ok=True, parents=True)
-VIDEO_CACHE_PATH = VIDEO_CACHE_PATH / "latest.mp4"
-VIDEO_LEN_SEC = 10
-FPS = 2
-PLAYLIST_URL = (
-    "https://live.hdontap.com/hls/hosb1/topanga_swellmagnet.stream/playlist.m3u8"
-)
-
-
-def get_latest_surfer_count():
-    vid_src = NetworkedCamera(PLAYLIST_URL, fps=FPS)
-    surfer_model = Predictor(st.session_state["endpoint_id"], api_key=st.session_state["api_key"])
-
-    frs = FrameSet()
-    for i, frame in enumerate(vid_src):
-        if i >= VIDEO_LEN_SEC * FPS:
-            break
-        frs.extend(frame.run_predict(predictor=surfer_model).overlay_predictions())
-    frs.save_video(str(VIDEO_CACHE_PATH), video_fps=FPS, image_src="overlay")
-    surfers = frs.get_class_counts()["surfer"] / (VIDEO_LEN_SEC * FPS)
-    st.video(open(VIDEO_CACHE_PATH, "rb").read())
-    st.write(f"Surfer count: **{surfers}**")
-
-st.title("Surfer Counter")
-button = st.button("Get Topanga Beach Surfer Count", on_click=get_latest_surfer_count)