app.py

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

# Path to the data file
DATA_FILE = os.path.join("data", "teamup_data.json")

# Load the ADMIN_CODE environment variable
ADMIN_CODE = os.getenv("ADMIN_CODE", "")

# Ensure data file exists
os.makedirs("data", exist_ok=True)
if not os.path.exists(DATA_FILE) or os.path.getsize(DATA_FILE) == 0:
    with open(DATA_FILE, "w") as f:
        json.dump([], f)

# Function to create a backup of the data
def backup_data():
    source_file = DATA_FILE
    backup_dir = './data/backup'
    os.makedirs(backup_dir, exist_ok=True)

    timestamp = datetime.datetime.now().strftime('%Y%m%d_%H%M%S')
    backup_file = os.path.join(backup_dir, f'teamup_data_backup_{timestamp}.json')

    shutil.copy(source_file, backup_file)
    print(f"✅ Backup created at {backup_file}")

# Function to submit or update a profile
def submit_profile(name, discord, city, country, address, looking, onlinecheck, languages, laptop, robot, skills, describe3, experience, idea):
    print("🟢 Submit button clicked.")
    print(f"Incoming: {discord=}, {city=}, {country=}, {languages=}, {laptop=}, {robot=}")

    if not discord or not city or not country or not laptop or not robot:
        return "❌ Please fill in all required fields."
    if not languages or not isinstance(languages, list) or len(languages) == 0:
        return "❌ Please select at least one language."

    # Ensure country is stored as a string (in case form allows lists)
    if isinstance(country, list):
        country = country[0] if country else ""

    with open(DATA_FILE, "r") as f:
        data = json.load(f)

    for d in data:
        if d["Discord"].lower() == discord.lower():
            d.update({
                "Name": name,
                "City": city,
                "Country": country,
                "Address": address,
                "Looking for Team": looking,
                "Onlinecheck": onlinecheck,
                "Languages": languages,
                "Laptop": laptop,
                "Robot": robot,
                "Skills": skills,
                "Describe3": describe3,
                "Experience": experience,
                "Project Idea": idea
            })
            break
    else:
        data.append({
            "Name": name,
            "Discord": discord,
            "City": city,
            "Country": country,
            "Address": address,
            "Looking for Team": looking,
            "Onlinecheck": onlinecheck,
            "Languages": languages,
            "Laptop": laptop,
            "Robot": robot,
            "Skills": skills,
            "Describe3": describe3,
            "Experience": experience,
            "Project Idea": idea
        })

    try:
        with open(DATA_FILE, "w") as f:
            json.dump(data, f, indent=2)
        print(f"✅ Successfully wrote {len(data)} profiles to {DATA_FILE}")
        backup_data()  # Backup after every update
    except Exception as e:
        print(f"❌ Failed to write to {DATA_FILE}: {e}")
        return "❌ Error saving your profile. Please try again."

    return "✅ Profile saved!"

# Function to filter participants by fields
def filter_by_fields(selected_country, selected_city, selected_language):
    with open(DATA_FILE, "r") as f:
        data = json.load(f)
    df = pd.DataFrame(data)

    # Convert list values to readable strings
    df["Languages"] = df["Languages"].apply(lambda langs: ", ".join(langs) if isinstance(langs, list) else langs)

    if df.empty or "Country" not in df.columns or "Languages" not in df.columns:
        return "<p>No participants found.</p>"

    if selected_country != "All":
        df = df[df["Country"] == selected_country]

    if selected_city != "All":
        df = df[df["City"] == selected_city]

    if selected_language != "All":
        df = df[df["Languages"].apply(lambda langs: selected_language in langs)]

    # Hide address
    if "Address" in df.columns:
        df = df.drop(columns=["Address"])

    # Rename column headers for display only
    display_names = {
        "Discord": "Discord",
        "Name": "Name",
        "City": "City",
        "Country": "Country",
        "Looking for Team": "Looking for Team",
        "Onlinecheck": "How?",
        "Languages": "Languages",
        "Laptop": "Laptop",
        "Robot": "Robot",
        "Skills": "Skills",
        "Describe3": "Describe",
        "Experience": "Experience",
        "Project Idea": "Project Idea"
    }

    html = '<h3 style="margin-top:20px;">🔍 Find your matching Teammates & Register your team <a href="https://forms.gle/gJEMGD4CEA2emhD18" target="_blank">here</a>👈👈</h3>'
    html += "<table style='width:100%; border-collapse: collapse;'>"

    # Header row
    html += "<tr>" + "".join(
        f"<th style='border: 1px solid #ccc; padding: 6px;'>{display_names.get(col, col)}</th>"
        for col in df.columns
    ) + "</tr>"

    # Data rows
    for _, row in df.iterrows():
        html += "<tr>"
        for col in df.columns:
            val = row[col]
            if col == "Discord":
                val = f"<a href='https://discord.com/users/{val}' target='_blank'>{val}</a>"
            html += f"<td style='border: 1px solid #eee; padding: 6px;'>{val}</td>"
        html += "</tr>"

    html += "</table>"
    return html

