Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,6 +6,7 @@ import dill as pickle
 import inspect
 
 from google_sheet import *
+from google_drive import upload_to_drive
 
 st.title("🏆 Hackathon Leaderboard")
 
@@ -13,7 +14,7 @@ st.title("🏆 Hackathon Leaderboard")
 # Submission Form
 # ========================
 
-uploaded_file = st.file_uploader("Upload your model/class (.pkl)", type=["pkl"])
+uploaded_file = st.file_uploader("Upload your submission (.zip)", type=["zip"])
 
 def is_valid_model(obj):
     """
@@ -31,29 +32,11 @@ if uploaded_file and st.button("Submit"):
     with open(submission_path, "wb") as f:
         f.write(uploaded_file.read())
 
-    team = "fail"
-
-    # Load and evaluate model
     try:
-        with open(submission_path, "rb") as f:
-            model = pickle.load(f)
-
-        if is_valid_model(model):
-            team = model.get_team()
-            score = 100  # Example valid score
-            st.success("Valid model uploaded. ✅")
-        else:
-            team = "fail"
-            score = 0
-            st.error("Uploaded object does not implement a `predict` method. ❌")
-
+        drive_file_id = upload_to_drive(submission_path, submission_filename)
+        st.success(f"Uploaded to Google Drive ✅ [File ID: {drive_file_id}]")
     except Exception as e:
-        score = 0
-        st.error(f"Failed to load or validate pickle file: {e}")
-
-    # Append result to leaderboard
-    append_score(timestamp, score, submission_filename)
-    st.success(f"Submission received! Score: {score}; Team {team}")
+        st.warning(f"Failed to upload to Google Drive: {e}")
 
 # ========================
 # Always Show Leaderboard