Update app.py

app.py

@@ -7,61 +7,12 @@ import datetime
 DATA_DIR = "data"
 os.makedirs(DATA_DIR, exist_ok=True)
 
-# Paths
 DB_PATH = os.path.join(DATA_DIR, "teamup.db")
-
-# Initialize SQLite DB
-def init_db():
-    with sqlite3.connect(DB_PATH) as conn:
-        conn.execute("""
-            CREATE TABLE IF NOT EXISTS teamup (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                Name TEXT,
-                Discord TEXT UNIQUE,
-                City TEXT,
-                Country TEXT,
-                Address TEXT,
-                Looking TEXT,
-                Onlinecheck TEXT,
-                Languages TEXT,
-                Laptop TEXT,
-                Robot TEXT,
-                Skills TEXT,
-                Describe3 TEXT,
-                Experience TEXT,
-                Idea TEXT
-            );
-        """)
-    print("✅ SQLite database initialized.")
-init_db()
-
-# Initialize database
-conn = sqlite3.connect(DB_PATH)
-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", "")
+PAGE_SIZE = 20
 
-ALL_COUNTRIES = sorted(list(set([
+LANGUAGES = ["English", "French", "Spanish", "German", "Portuguese", "Chinese", "Arabic", "Hindi"]
+ALL_COUNTRIES = sorted(set([
     "United States of America", "United Kingdom", "India", "France", "Germany", "Canada", "Australia", "Japan", "Brazil", "Mexico",
     "Afghanistan", "Albania", "Algeria", "Andorra", "Angola", "Argentina", "Armenia", "Austria", "Azerbaijan",
     "Bangladesh", "Belgium", "Bhutan", "Bolivia", "Bosnia and Herzegovina", "Botswana", "Bulgaria", "Cambodia",
@@ -72,63 +23,89 @@ ALL_COUNTRIES = sorted(list(set([
     "Qatar", "Romania", "Russia", "Saudi Arabia", "Serbia", "Singapore", "Slovakia", "Slovenia", "South Africa", "South Korea",
     "Spain", "Sri Lanka", "Sweden", "Switzerland", "Thailand", "Tunisia", "Turkey", "Ukraine", "United Arab Emirates", "Vietnam",
     "Zambia", "Zimbabwe"
-])))
+]))
 
-LANGUAGES = ["English", "French", "Spanish", "German", "Portuguese", "Chinese", "Arabic", "Hindi"]
+def init_db():
+    with sqlite3.connect(DB_PATH) as conn:
+        conn.execute("""
+            CREATE TABLE IF NOT EXISTS teamup (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                name TEXT,
+                discord TEXT UNIQUE,
+                city TEXT,
+                country TEXT,
+                address TEXT,
+                looking TEXT,
+                onlinecheck TEXT,
+                languages TEXT,
+                laptop TEXT,
+                robot TEXT,
+                skills TEXT,
+                describe3 TEXT,
+                experience TEXT,
+                idea TEXT,
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+            );
+        """)
+init_db()
 
-# 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."
 
     lang_str = ", ".join(languages) if isinstance(languages, list) else languages
 
-    conn = sqlite3.connect(DB_PATH)
-    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!"
+    with sqlite3.connect(DB_PATH) as conn:
+        conn.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))
 
-# Filter and return HTML table
-def filter_by_fields(selected_country, selected_city, selected_language):
-    conn = sqlite3.connect(DB_PATH)
-    df = pd.read_sql_query("SELECT * FROM teamup", conn)
-    conn.close()
+    return "✅ Profile saved!"
 
-    if df.empty:
-        return "<p>No data available.</p>"
+def filter_by_fields(selected_country, selected_city, selected_language, page):
+    offset = (page - 1) * PAGE_SIZE
 
-    df["City"] = df["city"].str.title()
-    df["Country"] = df["country"].str.title()
-    df["Languages"] = df["languages"]
+    query = "SELECT * FROM teamup"
+    filters = []
+    params = []
 
     if selected_country != "All":
-        df = df[df["Country"] == selected_country.title()]
+        filters.append("country = ?")
+        params.append(selected_country.title())
     if selected_city != "All":
-        df = df[df["City"] == selected_city.title()]
+        filters.append("city = ?")
+        params.append(selected_city.title())
     if selected_language != "All":
-        df = df[df["Languages"].str.contains(selected_language.lower(), na=False)]
+        filters.append("languages LIKE ?")
+        params.append(f"%{selected_language.lower()}%")
+
+    if filters:
+        query += " WHERE " + " AND ".join(filters)
+    query += " ORDER BY created_at DESC LIMIT ? OFFSET ?"
+    params.extend([PAGE_SIZE, offset])
+
+    with sqlite3.connect(DB_PATH) as conn:
+        df = pd.read_sql_query(query, conn, params=params)
+        total_query = "SELECT COUNT(*) FROM teamup" + (" WHERE " + " AND ".join(filters) if filters else "")
+        total_count = pd.read_sql_query(total_query, conn, params=params[:-2] if filters else []).iloc[0, 0]
 
     if df.empty:
-        return "<p>No participants match your filters.</p>"
+        return "<p>No participants match your filters.</p>", page, total_count
 
     html = "<table style='width:100%; border-collapse: collapse;'><tr>"
     headers = ["Discord", "Name", "City", "Country", "Looking", "Onlinecheck", "Languages", "Laptop", "Robot", "Skills", "Describe3", "Experience", "Idea"]
@@ -138,7 +115,7 @@ def filter_by_fields(selected_country, selected_city, selected_language):
 
     for _, row in df.iterrows():
         html += "<tr>"
-        for col in ["discord", "name", "City", "Country", "looking", "onlinecheck", "Languages", "laptop", "robot", "skills", "describe3", "experience", "idea"]:
+        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>"
@@ -146,72 +123,94 @@ def filter_by_fields(selected_country, selected_city, selected_language):
         html += "</tr>"
 
     html += "</table>"
-    return html
+    return html, page, total_count
 
-# Update city dropdown
 def update_city_filter(country):
-    conn = sqlite3.connect(DB_PATH)
-    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()
+    with sqlite3.connect(DB_PATH) as conn:
+        if country == "All":
+            result = conn.execute("SELECT DISTINCT city FROM teamup").fetchall()
+        else:
+            result = conn.execute("SELECT DISTINCT city FROM teamup WHERE country = ?", (country.title(),)).fetchall()
+    cities = [r[0].title() for r in result if r[0]]
     return gr.update(choices=["All"] + sorted(cities), value="All")
 
-# Delete
 def delete_by_discord(discord, code):
     if code != ADMIN_CODE:
         return "❌ Invalid admin code."
-    conn = sqlite3.connect(DB_PATH)
-    c = conn.cursor()
-    c.execute("DELETE FROM teamup WHERE lower(discord) = ?", (discord.lower(),))
-    conn.commit()
-    conn.close()
+    with sqlite3.connect(DB_PATH) as conn:
+        conn.execute("DELETE FROM teamup WHERE lower(discord) = ?", (discord.lower(),))
     return f"🗑️ Deleted user with Discord: {discord}"
 
-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")
-
-    with gr.Row():
-        with gr.Column():
-            name = gr.Text(label="Name")
-            discord = gr.Text(label="👤 Discord Username *")
-            city = gr.Text(label="📍 City *")
-            country = gr.Dropdown(label="🌍 Country *", choices=ALL_COUNTRIES, value="France")
-            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=LANGUAGES, label="Languages Spoken *")
-            laptop = gr.Text(label="💻 Laptop Setup *")
-            robot = gr.Text(label="🤖 Robot Setup *")
-            skills = gr.Text(label="🧠 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 ✅")
-            status = gr.Textbox(label="", interactive=False)
-
-        with gr.Column():
-            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"] + LANGUAGES, value="All")
-            table_html = gr.HTML(label="Matching Participants")
-
-    country_filter.change(fn=update_city_filter, inputs=[country_filter], outputs=[city_filter])
-    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])
-    submit_btn.click(fn=filter_by_fields, inputs=[country_filter, city_filter, language_filter], outputs=[table_html])
-
-    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])
-
+def download_csv(code):
+    if code != ADMIN_CODE:
+        raise gr.Error("❌ Invalid admin code.")
+    with sqlite3.connect(DB_PATH) as conn:
+        df = pd.read_sql_query("SELECT * FROM teamup", conn)
+    csv_path = os.path.join(DATA_DIR, "teamup_export.csv")
+    df.to_csv(csv_path, index=False)
+    return csv_path
+
+def 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")
+
+        with gr.Row():
+            with gr.Column():
+                name = gr.Text(label="Name")
+                discord = gr.Text(label="👤 Discord Username *")
+                city = gr.Text(label="📍 City *")
+                country = gr.Dropdown(label="🌍 Country *", choices=ALL_COUNTRIES, value="France")
+                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=LANGUAGES, label="Languages Spoken *")
+                laptop = gr.Text(label="💻 Laptop Setup *")
+                robot = gr.Text(label="🤖 Robot Setup *")
+                skills = gr.Text(label="🧠 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 ✅")
+                status = gr.Textbox(label="", interactive=False)
+
+            with gr.Column():
+                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"] + LANGUAGES, value="All")
+                page_state = gr.Number(value=1, visible=False)
+                table_html = gr.HTML(label="Matching Participants")
+                prev_btn = gr.Button("⬅️ Previous Page")
+                next_btn = gr.Button("➡️ Next Page")
+
+        submit_btn.click(submit_profile, inputs=[name, discord, city, country, address, looking, onlinecheck, languages, laptop, robot, skills, describe3, experience, idea], outputs=[status])
+        submit_btn.click(filter_by_fields, inputs=[country_filter, city_filter, language_filter, page_state], outputs=[table_html, page_state, gr.Number(visible=False)])
+
+        country_filter.change(update_city_filter, inputs=[country_filter], outputs=[city_filter])
+
+        for dropdown in [country_filter, city_filter, language_filter]:
+            dropdown.change(lambda c, ci, l: (1,), inputs=[country_filter, city_filter, language_filter], outputs=[page_state])
+            dropdown.change(filter_by_fields, inputs=[country_filter, city_filter, language_filter, page_state], outputs=[table_html, page_state, gr.Number(visible=False)])
+
+        prev_btn.click(lambda p: max(p - 1, 1), inputs=[page_state], outputs=[page_state])
+        prev_btn.click(filter_by_fields, inputs=[country_filter, city_filter, language_filter, page_state], outputs=[table_html, page_state, gr.Number(visible=False)])
+
+        next_btn.click(lambda p: p + 1, inputs=[page_state], outputs=[page_state])
+        next_btn.click(filter_by_fields, inputs=[country_filter, city_filter, language_filter, page_state], outputs=[table_html, page_state, gr.Number(visible=False)])
+
+        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(download_csv, inputs=[export_code], outputs=[download_file])
+
+    return demo
+
+demo = interface()
 demo.launch()
-