Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,152 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import pandas as pd
+import os
+import hashlib
+from datetime import datetime
+from pathlib import Path
+from huggingface_hub import CommitScheduler
+from localization_eval import evaluate_submission
+from PIL import Image
+
+# --- PERSISTENCE CONFIGURATION ---
+DATA_FILENAME = "submissions.csv"
+USER_FILENAME = "users.csv"
+DATA_DIR = Path("data")
+DATA_DIR.mkdir(exist_ok=True)
+
+DATA_PATH = DATA_DIR / DATA_FILENAME
+USER_PATH = DATA_DIR / USER_FILENAME
+
+# Initialize CommitScheduler
+# This will automatically sync everything in the /data folder to your HF Dataset
+repo_id = "your-username/your-private-dataset-name" # TODO: Change this
+
+scheduler = CommitScheduler(
+    repo_id=repo_id,
+    repo_type="dataset",
+    folder_path=DATA_DIR,
+    path_in_repo="data",
+    every=5,
+    token=os.getenv("HF_TOKEN")
+)
+
+# --- DB HELPERS (Replaced SQLite with Pandas/CSV) ---
+
+def init_db():
+    # Create files if they don't exist
+    if not USER_PATH.exists():
+        pd.DataFrame(columns=["username", "password"]).to_csv(USER_PATH, index=False)
+    if not DATA_PATH.exists():
+        pd.DataFrame(columns=["username", "bbox_mAP", "bbox_AP50", "segm_mAP", "segm_AP50", "timestamp"]).to_csv(DATA_PATH, index=False)
+
+def make_hashes(password):
+    return hashlib.sha256(str.encode(password)).hexdigest()
+
+def add_user(username, password):
+    with scheduler.lock:
+        df = pd.read_csv(USER_PATH)
+        if username in df['username'].values:
+            return False
+        new_user = pd.DataFrame([{"username": username, "password": make_hashes(password)}])
+        df = pd.concat([df, new_user], ignore_index=True)
+        df.to_csv(USER_PATH, index=False)
+        return True
+
+def login_user(username, password):
+    df = pd.read_csv(USER_PATH)
+    user_row = df[(df['username'] == username) & (df['password'] == make_hashes(password))]
+    return not user_row.empty
+
+def save_submission(username, bbox_mAP, bbox_AP50, segm_mAP, segm_AP50):
+    with scheduler.lock:
+        df = pd.read_csv(DATA_PATH)
+        new_sub = pd.DataFrame([{
+            "username": username,
+            "bbox_mAP": bbox_mAP,
+            "bbox_AP50": bbox_AP50,
+            "segm_mAP": segm_mAP,
+            "segm_AP50": segm_AP50,
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        }])
+        df = pd.concat([df, new_sub], ignore_index=True)
+        df.to_csv(DATA_PATH, index=False)
+
+def get_leaderboard_data():
+    if not DATA_PATH.exists():
+        return pd.DataFrame()
+    
+    df = pd.read_csv(DATA_PATH)
+    if df.empty:
+        return df
+
+    # Logic to get the BEST score per user
+    # Sorting by segm_mAP (desc) and timestamp (asc) to get best/earliest
+    df = df.sort_values(by=['segm_mAP', 'timestamp'], ascending=[False, True])
+    df_best = df.drop_duplicates(subset='username', keep='first')
+    
+    df_best = df_best.rename(columns={'segm_mAP': 'Best_segm_mAP'})
+    return df_best
+
+# --- CONFIGURATION & SETUP ---
+st.set_page_config(page_title="AI Benchmark Arena", page_icon="🏆", layout="wide")
+
+# (The rest of your ui_login_signup and main_app functions remain largely the same)
+# Just ensure you call the new CSV-based functions.
+
+# ... [KEEP YOUR ui_login_signup() and UI code here] ...
+def ui_login_signup():
+    
+    st.title("Welcome to Benchmark Arena 🏆")
+
+    tab1, tab2 = st.tabs(["Login", "Sign Up"])
+
+    with tab1:
+        st.subheader("Sign In")
+        username = st.text_input("Username", key="login_user")
+        password = st.text_input("Password", type='password', key="login_pass")
+
+        if st.button("Login"):
+            if login_user(username, password):
+                st.session_state['logged_in'] = True
+                st.session_state['username'] = username
+                st.rerun()
+            else:
+                st.error("Username or Password incorrect")
+
+    with tab2:
+        st.subheader("Create New Account")
+        new_user = st.text_input("Username", key="new_user")
+        new_pass = st.text_input("Password", type='password', key="new_pass")
+
+        if st.button("Sign Up"):
+            if add_user(new_user, new_pass):
+                st.success("Account created! Please navigate to Login.")
+            else:
+                st.warning("Username already exists.")
+
+
+
+def main_app():
+    # (Sidebar and Expanders code here - same as your original)
+    
+    # In your "Evaluate" button logic:
+    # Ensure variables are mapped correctly to the new save_submission()
+    if st.button("Evaluate"):
+        with st.spinner('Calculating...'):
+            bbox_mAP, bbox_AP50, segm_mAP, segm_AP50 = evaluate_submission("src/biv_query.json", save_path)
+            if bbox_mAP is not None:
+                save_submission(st.session_state['username'], bbox_mAP, bbox_AP50, segm_mAP, segm_AP50)
+                st.balloons()
+                st.success("Submission Successful!")
+
+if __name__ == '__main__':
+    init_db()
+    
+    if 'logged_in' not in st.session_state:
+        st.session_state['logged_in'] = False
+        st.session_state['username'] = None
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+    if not st.session_state['logged_in']:
+        ui_login_signup()
+    else:
+        main_app()