Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,32 +19,30 @@ def load_df(file):
     df["scenario"] = df["trigger"] + " | " + df["modifier"]
     return df
 
-st.title("🔎 AI Vulnerability Benchmark Explorer (Live)")
-
 st.sidebar.header("Dataset")
 
-# Try loading from repo
 DEFAULT_PATH = "data/benchmark_results.csv"
-
 df = None
 
+# Try default dataset from repo
 try:
     df = load_df(DEFAULT_PATH)
     st.sidebar.success("Loaded default dataset from repository.")
-except Exception as e:
-    st.sidebar.warning("Default dataset not found in repo. Upload required.")
+except:
+    st.sidebar.warning("Default dataset missing. Upload required.")
 
-uploaded = st.sidebar.file_uploader("Upload your own CSV", type=["csv"])
-
-if uploaded is not None:
+# Allow optional upload from user
+uploaded = st.sidebar.file_uploader("Upload custom benchmark_results.csv", type=["csv"])
+if uploaded:
     df = load_df(uploaded)
     st.sidebar.success("Using uploaded dataset.")
 
 if df is None:
-    st.error("No dataset available. Please upload a CSV file.")
+    st.error("No dataset found. Upload a CSV to continue.")
     st.stop()
 
 
+
 # -----------------------------
 # FILTERS
 # -----------------------------