Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import pandas as pd
 import os
 import datetime
+import pickle
+import inspect
 
 from google_sheet import *
 
@@ -11,27 +13,43 @@ st.title("🏆 Hackathon Leaderboard")
 # Submission Form
 # ========================
 
-uploaded_file = st.file_uploader("Upload your solution (.py)", type=["py"])
+uploaded_file = st.file_uploader("Upload your model/class (.pkl)", type=["pkl"])
+
+def is_valid_model(obj):
+    """
+    Check if the object is a class instance with a 'predict' method.
+    """
+    return hasattr(obj, 'get_team') and inspect.ismethod(getattr(obj, 'get_team', None)) or callable(getattr(obj, 'get_team', None))
 
 if uploaded_file and st.button("Submit"):
-    # Generate unique filename based on timestamp
     timestamp = datetime.datetime.now().isoformat()
     submission_filename = f"{timestamp.replace(':', '_')}_{uploaded_file.name}"
     submission_path = os.path.join("submissions", submission_filename)
-
-    # Ensure submissions folder exists
     os.makedirs("submissions", exist_ok=True)
 
     # Save uploaded file
     with open(submission_path, "wb") as f:
         f.write(uploaded_file.read())
 
-    # TODO: Add actual evaluation logic here
-    score = 42  # Dummy score
+    # Load and evaluate model
+    try:
+        with open(submission_path, "rb") as f:
+            model = pickle.load(f)
+
+        if is_valid_model(model):
+            score = 100  # Example valid score
+            st.success("Valid model uploaded. ✅")
+        else:
+            score = 0
+            st.error("Uploaded object does not implement a `predict` method. ❌")
+
+    except Exception as e:
+        score = 0
+        st.error(f"Failed to load or validate pickle file: {e}")
 
-    # Append to Google Sheet
+    # Append result to leaderboard
     append_score(timestamp, score, submission_filename)
-    st.success(f"Submission received! Score: {score}")
+    st.success(f"Submission received! Score: {score}; Team {team}")
 
 # ========================
 # Always Show Leaderboard