Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,48 +1,68 @@
-import streamlit as st
+import gradio as gr
 import pandas as pd
 import os
 import datetime
-import dill as pickle
-import inspect
 
-from google_sheet import *
+from google_sheet import fetch_leaderboard
 from google_drive import upload_to_drive
 
-st.title("🏆 Hackathon Leaderboard")
+# Ensure the submissions folder exists
+os.makedirs("submissions", exist_ok=True)
 
-# ========================
-# Submission Form
-# ========================
+# --- Submission Logic ---
+def handle_submission(file):
+    if file is None:
+        return "❌ No file uploaded.", None
 
-uploaded_file = st.file_uploader("Upload your submission (.zip)", type=["zip"], key="file-upload-1")
-
-if uploaded_file and st.button("Submit"):
-    timestamp = datetime.datetime.now().isoformat()
-    submission_filename = f"{timestamp.replace(':', '_')}_{uploaded_file.name}"
+    # Generate timestamped filename
+    timestamp = datetime.datetime.now().isoformat().replace(":", "_")
+    submission_filename = f"{timestamp}_{file.name}"
     submission_path = os.path.join("submissions", submission_filename)
-    os.makedirs("submissions", exist_ok=True)
 
-    # Save uploaded file
+    # Save file
     with open(submission_path, "wb") as f:
-        f.write(uploaded_file.read())
+        f.write(file.read())
 
     try:
         drive_file_id = upload_to_drive(submission_path, submission_filename)
-        st.success(f"Uploaded to Google Drive ✅ [File ID: {drive_file_id}]")
+        status = f"✅ Uploaded to Google Drive [File ID: {drive_file_id}]"
     except Exception as e:
-        st.warning(f"Failed to upload to Google Drive: {e}")
+        status = f"⚠️ Failed to upload to Google Drive: {e}"
+
+    # Return status and updated leaderboard
+    return status, get_leaderboard_html()
 
-# ========================
-# Always Show Leaderboard
-# ========================
-st.subheader("Leaderboard")
 
-try:
-    df = fetch_leaderboard()
-    if not df.empty:
+# --- Leaderboard Logic ---
+def get_leaderboard_html():
+    try:
+        df = fetch_leaderboard()
+        if df.empty:
+            return "<p>No submissions yet.</p>"
         df_sorted = df.sort_values(by="score", ascending=False)
-        st.dataframe(df_sorted)
-    else:
-        st.info("No submissions yet.")
-except Exception as e:
-    st.warning(f"Could not load leaderboard: {e}")
+        return df_sorted.to_html(index=False)
+    except Exception as e:
+        return f"<p>Could not load leaderboard: {e}</p>"
+
+
+# --- Gradio Interface ---
+with gr.Blocks(title="🏆 Hackathon Leaderboard") as demo:
+    gr.Markdown("## 🏆 Hackathon Leaderboard")
+
+    with gr.Row():
+        file_input = gr.File(label="Upload your submission (.zip)", file_types=[".zip"])
+        submit_btn = gr.Button("Submit")
+
+    status_output = gr.Markdown()
+    leaderboard_output = gr.HTML(get_leaderboard_html())
+
+    def submit_action(file):
+        return handle_submission(file)
+
+    submit_btn.click(
+        fn=submit_action,
+        inputs=file_input,
+        outputs=[status_output, leaderboard_output]
+    )
+
+demo.launch()