# Function to update the dropdown choices
def update_dropdown_choices():
    with open(DATA_FILE, "r") as f:
        data = json.load(f)
    df = pd.DataFrame(data)

    country_choices = sorted(df["Country"].dropna().unique()) if "Country" in df else []
    city_choices = sorted(df["City"].dropna().unique()) if "City" in df else []
    language_set = set()
    if "Languages" in df:
        for lang_list in df["Languages"].dropna():
            if isinstance(lang_list, list):
                language_set.update(lang_list)
            elif isinstance(lang_list, str):
                language_set.update(lang_list.split(", "))
    return (
        gr.update(choices=["All"] + list(country_choices), value="All"),
        gr.update(choices=["All"] + list(city_choices), value="All"),
        gr.update(choices=["All"] + sorted(language_set), value="All")
    )

# Setup Gradio interface
with gr.Blocks() as demo:
    gr.Markdown("# 🌍 LeRobot Worldwide Hackathon - Team-Up Dashboard")
    gr.Markdown("1. Submit or update your profile to find matching teammates and contact them on Discord. (Required fields marked with *.) ")

    # Input fields and buttons setup
    with gr.Row():
        with gr.Column():
            name = gr.Text(label="Name")
            discord = gr.Text(label="👤 Discord Username *")
            city = gr.Text(label="📍 City *")
            country = gr.Text(label="🌍 Country *")
            address = gr.Text(label="Address (optional)")
            looking = gr.Radio(["Yes", "No"], label="🔍 Looking for a team?")
            onlinecheck = gr.Radio(["Participate Online", "Join a Local Hackathon"], label="🚀 I will...")            
            languages = gr.CheckboxGroup(choices=["English", "French", "Spanish", "German", "Portuguese", "Chinese", "Arabic", "Hindi"], label="Languages Spoken *")
            laptop = gr.Text(label="💻 Laptop Setup *")
            robot = gr.Text(label="Robot Setup *")
            skills = gr.Text(label="🧠 Your Skills (comma separated)")
            describe3 = gr.Text(label="🤗 3 Words That Describe You")
            experience = gr.Dropdown(choices=["Beginner", "Intermediate", "Advanced"], label="Experience Level", value="Beginner")
            idea = gr.Textbox(label="Project Idea (optional)")
            submit_btn = gr.Button("Submit or Update Profile ✅")
            status = gr.Textbox(label="", interactive=False)

        with gr.Column():
            gr.Markdown("🎯 2. Choose your preferences to find your teammates (country, city or language)")
            country_filter = gr.Dropdown(label="Filter by Country", choices=["All"], value="All", allow_custom_value=False)
            city_filter = gr.Dropdown(label="Filter by City", choices=["All"], value="All", allow_custom_value=False)
            language_filter = gr.Dropdown(label="Filter by Language", choices=["All"], value="All", allow_custom_value=False)
            table_html = gr.HTML(label="Matching Participants")

    submit_btn.click(
        submit_profile,
        inputs=[name, discord, city, country, address, looking, onlinecheck, languages, laptop, robot, skills, describe3, experience, idea],
        outputs=[status]
    )

    country_filter.change(fn=filter_by_fields, inputs=[country_filter, city_filter, language_filter], outputs=table_html)
    city_filter.change(fn=filter_by_fields, inputs=[country_filter, city_filter, language_filter], outputs=table_html)
    language_filter.change(fn=filter_by_fields, inputs=[country_filter, city_filter, language_filter], outputs=table_html)

demo.launch()