Update app.py

app.py

@@ -1,27 +1,41 @@
 import gradio as gr
-import json
 import pandas as pd
+import sqlite3
 import os
-import shutil
 import datetime
 
 # Paths
-DATA_DIR = "data"
-DATA_FILE = os.path.join(DATA_DIR, "teamup_data.json")
-BACKUP_DIR = os.path.join(DATA_DIR, "backup")
-os.makedirs(DATA_DIR, exist_ok=True)
+DB_FILE = "data/teamup.db"
+os.makedirs("data", exist_ok=True)
+
+# Initialize database
+conn = sqlite3.connect(DB_FILE)
+c = conn.cursor()
+c.execute('''
+CREATE TABLE IF NOT EXISTS teamup (
+    discord TEXT PRIMARY KEY,
+    name TEXT,
+    city TEXT,
+    country TEXT,
+    address TEXT,
+    looking TEXT,
+    onlinecheck TEXT,
+    languages TEXT,
+    laptop TEXT,
+    robot TEXT,
+    skills TEXT,
+    describe3 TEXT,
+    experience TEXT,
+    idea TEXT
+)
+''')
+conn.commit()
+conn.close()
 
 ADMIN_CODE = os.getenv("ADMIN_CODE", "")
 
-# Init data file
-if not os.path.exists(DATA_FILE) or os.path.getsize(DATA_FILE) == 0:
-    with open(DATA_FILE, "w") as f:
-        json.dump([], f)
-
-# Country list
 ALL_COUNTRIES = sorted(list(set([
     "United States of America", "United Kingdom", "India", "France", "Germany", "Canada", "Australia", "Japan", "Brazil", "Mexico",
-] + [  # Full list
     "Afghanistan", "Albania", "Algeria", "Andorra", "Angola", "Argentina", "Armenia", "Austria", "Azerbaijan",
     "Bangladesh", "Belgium", "Bhutan", "Bolivia", "Bosnia and Herzegovina", "Botswana", "Bulgaria", "Cambodia",
     "Chile", "China", "Colombia", "Croatia", "Cuba", "Czech Republic", "Denmark", "Dominican Republic", "Ecuador",
@@ -33,178 +47,103 @@ ALL_COUNTRIES = sorted(list(set([
     "Zambia", "Zimbabwe"
 ])))
 
-# Backup
-
-def backup_data():
-    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(DATA_FILE, backup_file)
-
-# Save data safely
-
-def atomic_save(data, path):
-    tmp_path = path + ".tmp"
-    with open(tmp_path, "w") as f:
-        json.dump(data, f, indent=2)
-    os.replace(tmp_path, path)
-
-# Submit profile
+LANGUAGES = ["English", "French", "Spanish", "German", "Portuguese", "Chinese", "Arabic", "Hindi"]
 
+# Insert or update profile
 def submit_profile(name, discord, city, country, address, looking, onlinecheck, languages, laptop, robot, skills, describe3, experience, idea):
     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):
-        return "❌ Please select at least one language."
-
-    city = city.strip().title()
-    country = country.strip().title()
-    discord = discord.strip()
-
-    with open(DATA_FILE, "r") as f:
-        data = json.load(f)
-
-    for entry in data:
-        if entry["Discord"].lower() == discord.lower():
-            entry.update({
-                "Name": name,
-                "City": city,
-                "Country": country,
-                "Address": address,
-                "Looking for Team": looking,
-                "Onlinecheck": onlinecheck,
-                "Languages": languages if isinstance(languages, list) else [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 if isinstance(languages, list) else [languages],
-            "Laptop": laptop,
-            "Robot": robot,
-            "Skills": skills,
-            "Describe3": describe3,
-            "Experience": experience,
-            "Project Idea": idea
-        })
-
-    try:
-        atomic_save(data, DATA_FILE)
-        backup_data()
-        return "✅ Profile saved!"
-    except Exception as e:
-        return f"❌ Failed to save: {e}"
-
-# Filter dropdowns
-
-def update_city_filter(country):
-    with open(DATA_FILE, "r") as f:
-        data = json.load(f)
-    df = pd.DataFrame(data)
-
-    if country != "All":
-        df = df[df["Country"].str.title() == country.title()]
-
-    cities = sorted(set(df["City"].dropna().unique()))
-    return gr.update(choices=["All"] + cities, value="All")
-
-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")
-    )
-
-# Filter table
-
+    lang_str = ", ".join(languages) if isinstance(languages, list) else languages
+
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("""
+        INSERT INTO teamup (discord, name, city, country, address, looking, onlinecheck, languages, laptop, robot, skills, describe3, experience, idea)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        ON CONFLICT(discord) DO UPDATE SET
+            name=excluded.name,
+            city=excluded.city,
+            country=excluded.country,
+            address=excluded.address,
+            looking=excluded.looking,
+            onlinecheck=excluded.onlinecheck,
+            languages=excluded.languages,
+            laptop=excluded.laptop,
+            robot=excluded.robot,
+            skills=excluded.skills,
+            describe3=excluded.describe3,
+            experience=excluded.experience,
+            idea=excluded.idea
+    """, (discord, name, city.title(), country.title(), address, looking, onlinecheck, lang_str.lower(), laptop, robot, skills, describe3, experience, idea))
+    conn.commit()
+    conn.close()
+    return "✅ Profile saved!"
+
+# Filter and return HTML table
 def filter_by_fields(selected_country, selected_city, selected_language):
-    with open(DATA_FILE, "r") as f:
-        data = json.load(f)
+    conn = sqlite3.connect(DB_FILE)
+    df = pd.read_sql_query("SELECT * FROM teamup", conn)
+    conn.close()
 
-    if not data:
+    if df.empty:
         return "<p>No data available.</p>"
 
-    df = pd.DataFrame(data)
-
-    df["Country"] = df["Country"].astype(str).str.strip().str.title()
-    df["City"] = df["City"].astype(str).str.strip().str.title()
-    df["Languages"] = df["Languages"].apply(lambda x: ", ".join(x) if isinstance(x, list) else str(x)).str.strip().str.lower()
+    df["City"] = df["city"].str.title()
+    df["Country"] = df["country"].str.title()
+    df["Languages"] = df["languages"]
 
     if selected_country != "All":
-        df = df[df["Country"] == selected_country.strip().title()]
+        df = df[df["Country"] == selected_country.title()]
     if selected_city != "All":
-        df = df[df["City"] == selected_city.strip().title()]
+        df = df[df["City"] == selected_city.title()]
     if selected_language != "All":
-        df = df[df["Languages"].str.contains(selected_language.strip().lower(), na=False)]
+        df = df[df["Languages"].str.contains(selected_language.lower(), na=False)]
 
     if df.empty:
         return "<p>No participants match your filters.</p>"
 
-    if "Address" in df.columns:
-        df = df.drop(columns=["Address"])
+    html = "<table style='width:100%; border-collapse: collapse;'><tr>"
+    headers = ["Discord", "Name", "City", "Country", "Looking", "Onlinecheck", "Languages", "Laptop", "Robot", "Skills", "Describe3", "Experience", "Idea"]
+    for col in headers:
+        html += f"<th style='border:1px solid #ccc; padding:6px;'>{col}</th>"
+    html += "</tr>"
 
-    html = '<h3 style="margin-top:20px;">\U0001f50d 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;'>"
-    html += "<tr>" + "".join(f"<th style='border: 1px solid #ccc; padding: 6px;'>{col}</th>" for col in df.columns) + "</tr>"
     for _, row in df.iterrows():
-        html += "<tr>" + "".join(
-            f"<td style='border: 1px solid #eee; padding: 6px;'>" +
-            (f"<a href='https://discord.com/users/{val}' target='_blank'>{val}</a>" if col == "Discord" else str(val)) +
-            "</td>" for col, val in row.items()) + "</tr>"
+        html += "<tr>"
+        for col in ["discord", "name", "City", "Country", "looking", "onlinecheck", "Languages", "laptop", "robot", "skills", "describe3", "experience", "idea"]:
+            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
 
-# Delete
+# Update city dropdown
+def update_city_filter(country):
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    if country == "All":
+        c.execute("SELECT DISTINCT city FROM teamup")
+    else:
+        c.execute("SELECT DISTINCT city FROM teamup WHERE country = ?", (country.title(),))
+    cities = [r[0].title() for r in c.fetchall() if r[0]]
+    conn.close()
+    return gr.update(choices=["All"] + sorted(cities), value="All")
 
+# Delete
 def delete_by_discord(discord, code):
     if code != ADMIN_CODE:
         return "❌ Invalid admin code."
-    with open(DATA_FILE, "r") as f:
-        data = json.load(f)
-    new_data = [d for d in data if d["Discord"].lower() != discord.lower()]
-    with open(DATA_FILE, "w") as f:
-        json.dump(new_data, f, indent=2)
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("DELETE FROM teamup WHERE lower(discord) = ?", (discord.lower(),))
+    conn.commit()
+    conn.close()
     return f"🗑️ Deleted user with Discord: {discord}"
 
-# CSV export
-
-def download_csv(code):
-    if code != ADMIN_CODE:
-        raise gr.Error("❌ Invalid admin code.")
-    with open(DATA_FILE, "r") as f:
-        data = json.load(f)
-    df = pd.DataFrame(data)
-    csv_path = os.path.join("data", "teamup_export.csv")
-    df.to_csv(csv_path, index=False)
-    return csv_path
-
-# Interface
 with gr.Blocks(css=".gr-dropdown { max-height: 100px; overflow-y: auto; font-size: 12px; }") as demo:
     gr.Markdown("# 🌍 LeRobot Worldwide Hackathon - Team-Up Dashboard")
 
@@ -217,7 +156,7 @@ with gr.Blocks(css=".gr-dropdown { max-height: 100px; overflow-y: auto; font-siz
             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 *")
+            languages = gr.CheckboxGroup(choices=LANGUAGES, label="Languages Spoken *")
             laptop = gr.Text(label="💻 Laptop Setup *")
             robot = gr.Text(label="🤖 Robot Setup *")
             skills = gr.Text(label="🧠 Skills (comma separated)")
@@ -228,47 +167,23 @@ with gr.Blocks(css=".gr-dropdown { max-height: 100px; overflow-y: auto; font-siz
             status = gr.Textbox(label="", interactive=False)
 
         with gr.Column():
-            gr.Markdown("\n\n🎯 2. Choose your preferences to find your teammates")
             country_filter = gr.Dropdown(label="Filter by Country", choices=["All"] + ALL_COUNTRIES, value="All")
             city_filter = gr.Dropdown(label="Filter by City", choices=["All"], value="All")
-            language_filter = gr.Dropdown(label="Filter by Language", choices=["All", "English", "French", "Spanish", "German", "Portuguese", "Chinese", "Arabic", "Hindi"], value="All")
+            language_filter = gr.Dropdown(label="Filter by Language", choices=["All"] + LANGUAGES, value="All")
             table_html = gr.HTML(label="Matching Participants")
 
-    demo.load(fn=filter_by_fields, inputs=[country_filter, city_filter, language_filter], outputs=[table_html])
-    demo.load(fn=update_dropdown_choices, outputs=[country_filter, city_filter, language_filter])
-
     country_filter.change(fn=update_city_filter, inputs=[country_filter], outputs=[city_filter])
-    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])
+    for dropdown in [country_filter, city_filter, language_filter]:
+        dropdown.change(fn=filter_by_fields, inputs=[country_filter, city_filter, language_filter], outputs=[table_html])
 
-    submit_btn.click(
-        fn=submit_profile,
-        inputs=[name, discord, city, country, address, looking, onlinecheck, languages, laptop, robot, skills, describe3, experience, idea],
-        outputs=[status]
-    ).then(
-        fn=update_dropdown_choices,
-        inputs=[],
-        outputs=[country_filter, city_filter, language_filter]
-    ).then(
-        fn=filter_by_fields,
-        inputs=[country_filter, city_filter, language_filter],
-        outputs=[table_html]
-    )
+    submit_btn.click(fn=submit_profile, inputs=[name, discord, city, country, address, looking, onlinecheck, languages, laptop, robot, skills, describe3, experience, idea], outputs=[status])
+    submit_btn.click(fn=filter_by_fields, inputs=[country_filter, city_filter, language_filter], outputs=[table_html])
 
-    gr.Markdown("---\n### 🛡️ Admin Panel (delete by Discord)")
+    gr.Markdown("---\n### 🛡️ Admin Panel")
     admin_discord = gr.Text(label="Discord Username")
     admin_code = gr.Text(label="Admin Code", type="password")
     del_btn = gr.Button("Delete Profile")
     del_status = gr.Textbox(label="Status", interactive=False)
     del_btn.click(delete_by_discord, inputs=[admin_discord, admin_code], outputs=[del_status])
 
-    gr.Markdown("---\n### 📅 Admin Export CSV")
-    export_code = gr.Text(label="Admin Code", type="password")
-    download_btn = gr.Button("Generate and Download CSV")
-    download_file = gr.File(label="CSV Export", interactive=False)
-    download_btn.click(fn=download_csv, inputs=[export_code], outputs=[download_file])
-
-if __name__ == "__main__":
-    demo.launch()
-
+demo.launch()