src/streamlit_app.py

import gradio as gr
import pandas as pd
import os
import datetime

from google_sheet import fetch_leaderboard
from google_drive import upload_to_drive

# Ensure the submissions folder exists
os.makedirs("submissions", exist_ok=True)

# --- Submission Logic ---
def handle_submission(file):
    if file is None:
        return "❌ No file uploaded.", None

    # Generate timestamped filename
    timestamp = datetime.datetime.now().isoformat().replace(":", "_")
    submission_filename = f"{timestamp}_{file.name}"
    submission_path = os.path.join("submissions", submission_filename)

    # Save file
    with open(submission_path, "wb") as f:
        f.write(file.read())

    try:
        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 = f"⚠️ Failed to upload to Google Drive: {e}"

    # Return status and updated leaderboard
    return status, get_leaderboard_html()


# --- 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)
        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()