Update app.py

app.py

@@ -1,90 +1,81 @@
-import io
-import random
-from typing import List, Tuple
-
-import aiohttp
-import panel as pn
-from PIL import Image
-from transformers import CLIPModel, CLIPProcessor
-
-pn.extension(design="bootstrap", sizing_mode="stretch_width")
+import os
+import tempfile
 import panel as pn
 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()
+pn.extension(design="bootstrap", sizing_mode="stretch_width")
 
-# File upload widget
+# Widgets
 file_input = pn.widgets.FileInput(accept='.zip', multiple=False)
+submit_button = pn.widgets.Button(name="📤 Submit", button_type="primary")
+status = pn.pane.Alert("", alert_type="info", visible=False)
+leaderboard = pn.pane.DataFrame(pd.DataFrame(), width=800, height=400, index=False)
 
-# Status message
-status = pn.pane.Markdown("")
-
-# Leaderboard display
-leaderboard = pn.pane.DataFrame(pd.DataFrame(), width=600)
-import tempfile
-
+# Temp dir
 temp_dir = tempfile.gettempdir()
 
+# --- Submission logic ---
 def submit_file(event):
     if file_input.value is None:
         status.object = "⚠️ Please upload a .zip file before submitting."
+        status.alert_type = "warning"
+        status.visible = True
         return
 
     try:
-        # Extract filename
         filename = file_input.filename
         submission_path = os.path.join(temp_dir, filename)
 
-        # Save uploaded content to temp file
         with open(submission_path, "wb") as f:
             f.write(file_input.value)
 
-        # Upload to Google Drive
         drive_file_id = upload_to_drive(submission_path, filename)
-        status.object = f"✅ Uploaded to Google Drive [File ID: {drive_file_id}]"
+        status.object = f"✅ Uploaded to Google Drive\n**File ID**: `{drive_file_id}`"
+        status.alert_type = "success"
+        status.visible = True
     except Exception as e:
-        status.object = f"❌ Failed to upload to Google Drive: {e}"
-        return  # Exit early on failure
+        status.object = f"❌ Failed to upload: {e}"
+        status.alert_type = "danger"
+        status.visible = True
+        return
+
+    # Refresh leaderboard
+    update_leaderboard()
+
+submit_button.on_click(submit_file)
 
-    # Update leaderboard
+# --- Leaderboard logic ---
+def update_leaderboard():
     try:
         df = fetch_leaderboard()
         if not df.empty:
-            df_sorted = df.sort_values(by="score", ascending=False)
-            leaderboard.object = df_sorted
+            leaderboard.object = df.sort_values(by="score", ascending=False)
         else:
             leaderboard.object = pd.DataFrame()
     except Exception as e:
-        status.object += f"\n⚠️ Could not load leaderboard: {e}"
-
-submit_button = pn.widgets.Button(name="Submit", button_type="primary")
-submit_button.on_click(submit_file)
-
-# Initial leaderboard load
-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 on startup: {e}"
-
-# Layout
-app = pn.Column(
-    "## 🏆 Hackathon Leaderboard",
-    file_input,
-    submit_button,
-    status,
-    "### Leaderboard",
-    leaderboard
+        status.object = f"⚠️ Could not load leaderboard: {e}"
+        status.alert_type = "warning"
+        status.visible = True
+
+# Initial load
+update_leaderboard()
+
+# --- Layout ---
+app = pn.template.BootstrapTemplate(
+    title="🏆 Hackathon Leaderboard",
+    sidebar=[
+        pn.pane.Markdown("### 📁 Upload Your Submission (.zip)"),
+        file_input,
+        submit_button,
+        status,
+    ],
+    main=[
+        pn.pane.Markdown("## 🥇 Live Leaderboard"),
+        leaderboard
+    ]
 )
 
 app.servable()