Spaces:

alphabagibagi
/

iceboxai-admin

Running

App Files Files Community

alphabagibagi commited on Mar 3

Commit

807e987

verified ·

1 Parent(s): 834c5b2

Upload 3 files

Browse files

Files changed (3) hide show

admin.py +455 -0
requirements.txt +6 -0
start_admin.sh +3 -0

admin.py ADDED Viewed

	@@ -0,0 +1,455 @@

+import os
+import asyncio
+from datetime import datetime, timezone, timedelta
+from flask import Flask, render_template_string, request, redirect, url_for, flash, session, jsonify
+from functools import wraps
+from supabase import create_client, Client
+from dotenv import load_dotenv
+import requests
+# Load environment variables
+load_dotenv()
+# Configuration
+SUPABASE_URL = os.getenv("SUPABASE_URL")
+SUPABASE_KEY = os.getenv("SUPABASE_KEY")
+ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD", "admin123")
+FLASK_SECRET_KEY = os.getenv("FLASK_SECRET_KEY", "icebox-admin-secret-777")
+# Initialize Supabase Client
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+app = Flask(__name__)
+app.secret_key = FLASK_SECRET_KEY
+# Authentication Decorator
+def login_required(f):
+    @wraps(f)
+    def decorated_function(*args, **kwargs):
+        if 'logged_in' not in session:
+            return redirect(url_for('login'))
+        return f(*args, **kwargs)
+    return decorated_function
+# Templates
+LOGIN_TEMPLATE = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Admin Login | Icebox AI</title>
+    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/css/bootstrap.min.css" rel="stylesheet">
+    <style>
+        :root {
+            --primary-bg: #0f172a;
+            --accent: #6366f1;
+            --glass: rgba(255, 255, 255, 0.05);
+        }
+        body {
+            background-color: var(--primary-bg);
+            color: white;
+            height: 100vh;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            font-family: 'Inter', sans-serif;
+        }
+        .login-card {
+            background: var(--glass);
+            backdrop-filter: blur(12px);
+            border: 1px solid rgba(255, 255, 255, 0.1);
+            border-radius: 20px;
+            padding: 40px;
+            width: 100%;
+            max-width: 400px;
+            box-shadow: 0 8px 32px 0 rgba(0, 0, 0, 0.37);
+        }
+        .form-control {
+            background: rgba(255, 255, 255, 0.05);
+            border: 1px solid rgba(255, 255, 255, 0.1);
+            color: white;
+            border-radius: 10px;
+            padding: 12px;
+        }
+        .form-control:focus {
+            background: rgba(255, 255, 255, 0.1);
+            border-color: var(--accent);
+            color: white;
+            box-shadow: none;
+        }
+        .btn-primary {
+            background: var(--accent);
+            border: none;
+            border-radius: 10px;
+            padding: 12px;
+            font-weight: 600;
+        }
+        .logo {
+            text-align: center;
+            margin-bottom: 30px;
+        }
+        .logo h1 {
+            font-size: 24px;
+            font-weight: 800;
+            background: linear-gradient(to right, #818cf8, #c084fc);
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+        }
+    </style>
+</head>
+<body>
+    <div class="login-card">
+        <div class="logo">
+            <h1>ICEBOX AI ADMIN</h1>
+            <p class="text-secondary">Control Center</p>
+        </div>
+        {% with messages = get_flashed_messages() %}
+          {% if messages %}
+            {% for message in messages %}
+              <div class="alert alert-danger py-2" role="alert" style="background: rgba(220, 53, 69, 0.2); border: none; color: #ff8787;">
+                {{ message }}
+              </div>
+            {% endfor %}
+          {% endif %}
+        {% endwith %}
+        <form method="POST">
+            <div class="mb-4">
+                <label class="form-label text-secondary small">ACCESS TOKEN</label>
+                <input type="password" name="password" class="form-control" placeholder="••••••••" required>
+            </div>
+            <button type="submit" class="btn btn-primary w-100">AUTHENTICATE</button>
+        </form>
+    </div>
+</body>
+</html>
+"""
+INDEX_TEMPLATE = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Dashboard | Icebox AI Admin</title>
+    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/css/bootstrap.min.css" rel="stylesheet">
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.11.1/font/bootstrap-icons.css">
+    <script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
+    <style>
+        :root {
+            --bg: #0b0f19;
+            --sidebar: #111827;
+            --card: #1f2937;
+            --accent: #6366f1;
+            --text-muted: #9ca3af;
+        }
+        body {
+            background-color: var(--bg);
+            color: #f3f4f6;
+            font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+            overflow-x: hidden;
+        }
+        .sidebar {
+            width: 250px;
+            background-color: var(--sidebar);
+            height: 100vh;
+            position: fixed;
+            padding: 20px;
+            border-right: 1px solid rgba(255, 255, 255, 0.05);
+        }
+        .main-content {
+            margin-left: 250px;
+            padding: 30px;
+        }
+        .nav-link {
+            color: var(--text-muted);
+            padding: 12px 15px;
+            border-radius: 10px;
+            margin-bottom: 5px;
+            transition: 0.3s;
+        }
+        .nav-link:hover, .nav-link.active {
+            color: white;
+            background: rgba(99, 102, 241, 0.1);
+        }
+        .nav-link i {
+            margin-right: 10px;
+        }
+        .stat-card {
+            background: var(--card);
+            border-radius: 15px;
+            padding: 24px;
+            border: 1px solid rgba(255, 255, 255, 0.05);
+            height: 100%;
+        }
+        .stat-icon {
+            width: 48px;
+            height: 48px;
+            border-radius: 12px;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            font-size: 24px;
+            margin-bottom: 15px;
+        }
+        .table-container {
+            background: var(--card);
+            border-radius: 15px;
+            padding: 20px;
+            border: 1px solid rgba(255, 255, 255, 0.05);
+            margin-top: 25px;
+        }
+        .table {
+            color: #e5e7eb;
+            vertical-align: middle;
+        }
+        .table thead th {
+            color: var(--text-muted);
+            font-weight: 500;
+            text-transform: uppercase;
+            font-size: 11px;
+            letter-spacing: 0.05em;
+            border-bottom: 1px solid rgba(255, 255, 255, 0.05);
+        }
+        .table td {
+            border-bottom: 1px solid rgba(255, 255, 255, 0.05);
+            padding: 15px 10px;
+        }
+        .badge-premium { background: rgba(168, 85, 247, 0.2); color: #c084fc; border: 1px solid rgba(168, 85, 247, 0.3); }
+        .badge-free { background: rgba(59, 130, 246, 0.2); color: #60a5fa; border: 1px solid rgba(59, 130, 246, 0.3); }
+        .search-bar {
+            background: rgba(255, 255, 255, 0.05);
+            border: 1px solid rgba(255, 255, 255, 0.1);
+            border-radius: 10px;
+            color: white;
+            padding: 10px 15px;
+        }
+    </style>
+</head>
+<body>
+    <div class="sidebar">
+        <h4 class="fw-bold mb-4 px-3" style="color: var(--accent);">Icebox Admin</h4>
+        <nav class="nav flex-column">
+            <a class="nav-link active" href="/"><i class="bi bi-grid-1x2"></i> Overview</a>
+            <a class="nav-link" href="#users"><i class="bi bi-people"></i> Users</a>
+            <a class="nav-link" href="#generations"><i class="bi bi-robot"></i> Generations</a>
+            <hr class="opacity-10 my-4">
+            <a class="nav-link text-danger" href="/logout"><i class="bi bi-box-arrow-left"></i> Logout</a>
+        </nav>
+    </div>
+    <div class="main-content">
+        <div class="d-flex justify-content-between align-items-center mb-5">
+            <div>
+                <h2 class="fw-bold mb-0">Platform Analytics</h2>
+                <p class="text-secondary small">Real-time update as of {{ now }}</p>
+            </div>
+            <div class="d-flex gap-3">
+                <input type="text" class="search-bar" placeholder="Search users...">
+                <button class="btn btn-primary px-4"><i class="bi bi-arrow-clockwise"></i></button>
+            </div>
+        </div>
+        <div class="row g-4">
+            <div class="col-md-3">
+                <div class="stat-card">
+                    <div class="stat-icon" style="background: rgba(59, 130, 246, 0.1); color: #3b82f6;">
+                        <i class="bi bi-people"></i>
+                    </div>
+                    <div class="text-secondary small fw-medium">TOTAL USERS</div>
+                    <h3 class="fw-bold mt-1">{{ stats.total_users }}</h3>
+                </div>
+            </div>
+            <div class="col-md-3">
+                <div class="stat-card">
+                    <div class="stat-icon" style="background: rgba(16, 185, 129, 0.1); color: #10b981;">
+                        <i class="bi bi-image"></i>
+                    </div>
+                    <div class="text-secondary small fw-medium">TOTAL IMAGES</div>
+                    <h3 class="fw-bold mt-1">{{ stats.total_images }}</h3>
+                </div>
+            </div>
+            <div class="col-md-3">
+                <div class="stat-card">
+                    <div class="stat-icon" style="background: rgba(245, 158, 11, 0.1); color: #f59e0b;">
+                        <i class="bi bi-mic"></i>
+                    </div>
+                    <div class="text-secondary small fw-medium">TOTAL VOICES</div>
+                    <h3 class="fw-bold mt-1">{{ stats.total_voices }}</h3>
+                </div>
+            </div>
+            <div class="col-md-3">
+                <div class="stat-card">
+                    <div class="stat-icon" style="background: rgba(139, 92, 246, 0.1); color: #8b5cf6;">
+                        <i class="bi bi-gem"></i>
+                    </div>
+                    <div class="text-secondary small fw-medium">PREMIUM USERS</div>
+                    <h3 class="fw-bold mt-1">{{ stats.premium_users }}</h3>
+                </div>
+            </div>
+        </div>
+        <div class="row mt-4 g-4">
+            <div class="col-md-8">
+                <div class="table-container h-100">
+                    <div class="d-flex justify-content-between align-items-center mb-4">
+                        <h5 class="fw-bold mb-0">Recent Users</h5>
+                        <a href="#" class="text-accent small text-decoration-none">View All</a>
+                    </div>
+                    <div class="table-responsive">
+                        <table class="table">
+                            <thead>
+                                <tr>
+                                    <th>User</th>
+                                    <th>Chat ID</th>
+                                    <th>Tier</th>
+                                    <th>Activity</th>
+                                    <th>Action</th>
+                                </tr>
+                            </thead>
+                            <tbody>
+                                {% for user in users %}
+                                <tr>
+                                    <td>
+                                        <div class="d-flex align-items-center">
+                                            <div class="rounded-circle bg-secondary d-flex align-items-center justify-content-center me-3" style="width: 35px; height: 35px; font-size: 14px;">
+                                                {{ user.first_name[0] if user.first_name else 'U' }}
+                                            </div>
+                                            <div>
+                                                <div class="fw-medium">{{ user.first_name or 'User' }}</div>
+                                                <div class="text-secondary x-small">@{{ user.username or 'unknown' }}</div>
+                                            </div>
+                                        </div>
+                                    </td>
+                                    <td class="small">{{ user.chat_id }}</td>
+                                    <td>
+                                        <span class="badge {{ 'badge-premium' if user.tier == 'paid' else 'badge-free' }}">
+                                            {{ user.tier.upper() }}
+                                        </span>
+                                    </td>
+                                    <td>
+                                        <div class="text-secondary small">Img: {{ user.total_images_generated }}</div>
+                                        <div class="text-secondary small">Token: {{ user.token_balance }}</div>
+                                    </td>
+                                    <td>
+                                        <button class="btn btn-sm btn-outline-secondary border-0" onclick="editUser('{{ user.chat_id }}')">
+                                            <i class="bi bi-pencil-square"></i>
+                                        </button>
+                                    </td>
+                                </tr>
+                                {% endfor %}
+                            </tbody>
+                        </table>
+                    </div>
+                </div>
+            </div>
+            <div class="col-md-4">
+                <div class="table-container h-100">
+                    <h5 class="fw-bold mb-4">Activity Mix</h5>
+                    <canvas id="activityChart" height="300"></canvas>
+                </div>
+            </div>
+        </div>
+    </div>
+    <script>
+        const ctx = document.getElementById('activityChart').getContext('2d');
+        new Chart(ctx, {
+            type: 'doughnut',
+            data: {
+                labels: ['Images', 'Voices'],
+                datasets: [{
+                    data: [{{ stats.total_images }}, {{ stats.total_voices }}],
+                    backgroundColor: ['#6366f1', '#f59e0b'],
+                    borderWidth: 0,
+                    hoverOffset: 4
+                }]
+            },
+            options: {
+                plugins: {
+                    legend: {
+                        position: 'bottom',
+                        labels: { color: '#9ca3af', padding: 20 }
+                    }
+                },
+                cutout: '70%'
+            }
+        });
+        function editUser(chat_id) {
+            alert('User management for ' + chat_id + ' coming soon in next update');
+        }
+    </script>
+</body>
+</html>
+"""
+# Routes
+@app.route('/login', methods=['GET', 'POST'])
+def login():
+    if request.method == 'POST':
+        password = request.form.get('password')
+        if password == ADMIN_PASSWORD:
+            session['logged_in'] = True
+            return redirect(url_for('index'))
+        else:
+            flash('Invalid access token')
+    return render_template_string(LOGIN_TEMPLATE)
+@app.route('/logout')
+def logout():
+    session.pop('logged_in', None)
+    return redirect(url_for('login'))
+@app.route('/')
+@login_required
+def index():
+    try:
+        # Fetch stats
+        users_res = supabase.table("telegram_users").select("*", count="exact").order("created_at", desc=True).limit(50).execute()
+        # Aggregate stats
+        total_users = users_res.count if users_res.count else 0
+        # Get generation counts using counts from all users (simple aggregation for demo)
+        images_gen = 0
+        voices_gen = 0
+        premium_count = 0
+        # For a real large DB, we should use separate rollup tables or specialized RPCs
+        # But for this case, we aggregate from the user management data
+        all_users = supabase.table("telegram_users").select("tier, total_images_generated").execute()
+        for u in all_users.data:
+            images_gen += u.get('total_images_generated', 0)
+            if u.get('tier') == 'paid':
+                premium_count += 1
+        # Count voices from log table
+        voices_count_res = supabase.table("voice_generation_logs").select("id", count="exact").execute()
+        voices_gen = voices_count_res.count or 0
+        stats = {
+            "total_users": total_users,
+            "total_images": images_gen,
+            "total_voices": voices_gen,
+            "premium_users": premium_count
+        }
+        return render_template_string(
+            INDEX_TEMPLATE,
+            stats=stats,
+            users=users_res.data,
+            now=datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        )
+    except Exception as e:
+        return f"Database Error: {str(e)}"
+# Health check endpoint for Space
+@app.route('/health')
+def health():
+    return jsonify({"status": "healthy", "time": datetime.now().isoformat()}), 200
+if __name__ == "__main__":
+    port = int(os.getenv("PORT", 7860))
+    app.run(host='0.0.0.0', port=port, debug=True)

requirements.txt ADDED Viewed

	@@ -0,0 +1,6 @@

+python-telegram-bot
+supabase
+python-dotenv
+flask
+requests
+APScheduler>=3.10.0

start_admin.sh ADDED Viewed

	@@ -0,0 +1,3 @@

+#!/bin/bash
+# Start the Admin Dashboard
+python admin.py