Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,64 +1,57 @@
-import panel as pn
+import gradio as gr
 import pandas as pd
 import os
 import datetime
-import io
 
 from google_sheet import fetch_leaderboard
 from google_drive import upload_to_drive
 
-pn.extension()
+os.makedirs("submissions", exist_ok=True)
 
-# File upload widget
-file_input = pn.widgets.FileInput(accept='.zip', multiple=False)
-
-# Status message
-status = pn.pane.Markdown("")
-
-# Leaderboard display
-leaderboard = pn.pane.DataFrame(pd.DataFrame(), width=600)
+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)
+        return df_sorted.to_html(index=False)
+    except Exception as e:
+        return f"<p>Could not load leaderboard: {e}</p>"
 
-def submit_file(event):
-    if file_input.value is None:
-        status.object = "⚠️ Please upload a .zip file before submitting."
-        return
+def handle_submission(file):
+    if file is None:
+        return "❌ No file uploaded.", "<p>No leaderboard to show.</p>"
 
-    # Save uploaded file
     timestamp = datetime.datetime.now().isoformat().replace(":", "_")
-    filename = f"{timestamp}_{file_input.filename}"
-    submission_path = os.path.join("submissions", filename)
-    os.makedirs("submissions", exist_ok=True)
+    submission_filename = f"{timestamp}_{file.name}"
+    submission_path = os.path.join("submissions", submission_filename)
+
     with open(submission_path, "wb") as f:
-        f.write(file_input.value)
+        f.write(file.read())
 
     try:
-        drive_file_id = upload_to_drive(submission_path, filename)
-        status.object = f"✅ Uploaded to Google Drive [File ID: {drive_file_id}]"
+        drive_file_id = upload_to_drive(submission_path, submission_filename)
+        status = f"✅ Uploaded to Google Drive [File ID: {drive_file_id}]"
     except Exception as e:
-        status.object = f"❌ Failed to upload to Google Drive: {e}"
+        status = f"⚠️ Failed to upload to Google Drive: {e}"
 
-    # Update leaderboard
-    try:
-        df = fetch_leaderboard()
-        if not df.empty:
-            df_sorted = df.sort_values(by="score", ascending=False)
-            leaderboard.object = df_sorted
-        else:
-            leaderboard.object = pd.DataFrame()
-    except Exception as e:
-        status.object += f"\n⚠️ Could not load leaderboard: {e}"
+    leaderboard_html = get_leaderboard_html()
+    return status, leaderboard_html
+
+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")
 
-submit_button = pn.widgets.Button(name="Submit", button_type="primary")
-submit_button.on_click(submit_file)
+    status_output = gr.Markdown()
+    leaderboard_output = gr.HTML(get_leaderboard_html())
 
-# Layout
-app = pn.Column(
-    "## 🏆 Hackathon Leaderboard",
-    file_input,
-    submit_button,
-    status,
-    "### Leaderboard",
-    leaderboard
-)
+    submit_btn.click(
+        fn=handle_submission,
+        inputs=file_input,
+        outputs=[status_output, leaderboard_output]
+    )
 
-app.servable()
+demo.launch(share=True)