Update app-backup1.py

app-backup1.py

@@ -3,392 +3,644 @@ import requests
 import json
 import os
 import time
+import sqlite3
+import hashlib
+import secrets
+from datetime import datetime, timedelta
 
-# 환경변수에서 API 키 로드
+# ============================================
+# 환경변수
+# ============================================
 FIREWORKS_API_KEY = os.getenv("FIREWORKS_API_KEY", "")
 FAL_KEY = os.getenv("FAL_KEY", "")
+ADMIN_EMAIL = os.getenv("ADMIN_EMAIL", "arxivgpt@gmail.com")
+SECRET_KEY = os.getenv("SECRET_KEY", "change-this-secret-key-in-production")
+
+# Persistent Storage 경로 (HuggingFace Spaces)
+DATA_DIR = "/data" if os.path.exists("/data") else "./data"
+DB_PATH = os.path.join(DATA_DIR, "app.db")
+
+# 설정
+DAILY_LIMIT_FREE = 10
+SESSION_EXPIRE_HOURS = 24 * 7
 
 # ============================================
-# 드라마틱한 커스텀 CSS - 사이버펑크 네온 테마
+# 데이터베이스
 # ============================================
-CUSTOM_CSS = """
-@import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;500;600;700;800;900&family=Rajdhani:wght@300;400;500;600;700&family=Share+Tech+Mono&display=swap');
+def get_db():
+    conn = sqlite3.connect(DB_PATH, check_same_thread=False)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+def init_db():
+    os.makedirs(DATA_DIR, exist_ok=True)
+    conn = get_db()
+    cursor = conn.cursor()
+    
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            email TEXT UNIQUE NOT NULL,
+            password_hash TEXT NOT NULL,
+            is_admin INTEGER DEFAULT 0,
+            is_active INTEGER DEFAULT 1,
+            daily_limit INTEGER DEFAULT 10,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            last_login TIMESTAMP
+        )
+    ''')
+    
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS sessions (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            user_id INTEGER NOT NULL,
+            token TEXT UNIQUE NOT NULL,
+            expires_at TIMESTAMP NOT NULL,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (user_id) REFERENCES users (id)
+        )
+    ''')
+    
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS generations (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            user_id INTEGER NOT NULL,
+            original_prompt TEXT,
+            enhanced_prompt TEXT,
+            image_url TEXT,
+            status TEXT DEFAULT 'success',
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (user_id) REFERENCES users (id)
+        )
+    ''')
+    
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS daily_usage (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            user_id INTEGER NOT NULL,
+            date DATE NOT NULL,
+            count INTEGER DEFAULT 0,
+            UNIQUE(user_id, date),
+            FOREIGN KEY (user_id) REFERENCES users (id)
+        )
+    ''')
+    
+    conn.commit()
+    
+    # 기본 관리자 계정
+    cursor.execute("SELECT id FROM users WHERE email = ?", (ADMIN_EMAIL,))
+    if not cursor.fetchone():
+        admin_password = secrets.token_urlsafe(12)
+        password_hash = hash_password(admin_password)
+        cursor.execute(
+            "INSERT INTO users (email, password_hash, is_admin, daily_limit) VALUES (?, ?, 1, 9999)",
+            (ADMIN_EMAIL, password_hash)
+        )
+        conn.commit()
+        print(f"[ADMIN] 관리자 계정: {ADMIN_EMAIL} / {admin_password}")
+    
+    conn.close()
 
-:root {
-    --neon-cyan: #00f5ff;
-    --neon-magenta: #ff00ff;
-    --neon-yellow: #ffff00;
-    --neon-green: #39ff14;
-    --deep-black: #0a0a0f;
-    --dark-surface: #12121a;
-    --glass-bg: rgba(18, 18, 26, 0.85);
-    --glow-cyan: 0 0 20px rgba(0, 245, 255, 0.5), 0 0 40px rgba(0, 245, 255, 0.3);
-    --glow-magenta: 0 0 20px rgba(255, 0, 255, 0.5), 0 0 40px rgba(255, 0, 255, 0.3);
-}
-
-/* 메인 컨테이너 */
-.gradio-container {
-    background: var(--deep-black) !important;
-    font-family: 'Rajdhani', sans-serif !important;
-    min-height: 100vh;
-    position: relative;
-    overflow-x: hidden;
-}
+# ============================================
+# 인증 유틸리티
+# ============================================
+def hash_password(password: str) -> str:
+    return hashlib.sha256((password + SECRET_KEY).encode()).hexdigest()
+
+def verify_password(password: str, password_hash: str) -> bool:
+    return hash_password(password) == password_hash
+
+def generate_session_token() -> str:
+    return secrets.token_urlsafe(32)
+
+def create_session(user_id: int) -> str:
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute("DELETE FROM sessions WHERE user_id = ?", (user_id,))
+    token = generate_session_token()
+    expires_at = datetime.now() + timedelta(hours=SESSION_EXPIRE_HOURS)
+    cursor.execute(
+        "INSERT INTO sessions (user_id, token, expires_at) VALUES (?, ?, ?)",
+        (user_id, token, expires_at)
+    )
+    cursor.execute("UPDATE users SET last_login = ? WHERE id = ?", (datetime.now(), user_id))
+    conn.commit()
+    conn.close()
+    return token
 
-/* 애니메이션 배경 그리드 */
-.gradio-container::before {
-    content: '';
-    position: fixed;
-    top: 0;
-    left: 0;
-    right: 0;
-    bottom: 0;
-    background: 
-        linear-gradient(90deg, rgba(0, 245, 255, 0.03) 1px, transparent 1px),
-        linear-gradient(rgba(0, 245, 255, 0.03) 1px, transparent 1px);
-    background-size: 50px 50px;
-    animation: gridMove 20s linear infinite;
-    pointer-events: none;
-    z-index: 0;
-}
+def validate_session(token: str) -> dict:
+    if not token:
+        return None
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute('''
+        SELECT u.id, u.email, u.is_admin, u.daily_limit, u.is_active
+        FROM sessions s JOIN users u ON s.user_id = u.id
+        WHERE s.token = ? AND s.expires_at > ? AND u.is_active = 1
+    ''', (token, datetime.now()))
+    row = cursor.fetchone()
+    conn.close()
+    if row:
+        return {"id": row["id"], "email": row["email"], "is_admin": bool(row["is_admin"]), "daily_limit": row["daily_limit"]}
+    return None
 
-@keyframes gridMove {
-    0% { transform: translate(0, 0); }
-    100% { transform: translate(50px, 50px); }
-}
+def delete_session(token: str):
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute("DELETE FROM sessions WHERE token = ?", (token,))
+    conn.commit()
+    conn.close()
 
-/* 플로팅 오브 효과 */
-.gradio-container::after {
-    content: '';
-    position: fixed;
-    width: 600px;
-    height: 600px;
-    background: radial-gradient(circle, rgba(255, 0, 255, 0.15) 0%, transparent 70%);
-    top: -200px;
-    right: -200px;
-    animation: floatOrb 15s ease-in-out infinite;
-    pointer-events: none;
-    z-index: 0;
-}
+# ============================================
+# 사용량 관리
+# ============================================
+def get_daily_usage(user_id: int) -> int:
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute("SELECT count FROM daily_usage WHERE user_id = ? AND date = ?", (user_id, datetime.now().date()))
+    row = cursor.fetchone()
+    conn.close()
+    return row["count"] if row else 0
+
+def increment_usage(user_id: int) -> int:
+    conn = get_db()
+    cursor = conn.cursor()
+    today = datetime.now().date()
+    cursor.execute('''
+        INSERT INTO daily_usage (user_id, date, count) VALUES (?, ?, 1)
+        ON CONFLICT(user_id, date) DO UPDATE SET count = count + 1
+    ''', (user_id, today))
+    conn.commit()
+    cursor.execute("SELECT count FROM daily_usage WHERE user_id = ? AND date = ?", (user_id, today))
+    row = cursor.fetchone()
+    conn.close()
+    return row["count"]
+
+def check_usage_limit(user_id: int, daily_limit: int) -> tuple:
+    current = get_daily_usage(user_id)
+    return (current < daily_limit, current, daily_limit)
 
-@keyframes floatOrb {
-    0%, 100% { transform: translate(0, 0) scale(1); opacity: 0.5; }
-    50% { transform: translate(-100px, 100px) scale(1.2); opacity: 0.8; }
-}
+# ============================================
+# 생성 이력
+# ============================================
+def save_generation(user_id: int, original: str, enhanced: str, image_url: str, status: str = "success"):
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute('''
+        INSERT INTO generations (user_id, original_prompt, enhanced_prompt, image_url, status)
+        VALUES (?, ?, ?, ?, ?)
+    ''', (user_id, original, enhanced, image_url, status))
+    conn.commit()
+    conn.close()
+
+def get_user_generations(user_id: int, limit: int = 20) -> list:
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute('''
+        SELECT original_prompt, enhanced_prompt, image_url, status, created_at
+        FROM generations WHERE user_id = ? ORDER BY created_at DESC LIMIT ?
+    ''', (user_id, limit))
+    rows = cursor.fetchall()
+    conn.close()
+    return [dict(row) for row in rows]
+
+# ============================================
+# 관리자 기능
+# ============================================
+def get_all_users() -> list:
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute('SELECT id, email, is_admin, is_active, daily_limit, created_at, last_login FROM users ORDER BY created_at DESC')
+    rows = cursor.fetchall()
+    conn.close()
+    return [dict(row) for row in rows]
+
+def get_stats() -> dict:
+    conn = get_db()
+    cursor = conn.cursor()
+    today = datetime.now().date()
+    
+    cursor.execute("SELECT COUNT(*) as count FROM users")
+    total_users = cursor.fetchone()["count"]
+    
+    cursor.execute("SELECT COUNT(DISTINCT user_id) as count FROM daily_usage WHERE date = ?", (today,))
+    active_today = cursor.fetchone()["count"]
+    
+    cursor.execute("SELECT COUNT(*) as count FROM generations")
+    total_generations = cursor.fetchone()["count"]
+    
+    cursor.execute("SELECT COUNT(*) as count FROM generations WHERE DATE(created_at) = ?", (today,))
+    generations_today = cursor.fetchone()["count"]
+    
+    conn.close()
+    return {"total_users": total_users, "active_today": active_today, "total_generations": total_generations, "generations_today": generations_today}
+
+def update_user_limit(user_id: int, daily_limit: int) -> bool:
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute("UPDATE users SET daily_limit = ? WHERE id = ?", (daily_limit, user_id))
+    conn.commit()
+    conn.close()
+    return True
+
+# ============================================
+# 회원가입 / 로그인
+# ============================================
+def register_user(email: str, password: str, confirm_password: str) -> tuple:
+    if not email or not password:
+        return None, "❌ 이메일과 비밀번호를 입력해주세요."
+    if password != confirm_password:
+        return None, "❌ 비밀번호가 일치하지 않습니다."
+    if len(password) < 6:
+        return None, "❌ 비밀번호는 6자 이상이어야 합니다."
+    if "@" not in email:
+        return None, "❌ 올바른 이메일 형식이 아닙니다."
+    
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute("SELECT id FROM users WHERE email = ?", (email,))
+    if cursor.fetchone():
+        conn.close()
+        return None, "❌ 이미 등록된 이메일입니다."
+    
+    password_hash = hash_password(password)
+    cursor.execute("INSERT INTO users (email, password_hash, daily_limit) VALUES (?, ?, ?)", (email, password_hash, DAILY_LIMIT_FREE))
+    conn.commit()
+    user_id = cursor.lastrowid
+    conn.close()
+    
+    token = create_session(user_id)
+    return token, f"✅ 회원가입 완료! 환영합니다, {email}"
+
+def login_user(email: str, password: str) -> tuple:
+    if not email or not password:
+        return None, "❌ 이메일과 비밀번호를 입력해주세요."
+    
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute("SELECT id, password_hash, is_active FROM users WHERE email = ?", (email,))
+    row = cursor.fetchone()
+    conn.close()
+    
+    if not row:
+        return None, "❌ 등록되지 않은 이메일입니다."
+    if not row["is_active"]:
+        return None, "❌ 비활성화된 계정입니다."
+    if not verify_password(password, row["password_hash"]):
+        return None, "❌ 비밀번호가 올바르지 않습니다."
+    
+    token = create_session(row["id"])
+    return token, "✅ 로그인 성공!"
+
+def logout_user(token: str) -> str:
+    if token:
+        delete_session(token)
+    return "✅ 로그아웃되었습니다."
+
+# ============================================
+# API 함수들
+# ============================================
+def enhance_prompt(prompt: str, fireworks_key: str) -> str:
+    if not prompt.strip():
+        return "❌ 프롬프트를 입력해주세요."
+    if not fireworks_key.strip():
+        return "❌ Fireworks API 키가 필요합니다."
+    
+    system_message = """You are a professional prompt engineer for AI image generation.
+Enhance prompts with: visual details, lighting, composition, style, mood, camera angles, color palette, quality modifiers.
+Output ONLY the enhanced prompt in English, no explanations."""
+
+    url = "https://api.fireworks.ai/inference/v1/chat/completions"
+    payload = {
+        "model": "accounts/fireworks/models/gpt-oss-120b",
+        "max_tokens": 4096,
+        "top_p": 1,
+        "top_k": 40,
+        "presence_penalty": 0,
+        "frequency_penalty": 0,
+        "temperature": 0.6,
+        "messages": [
+            {"role": "system", "content": system_message},
+            {"role": "user", "content": f"Enhance this prompt:\n\n\"{prompt}\""}
+        ]
+    }
+    headers = {
+        "Accept": "application/json",
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {fireworks_key}"
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=60)
+        response.raise_for_status()
+        data = response.json()
+        enhanced = data.get("choices", [{}])[0].get("message", {}).get("content", "")
+        return enhanced if enhanced else "❌ 증강 실패"
+    except requests.exceptions.HTTPError as e:
+        return f"❌ HTTP 오류: {e.response.status_code}"
+    except Exception as e:
+        return f"❌ 오류: {str(e)}"
+
+def generate_image(prompt: str, fal_key: str, aspect_ratio: str = "1:1", resolution: str = "1K") -> str:
+    if not prompt.strip() or not fal_key.strip():
+        return None
+    
+    # FAL API - nano-banana-pro
+    url = "https://fal.run/fal-ai/nano-banana-pro"
+    headers = {"Authorization": f"Key {fal_key}", "Content-Type": "application/json"}
+    
+    payload = {
+        "prompt": prompt,
+        "num_images": 1,
+        "aspect_ratio": aspect_ratio,
+        "output_format": "png",
+        "resolution": resolution
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, json=payload, timeout=120)
+        response.raise_for_status()
+        data = response.json()
+        if "images" in data and data["images"]:
+            return data["images"][0].get("url")
+    except Exception as e:
+        print(f"이미지 생성 오류: {e}")
+    return None
+
+def process_comparison_with_auth(original_prompt, session_token, fireworks_key, fal_key, aspect_ratio, resolution):
+    user = validate_session(session_token)
+    if not user:
+        yield "❌ 로그인이 필요합니다.", None, None, None, ""
+        return
+    
+    allowed, current, limit = check_usage_limit(user["id"], user["daily_limit"])
+    if not allowed:
+        yield f"❌ 일일 한도({limit}회) 초과", None, None, None, f"{current}/{limit}"
+        return
+    
+    fw_key = fireworks_key if fireworks_key.strip() else FIREWORKS_API_KEY
+    fl_key = fal_key if fal_key.strip() else FAL_KEY
+    
+    yield f"⚡ 처리 중... ({current}/{limit})", None, None, None, f"{current}/{limit}"
+    
+    enhanced = enhance_prompt(original_prompt, fw_key)
+    if enhanced.startswith("❌"):
+        yield enhanced, None, None, None, f"{current}/{limit}"
+        return
+    
+    yield "✅ 프롬프트 증강 완료\n⚡ 원본 이미지 생성 중...", enhanced, None, None, f"{current}/{limit}"
+    original_image = generate_image(original_prompt, fl_key, aspect_ratio, resolution)
+    
+    yield "✅ 원본 이미지 완료\n⚡ 증강 이미지 생성 중...", enhanced, original_image, None, f"{current}/{limit}"
+    enhanced_image = generate_image(enhanced, fl_key, aspect_ratio, resolution)
+    
+    new_count = increment_usage(user["id"])
+    save_generation(user["id"], original_prompt, enhanced, enhanced_image or original_image, "success" if enhanced_image else "partial")
+    
+    yield f"✅ 완료! 사용량: {new_count}/{limit}", enhanced, original_image, enhanced_image, f"{new_count}/{limit}"
 
-/* 모든 요소 z-index 조정 */
-.gradio-container > * {
-    position: relative;
-    z-index: 1;
+# ============================================
+# CSS - 밝은 프로페셔널 테마
+# ============================================
+CUSTOM_CSS = """
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&family=Poppins:wght@400;500;600;700&display=swap');
+
+:root {
+    --primary-blue: #2563eb;
+    --primary-blue-light: #3b82f6;
+    --primary-blue-dark: #1d4ed8;
+    --accent-purple: #7c3aed;
+    --accent-green: #10b981;
+    --accent-orange: #f59e0b;
+    --bg-white: #ffffff;
+    --bg-gray-50: #f9fafb;
+    --bg-gray-100: #f3f4f6;
+    --bg-gray-200: #e5e7eb;
+    --text-primary: #111827;
+    --text-secondary: #4b5563;
+    --text-muted: #9ca3af;
+    --border-color: #e5e7eb;
+    --shadow-sm: 0 1px 2px 0 rgba(0, 0, 0, 0.05);
+    --shadow-md: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06);
+    --shadow-lg: 0 10px 15px -3px rgba(0, 0, 0, 0.1), 0 4px 6px -2px rgba(0, 0, 0, 0.05);
+    --radius: 12px;
+}
+
+/* 메인 컨테이너 - 밝은 배경 */
+.gradio-container {
+    background: linear-gradient(135deg, var(--bg-gray-50) 0%, var(--bg-white) 50%, var(--bg-gray-100) 100%) !important;
+    font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif !important;
+    min-height: 100vh;
 }
 
 /* 타이틀 스타일 */
 .cyber-title {
-    font-family: 'Orbitron', monospace !important;
-    font-size: 3.5rem !important;
-    font-weight: 900 !important;
-    text-transform: uppercase;
-    letter-spacing: 8px;
-    background: linear-gradient(135deg, var(--neon-cyan), var(--neon-magenta), var(--neon-yellow));
-    background-size: 300% 300%;
+    font-family: 'Poppins', sans-serif !important;
+    font-size: 2.5rem !important;
+    font-weight: 700 !important;
+    background: linear-gradient(135deg, var(--primary-blue) 0%, var(--accent-purple) 100%);
     -webkit-background-clip: text;
     -webkit-text-fill-color: transparent;
     background-clip: text;
-    animation: gradientShift 5s ease infinite;
-    text-shadow: none;
-    filter: drop-shadow(0 0 30px rgba(0, 245, 255, 0.5));
-}
-
-@keyframes gradientShift {
-    0% { background-position: 0% 50%; }
-    50% { background-position: 100% 50%; }
-    100% { background-position: 0% 50%; }
-}
-
-/* 서브타이틀 */
-.cyber-subtitle {
-    font-family: 'Share Tech Mono', monospace !important;
-    color: var(--neon-cyan) !important;
-    font-size: 1.1rem !important;
-    letter-spacing: 4px;
-    opacity: 0.9;
-    animation: flicker 4s infinite;
-}
-
-@keyframes flicker {
-    0%, 100% { opacity: 0.9; }
-    92% { opacity: 0.9; }
-    93% { opacity: 0.4; }
-    94% { opacity: 0.9; }
-    96% { opacity: 0.4; }
-    97% { opacity: 0.9; }
-}
-
-/* 글래스모피즘 패널 */
-.glass-panel {
-    background: var(--glass-bg) !important;
-    backdrop-filter: blur(20px) !important;
-    -webkit-backdrop-filter: blur(20px) !important;
-    border: 1px solid rgba(0, 245, 255, 0.2) !important;
-    border-radius: 16px !important;
-    box-shadow: 
-        0 8px 32px rgba(0, 0, 0, 0.4),
-        inset 0 1px 0 rgba(255, 255, 255, 0.05) !important;
-    position: relative;
-    overflow: hidden;
+    text-align: center;
+    letter-spacing: -0.5px;
 }
 
-.glass-panel::before {
-    content: '';
-    position: absolute;
-    top: 0;
-    left: -100%;
-    width: 100%;
-    height: 100%;
-    background: linear-gradient(90deg, transparent, rgba(0, 245, 255, 0.1), transparent);
-    animation: shimmer 3s infinite;
-}
-
-@keyframes shimmer {
-    0% { left: -100%; }
-    100% { left: 100%; }
-}
-
-/* 입력 필드 */
+/* 입력 필드 - 밝은 스타일 */
 .gradio-container textarea,
 .gradio-container input[type="text"],
-.gradio-container input[type="password"] {
-    background: rgba(0, 0, 0, 0.6) !important;
-    border: 2px solid rgba(0, 245, 255, 0.3) !important;
-    border-radius: 12px !important;
-    color: #ffffff !important;
-    font-family: 'Rajdhani', sans-serif !important;
-    font-size: 1.1rem !important;
-    font-weight: 500 !important;
-    padding: 16px !important;
-    transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1) !important;
+.gradio-container input[type="password"],
+.gradio-container input[type="email"] {
+    background: var(--bg-white) !important;
+    border: 2px solid var(--border-color) !important;
+    border-radius: var(--radius) !important;
+    color: var(--text-primary) !important;
+    font-family: 'Inter', sans-serif !important;
+    font-size: 0.95rem !important;
+    padding: 12px 16px !important;
+    transition: all 0.2s ease !important;
 }
 
 .gradio-container textarea:focus,
 .gradio-container input:focus {
-    border-color: var(--neon-cyan) !important;
-    box-shadow: var(--glow-cyan), inset 0 0 20px rgba(0, 245, 255, 0.1) !important;
+    border-color: var(--primary-blue) !important;
+    box-shadow: 0 0 0 3px rgba(37, 99, 235, 0.1) !important;
     outline: none !important;
 }
 
 .gradio-container textarea::placeholder,
 .gradio-container input::placeholder {
-    color: rgba(0, 245, 255, 0.4) !important;
+    color: var(--text-muted) !important;
 }
 
-/* 라벨 */
+/* 라벨 스타일 */
 .gradio-container label {
-    font-family: 'Orbitron', monospace !important;
-    color: var(--neon-cyan) !important;
-    font-size: 0.9rem !important;
+    font-family: 'Inter', sans-serif !important;
+    color: var(--text-primary) !important;
     font-weight: 600 !important;
-    letter-spacing: 2px !important;
-    text-transform: uppercase !important;
+    font-size: 0.9rem !important;
+    margin-bottom: 6px !important;
 }
 
 /* 드롭다운 */
 .gradio-container select,
 .gradio-container .wrap-inner {
-    background: rgba(0, 0, 0, 0.6) !important;
-    border: 2px solid rgba(0, 245, 255, 0.3) !important;
-    border-radius: 12px !important;
-    color: #ffffff !important;
-    font-family: 'Rajdhani', sans-serif !important;
+    background: var(--bg-white) !important;
+    border: 2px solid var(--border-color) !important;
+    border-radius: var(--radius) !important;
+    color: var(--text-primary) !important;
 }
 
-/* 메인 버튼 - 네온 글로우 */
+/* 메인 버튼 - 블루 그라디언트 */
 .neon-button {
-    background: linear-gradient(135deg, rgba(0, 245, 255, 0.2), rgba(255, 0, 255, 0.2)) !important;
-    border: 2px solid var(--neon-cyan) !important;
-    border-radius: 12px !important;
+    background: linear-gradient(135deg, var(--primary-blue) 0%, var(--primary-blue-dark) 100%) !important;
+    border: none !important;
+    border-radius: var(--radius) !important;
     color: #ffffff !important;
-    font-family: 'Orbitron', monospace !important;
-    font-size: 1.2rem !important;
-    font-weight: 700 !important;
-    letter-spacing: 3px !important;
-    text-transform: uppercase !important;
-    padding: 20px 40px !important;
-    position: relative !important;
-    overflow: hidden !important;
-    transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1) !important;
-    box-shadow: var(--glow-cyan) !important;
+    font-family: 'Inter', sans-serif !important;
+    font-size: 0.95rem !important;
+    font-weight: 600 !important;
+    padding: 12px 24px !important;
+    box-shadow: var(--shadow-md), 0 4px 14px rgba(37, 99, 235, 0.25) !important;
+    transition: all 0.2s ease !important;
 }
 
-.neon-button::before {
-    content: '';
-    position: absolute;
-    top: 0;
-    left: -100%;
-    width: 100%;
-    height: 100%;
-    background: linear-gradient(90deg, transparent, rgba(255, 255, 255, 0.3), transparent);
-    transition: left 0.5s;
+.neon-button:hover {
+    background: linear-gradient(135deg, var(--primary-blue-light) 0%, var(--primary-blue) 100%) !important;
+    box-shadow: var(--shadow-lg), 0 6px 20px rgba(37, 99, 235, 0.35) !important;
+    transform: translateY(-2px) !important;
 }
 
-.neon-button:hover {
-    background: linear-gradient(135deg, rgba(0, 245, 255, 0.4), rgba(255, 0, 255, 0.4)) !important;
-    border-color: var(--neon-magenta) !important;
-    box-shadow: var(--glow-magenta), 0 10px 40px rgba(255, 0, 255, 0.3) !important;
-    transform: translateY(-3px) !important;
+/* 일반 버튼 */
+.gradio-container button {
+    border-radius: var(--radius) !important;
+    font-family: 'Inter', sans-serif !important;
+    transition: all 0.2s ease !important;
 }
 
-.neon-button:hover::before {
-    left: 100%;
+/* 탭 스타일 */
+.gradio-container .tabs {
+    background: var(--bg-white) !important;
+    border-radius: 16px !important;
+    padding: 8px !important;
+    box-shadow: var(--shadow-sm) !important;
 }
 
-/* 상태 표시 박스 */
-.status-box {
-    background: rgba(0, 0, 0, 0.6) !important;
-    border-left: 4px solid var(--neon-cyan) !important;
-    border-radius: 8px !important;
-    padding: 16px !important;
-    font-family: 'Share Tech Mono', monospace !important;
-    color: var(--neon-green) !important;
+.gradio-container .tab-nav button {
+    background: transparent !important;
+    color: var(--text-secondary) !important;
+    border-radius: 10px !important;
+    font-weight: 500 !important;
+    padding: 10px 20px !important;
 }
 
-/* 증강 프롬프트 출력 */
-.enhanced-output textarea {
-    background: linear-gradient(135deg, rgba(57, 255, 20, 0.05), rgba(0, 245, 255, 0.05)) !important;
-    border: 2px solid rgba(57, 255, 20, 0.4) !important;
-    color: var(--neon-green) !important;
-    font-family: 'Share Tech Mono', monospace !important;
-    text-shadow: 0 0 10px rgba(57, 255, 20, 0.3);
+.gradio-container .tab-nav button.selected {
+    background: var(--primary-blue) !important;
+    color: #ffffff !important;
 }
 
-/* 이미지 카드 */
-.image-card {
-    background: var(--glass-bg) !important;
-    border: 2px solid rgba(0, 245, 255, 0.2) !important;
-    border-radius: 16px !important;
+/* 아코디언 */
+.gradio-container .accordion {
+    background: var(--bg-white) !important;
+    border: 1px solid var(--border-color) !important;
+    border-radius: var(--radius) !important;
+    box-shadow: var(--shadow-sm) !important;
+}
+
+/* 마크다운 텍스트 */
+.gradio-container .markdown-text {
+    color: var(--text-primary) !important;
+}
+
+.gradio-container .markdown-text h1,
+.gradio-container .markdown-text h2,
+.gradio-container .markdown-text h3 {
+    color: var(--text-primary) !important;
+}
+
+/* 테이블 */
+.gradio-container table {
+    background: var(--bg-white) !important;
+    border-radius: var(--radius) !important;
     overflow: hidden !important;
-    transition: all 0.4s cubic-bezier(0.4, 0, 0.2, 1) !important;
 }
 
-.image-card:hover {
-    border-color: var(--neon-magenta) !important;
-    box-shadow: var(--glow-magenta) !important;
-    transform: scale(1.02) !important;
+.gradio-container th {
+    background: var(--bg-gray-100) !important;
+    color: var(--text-primary) !important;
+    font-weight: 600 !important;
+}
+
+.gradio-container td {
+    color: var(--text-secondary) !important;
+    border-color: var(--border-color) !important;
 }
 
 /* 이미지 컨테이너 */
-.gradio-container .image-container img {
-    border-radius: 12px !important;
-    box-shadow: 0 0 30px rgba(0, 245, 255, 0.2) !important;
+.gradio-container .image-container {
+    background: var(--bg-white) !important;
+    border-radius: var(--radius) !important;
+    box-shadow: var(--shadow-md) !important;
 }
 
-/* 아코디언 */
-.gradio-container .accordion {
-    background: var(--glass-bg) !important;
-    border: 1px solid rgba(0, 245, 255, 0.2) !important;
-    border-radius: 12px !important;
+/* 카드 스타일 박스 */
+.gradio-container .block {
+    background: var(--bg-white) !important;
+    border: 1px solid var(--border-color) !important;
+    border-radius: var(--radius) !important;
 }
 
-/* 예시 버튼 */
-.gradio-container .examples-button {
-    background: rgba(0, 245, 255, 0.1) !important;
-    border: 1px solid rgba(0, 245, 255, 0.3) !important;
-    border-radius: 20px !important;
-    color: var(--neon-cyan) !important;
-    font-family: 'Rajdhani', sans-serif !important;
-    transition: all 0.3s ease !important;
+/* 슬라이더 */
+.gradio-container input[type="range"] {
+    accent-color: var(--primary-blue) !important;
 }
 
-.gradio-container .examples-button:hover {
-    background: rgba(0, 245, 255, 0.2) !important;
-    box-shadow: var(--glow-cyan) !important;
+/* 체크박스 */
+.gradio-container input[type="checkbox"] {
+    accent-color: var(--primary-blue) !important;
 }
 
-/* 섹션 헤더 */
-.section-header {
-    font-family: 'Orbitron', monospace !important;
-    color: var(--neon-magenta) !important;
-    font-size: 1.5rem !important;
-    font-weight: 700 !important;
-    letter-spacing: 4px !important;
-    text-transform: uppercase !important;
-    text-shadow: 0 0 20px rgba(255, 0, 255, 0.5);
-}
-
-/* 비교 라벨 */
-.compare-label-original {
-    background: linear-gradient(90deg, rgba(255, 107, 107, 0.2), transparent) !important;
-    border-left: 4px solid #ff6b6b !important;
-    color: #ff6b6b !important;
-    font-family: 'Orbitron', monospace !important;
-    padding: 12px 20px !important;
-    font-weight: 600 !important;
-    letter-spacing: 2px !important;
+/* 상태 메시지 */
+.gradio-container .message {
+    border-radius: var(--radius) !important;
 }
 
-.compare-label-enhanced {
-    background: linear-gradient(90deg, rgba(57, 255, 20, 0.2), transparent) !important;
-    border-left: 4px solid var(--neon-green) !important;
-    color: var(--neon-green) !important;
-    font-family: 'Orbitron', monospace !important;
-    padding: 12px 20px !important;
-    font-weight: 600 !important;
-    letter-spacing: 2px !important;
-}
-
-/* 스캔라인 효과 */
-.scanlines::after {
-    content: '';
-    position: fixed;
-    top: 0;
-    left: 0;
-    right: 0;
-    bottom: 0;
-    background: repeating-linear-gradient(
-        0deg,
-        rgba(0, 0, 0, 0.1) 0px,
-        rgba(0, 0, 0, 0.1) 1px,
-        transparent 1px,
-        transparent 2px
-    );
-    pointer-events: none;
-    z-index: 9999;
-    opacity: 0.3;
-}
-
-/* 푸터 */
-.cyber-footer {
-    font-family: 'Share Tech Mono', monospace !important;
-    color: rgba(0, 245, 255, 0.6) !important;
-    font-size: 0.85rem !important;
-    letter-spacing: 2px !important;
-}
-
-/* 로딩 애니메이션 */
-@keyframes pulse-glow {
-    0%, 100% { box-shadow: var(--glow-cyan); }
-    50% { box-shadow: var(--glow-magenta); }
-}
-
-.loading {
-    animation: pulse-glow 1.5s ease-in-out infinite;
-}
-
-/* 반응형 */
-@media (max-width: 768px) {
-    .cyber-title {
-        font-size: 2rem !important;
-        letter-spacing: 4px;
-    }
-    
-    .cyber-subtitle {
-        font-size: 0.9rem !important;
-        letter-spacing: 2px;
-    }
+/* 성공/에러 색상 */
+.success-text {
+    color: var(--accent-green) !important;
+}
+
+.error-text {
+    color: #ef4444 !important;
+}
+
+/* 데이터프레임 */
+.gradio-container .dataframe {
+    border-radius: var(--radius) !important;
+    overflow: hidden !important;
 }
 
-/* ============================================
-   허깅페이스 스페이스 배지/링크 완전 숨기기
-   ============================================ */
+/* 스크롤바 */
+::-webkit-scrollbar {
+    width: 8px;
+    height: 8px;
+}
+
+::-webkit-scrollbar-track {
+    background: var(--bg-gray-100);
+    border-radius: 4px;
+}
+
+::-webkit-scrollbar-thumb {
+    background: var(--bg-gray-300);
+    border-radius: 4px;
+}
+
+::-webkit-scrollbar-thumb:hover {
+    background: var(--text-muted);
+}
+
+/* 허깅페이스 배지 숨기기 */
 .built-with,
 .built-with-badge,
 a[href*="huggingface.co/spaces"],
@@ -397,435 +649,197 @@ footer,
 #footer,
 .gradio-container > footer,
 div[class*="footer"],
-.gradio-container a[target="_blank"][href*="huggingface"],
-.gradio-container .wrap:has(a[href*="huggingface"]),
 .space-info,
 .hf-space-header,
 [class*="space-header"],
-.wrap.default.minimal.svelte-1rjryqp,
+div.wrap.svelte-1rjryqp,
 .svelte-1rjryqp:has(a[href*="huggingface"]),
-.wrap:has(> a[href*="huggingface"]) {
+a.svelte-1rjryqp[href*="huggingface"] {
     display: none !important;
     visibility: hidden !important;
     opacity: 0 !important;
     height: 0 !important;
     width: 0 !important;
-    overflow: hidden !important;
     position: absolute !important;
     pointer-events: none !important;
-    z-index: -9999 !important;
 }
 
-/* 우측 상단 고정 위치 요소 숨기기 (허깅페이스 배지) */
-.gradio-container > div:first-child > a[href*="huggingface"],
-body > div > a[href*="huggingface"],
 div[style*="position: fixed"][style*="right"],
-div[style*="position: fixed"][style*="top: 0"],
-.fixed.top-0.right-0,
-a.svelte-1rjryqp[href*="huggingface"],
-div.wrap.svelte-1rjryqp {
+div[style*="position: fixed"][style*="top: 0"] {
     display: none !important;
-    visibility: hidden !important;
-    opacity: 0 !important;
 }
 """
 
 # ============================================
-# API 함수들
-# ============================================
-
-def enhance_prompt(prompt: str, fireworks_key: str) -> str:
-    """Fireworks AI LLM API를 사용하여 프롬프트를 증강합니다."""
-    if not prompt.strip():
-        return "❌ 프롬프트를 입력해주세요."
-    
-    if not fireworks_key.strip():
-        return "❌ Fireworks API 키를 입력해주세요."
-    
-    system_message = """You are a professional prompt engineer specializing in AI image and video generation. 
-Your task is to enhance user prompts to create more detailed, vivid, and effective prompts for AI image/video generation models.
-
-When enhancing prompts:
-1. Add specific visual details (lighting, composition, style, mood)
-2. Include artistic references or styles when appropriate
-3. Specify camera angles, perspectives, and depth of field
-4. Add color palette suggestions
-5. Include quality modifiers (8K, cinematic, professional, etc.)
-6. Maintain the original intent while enriching the description
-7. Output the enhanced prompt in English for best AI model compatibility
-
-Respond ONLY with the enhanced prompt, no explanations or additional text."""
-
-    url = "https://api.fireworks.ai/inference/v1/chat/completions"
-    
-    payload = {
-        "model": "accounts/fireworks/models/gpt-oss-120b",
-        "max_tokens": 4096,
-        "top_p": 1,
-        "top_k": 40,
-        "presence_penalty": 0,
-        "frequency_penalty": 0,
-        "temperature": 0.6,
-        "messages": [
-            {"role": "system", "content": system_message},
-            {"role": "user", "content": f"다음 프롬프트를 이미지/영상 생성 AI에 최적화된 상세하고 풍부한 프롬프트로 증강해주세요:\n\n\"{prompt}\""}
-        ]
-    }
-    
-    headers = {
-        "Accept": "application/json",
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {fireworks_key}"
-    }
-    
-    try:
-        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=60)
-        response.raise_for_status()
-        data = response.json()
-        enhanced = data.get("choices", [{}])[0].get("message", {}).get("content", "")
-        return enhanced if enhanced else "❌ 응답에서 증강된 프롬프트를 찾을 수 없습니다."
-    except requests.exceptions.Timeout:
-        return "❌ API 요청 시간 초과. 다시 시도해주세요."
-    except requests.exceptions.HTTPError as e:
-        return f"❌ API 오류: {e.response.status_code}"
-    except Exception as e:
-        return f"❌ 오류 발생: {str(e)}"
-
-
-def generate_image(prompt: str, fal_key: str, aspect_ratio: str = "1:1", resolution: str = "1K") -> str:
-    """FAL AI의 nano-banana-pro 모델을 사용하여 이미지를 생성합니다."""
-    if not prompt.strip() or not fal_key.strip():
-        return None
-    
-    url = "https://queue.fal.run/fal-ai/nano-banana-pro"
-    headers = {"Authorization": f"Key {fal_key}", "Content-Type": "application/json"}
-    payload = {
-        "prompt": prompt,
-        "num_images": 1,
-        "aspect_ratio": aspect_ratio,
-        "output_format": "png",
-        "resolution": resolution
-    }
-    
-    try:
-        response = requests.post(url, headers=headers, json=payload, timeout=120)
-        response.raise_for_status()
-        data = response.json()
-        
-        if "images" in data:
-            images = data.get("images", [])
-            if images:
-                return images[0].get("url", None)
-        
-        request_id = data.get("request_id")
-        if request_id:
-            return poll_for_result(request_id, fal_key)
-        return None
-    except Exception as e:
-        print(f"이미지 생성 오류: {e}")
-        return None
-
-
-def poll_for_result(request_id: str, fal_key: str, max_attempts: int = 60) -> str:
-    """Queue 결과를 폴링하여 이미지 URL을 가져옵니다."""
-    status_url = f"https://queue.fal.run/fal-ai/nano-banana-pro/requests/{request_id}/status"
-    result_url = f"https://queue.fal.run/fal-ai/nano-banana-pro/requests/{request_id}"
-    headers = {"Authorization": f"Key {fal_key}", "Content-Type": "application/json"}
-    
-    for _ in range(max_attempts):
-        try:
-            status_response = requests.get(status_url, headers=headers, timeout=30)
-            status_data = status_response.json()
-            status = status_data.get("status", "")
-            
-            if status == "COMPLETED":
-                result_response = requests.get(result_url, headers=headers, timeout=30)
-                result_data = result_response.json()
-                images = result_data.get("images", [])
-                return images[0].get("url", None) if images else None
-            elif status == "FAILED":
-                return None
-            time.sleep(2)
-        except Exception:
-            time.sleep(2)
-    return None
-
-
-def process_comparison(original_prompt: str, fireworks_key: str, fal_key: str, aspect_ratio: str, resolution: str):
-    """원본 프롬프트 증강 → 두 이미지 생성 → 비교 결과 반환"""
-    
-    # API 키 기본값 적용
-    fw_key = fireworks_key if fireworks_key.strip() else FIREWORKS_API_KEY
-    fl_key = fal_key if fal_key.strip() else FAL_KEY
-    
-    yield "⚡ INITIALIZING PROMPT ENHANCEMENT SEQUENCE...", None, None, None
-    
-    enhanced = enhance_prompt(original_prompt, fw_key)
-    
-    if enhanced.startswith("❌"):
-        yield enhanced, None, None, None
-        return
-    
-    yield "✅ PROMPT ENHANCED\n\n⚡ GENERATING ORIGINAL IMAGE...", enhanced, None, None
-    
-    original_image = generate_image(original_prompt, fl_key, aspect_ratio, resolution)
-    
-    if not original_image:
-        yield "✅ PROMPT ENHANCED\n\n❌ ORIGINAL IMAGE GENERATION FAILED", enhanced, None, None
-    else:
-        yield "✅ PROMPT ENHANCED\n✅ ORIGINAL IMAGE COMPLETE\n\n⚡ GENERATING ENHANCED IMAGE...", enhanced, original_image, None
-    
-    enhanced_image = generate_image(enhanced, fl_key, aspect_ratio, resolution)
-    
-    if not enhanced_image:
-        yield "✅ PROMPT ENHANCED\n✅ ORIGINAL IMAGE COMPLETE\n\n❌ ENHANCED IMAGE GENERATION FAILED", enhanced, original_image, None
-    else:
-        yield "✅ ALL SYSTEMS OPERATIONAL\n✅ COMPARISON READY", enhanced, original_image, enhanced_image
-
-
-# ============================================
-# Gradio UI 구성
+# Gradio UI
 # ============================================
+init_db()
 
 examples = [
     ["한복을 입은 여성이 전통 한옥 마당에서 부채를 들고 있다"],
     ["사이버펑크 도시의 네온사인이 빛나는 밤거리"],
     ["바다 위 일몰과 작은 돛단배"],
-    ["눈 덮인 산속 작은 오두막에서 연기가 피어오른다"],
-    ["미래 도시의 하늘을 나는 자동차들"],
 ]
 
 with gr.Blocks(title="AI PROMPT ENHANCER", css=CUSTOM_CSS) as demo:
+    session_token = gr.State(value="")
     
-    # 헤더
-    gr.HTML("""
-        <div class="scanlines">
-            <div style="text-align: center; padding: 60px 20px 40px 20px; position: relative;">
-                <!-- 배경 글로우 -->
-                <div style="
-                    position: absolute;
-                    top: 50%;
-                    left: 50%;
-                    transform: translate(-50%, -50%);
-                    width: 400px;
-                    height: 400px;
-                    background: radial-gradient(circle, rgba(0, 245, 255, 0.15) 0%, transparent 70%);
-                    filter: blur(40px);
-                    pointer-events: none;
-                "></div>
-                
-                <!-- 아이콘 -->
-                <div style="
-                    width: 80px;
-                    height: 80px;
-                    margin: 0 auto 24px auto;
-                    background: linear-gradient(135deg, rgba(0, 245, 255, 0.3), rgba(255, 0, 255, 0.3));
-                    border: 2px solid rgba(0, 245, 255, 0.5);
-                    border-radius: 20px;
-                    display: flex;
-                    align-items: center;
-                    justify-content: center;
-                    font-size: 36px;
-                    box-shadow: 0 0 40px rgba(0, 245, 255, 0.4);
-                    animation: pulse-glow 3s ease-in-out infinite;
-                ">🚀</div>
-                
-                <!-- 타이틀 -->
-                <h1 class="cyber-title">PROMPT ENHANCER</h1>
-                
-                <!-- 서브타이틀 -->
-                <p class="cyber-subtitle" style="margin-top: 16px;">
-                    [ NEURAL NETWORK ACTIVATED ] — IMAGE GENERATION PROTOCOL v2.0
-                </p>
-                
-                <!-- 데코 라인 -->
-                <div style="
-                    width: 200px;
-                    height: 2px;
-                    background: linear-gradient(90deg, transparent, var(--neon-cyan), transparent);
-                    margin: 24px auto 0 auto;
-                "></div>
-            </div>
-        </div>
-    """)
-    
-    # API Keys (접힌 상태)
-    with gr.Accordion("⚙️ SYSTEM CONFIGURATION", open=not (FIREWORKS_API_KEY and FAL_KEY), elem_classes=["glass-panel"]):
-        with gr.Row():
-            fireworks_key_input = gr.Textbox(
-                label="FIREWORKS API KEY",
-                placeholder="fw_...",
-                type="password",
-                value=FIREWORKS_API_KEY,
-            )
-            fal_key_input = gr.Textbox(
-                label="FAL API KEY", 
-                placeholder="Enter your FAL key...",
-                type="password",
-                value=FAL_KEY,
-            )
-    
-    gr.HTML("<div style='height: 20px;'></div>")
-    
-    # 메인 입력 섹션
-    with gr.Row():
-        with gr.Column(scale=2):
-            gr.HTML("""
-                <div style="
-                    font-family: 'Orbitron', monospace;
-                    color: #00f5ff;
-                    font-size: 0.9rem;
-                    letter-spacing: 3px;
-                    margin-bottom: 12px;
-                    text-transform: uppercase;
-                ">📝 INPUT PROMPT</div>
-            """)
-            prompt_input = gr.Textbox(
-                label="",
-                placeholder="[ ENTER YOUR VISION HERE ]\n예: 한복을 입은 여성이 벚꽃 아래에서 걷고 있다",
-                lines=5,
-                show_label=False,
-                elem_classes=["glass-panel"]
-            )
-            
-            gr.Examples(
-                examples=examples,
-                inputs=prompt_input,
-                label="💡 QUICK PROMPTS"
-            )
+    gr.HTML('''
+    <div style="text-align:center; padding:40px 20px; background: linear-gradient(135deg, #ffffff 0%, #f0f9ff 100%); border-radius: 16px; margin-bottom: 20px; box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);">
+        <h1 class="cyber-title">🚀 PROMPT ENHANCER</h1>
+        <p style="color: #4b5563; font-size: 1.1rem; margin-top: 8px;">AI-Powered Image Generation Platform</p>
+    </div>
+    ''')
+    
+    with gr.Tabs() as tabs:
+        # Tab 1: 로그인
+        with gr.Tab("🔐 로그인"):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    pass
+                with gr.Column(scale=2):
+                    with gr.Tabs():
+                        with gr.Tab("로그인"):
+                            login_email = gr.Textbox(label="이메일", placeholder="your@email.com")
+                            login_password = gr.Textbox(label="비밀번호", type="password")
+                            login_btn = gr.Button("로그인", elem_classes=["neon-button"])
+                            login_status = gr.Markdown("")
+                        with gr.Tab("회원가입"):
+                            reg_email = gr.Textbox(label="이메일", placeholder="your@email.com")
+                            reg_password = gr.Textbox(label="비밀번호", type="password", placeholder="6자 이상")
+                            reg_confirm = gr.Textbox(label="비밀번호 확인", type="password")
+                            reg_btn = gr.Button("회원가입", elem_classes=["neon-button"])
+                            reg_status = gr.Markdown("")
+                    current_user_display = gr.Markdown("로그인하지 않음")
+                    logout_btn = gr.Button("로그아웃", visible=False)
+                with gr.Column(scale=1):
+                    pass
         
-        with gr.Column(scale=1):
-            gr.HTML("""
-                <div style="
-                    font-family: 'Orbitron', monospace;
-                    color: #ff00ff;
-                    font-size: 0.9rem;
-                    letter-spacing: 3px;
-                    margin-bottom: 12px;
-                    text-transform: uppercase;
-                ">⚙️ PARAMETERS</div>
-            """)
-            aspect_ratio = gr.Dropdown(
-                label="ASPECT RATIO",
-                choices=["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3", "21:9"],
-                value="1:1",
-            )
-            resolution = gr.Dropdown(
-                label="RESOLUTION",
-                choices=["1K", "2K", "4K"],
-                value="1K",
-            )
+        # Tab 2: 메인 서비스
+        with gr.Tab("✨ 이미지 생성"):
+            usage_display = gr.Markdown("사용량: 로그인 필요")
             
-            gr.HTML("<div style='height: 16px;'></div>")
+            with gr.Accordion("⚙️ API 설정", open=False):
+                with gr.Row():
+                    fireworks_key_input = gr.Textbox(label="FIREWORKS API KEY", placeholder="환경변수 사용", type="password")
+                    fal_key_input = gr.Textbox(label="FAL API KEY", placeholder="환경변수 사용", type="password")
             
-            generate_btn = gr.Button(
-                "⚡ EXECUTE ENHANCEMENT",
-                variant="primary",
-                size="lg",
-                elem_classes=["neon-button"]
-            )
-    
-    gr.HTML("<div style='height: 30px;'></div>")
-    
-    # 상태 표시
-    status_text = gr.Textbox(
-        label="📊 SYSTEM STATUS",
-        value="[ AWAITING INPUT ]",
-        interactive=False,
-        lines=3,
-        elem_classes=["status-box"]
-    )
+            with gr.Row():
+                with gr.Column(scale=2):
+                    prompt_input = gr.Textbox(label="프롬프트 입력", placeholder="생성할 이미지를 설명해주세요...", lines=4)
+                    gr.Examples(examples=examples, inputs=prompt_input, label="예시")
+                with gr.Column(scale=1):
+                    aspect_ratio = gr.Dropdown(label="비율", choices=["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3", "21:9"], value="1:1")
+                    resolution = gr.Dropdown(label="해상도", choices=["1K", "2K", "4K"], value="1K")
+                    generate_btn = gr.Button("⚡ 생성하기", elem_classes=["neon-button"], size="lg")
+            
+            status_text = gr.Textbox(label="상태", interactive=False, lines=2)
+            enhanced_output = gr.Textbox(label="증강된 프롬프트", lines=3)
+            
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### 📌 원본")
+                    original_image_output = gr.Image(label="", height=400)
+                with gr.Column():
+                    gr.Markdown("### ✨ 증강")
+                    enhanced_image_output = gr.Image(label="", height=400)
+        
+        # Tab 3: 내 계정
+        with gr.Tab("👤 내 계정") as account_tab:
+            account_info = gr.Markdown("로그인이 필요합니다.")
+            history_display = gr.Dataframe(headers=["프롬프트", "생성일시", "상태"], interactive=False)
+            refresh_history_btn = gr.Button("🔄 새로고침")
+        
+        # Tab 4: 관리자
+        with gr.Tab("🛠️ 관리자") as admin_tab:
+            admin_auth_status = gr.Markdown("관리자 권한이 필요합니다.")
+            with gr.Row(visible=False) as admin_panel:
+                with gr.Column():
+                    gr.Markdown("### 📊 통계")
+                    stats_display = gr.Markdown("")
+                    refresh_stats_btn = gr.Button("통계 새로고침")
+                with gr.Column():
+                    gr.Markdown("### 👥 사용자 관리")
+                    users_table = gr.Dataframe(headers=["ID", "이메일", "관리자", "활성", "일일제한", "가입일"], interactive=False)
+                    refresh_users_btn = gr.Button("사용자 새로고침")
+            with gr.Row(visible=False) as admin_actions:
+                user_id_input = gr.Number(label="사용자 ID", precision=0)
+                toggle_active_btn = gr.Button("활성화/비활성화")
+                new_limit = gr.Number(label="새 일일 제한", value=10, precision=0)
+                update_limit_btn = gr.Button("제한 변경")
+                admin_action_status = gr.Markdown("")
+
+    # ========== 이벤트 핸들러 ==========
+    def do_login(email, password):
+        token, msg = login_user(email, password)
+        if token:
+            user = validate_session(token)
+            user_info = f"✅ **{user['email']}**" + (" (관리자)" if user['is_admin'] else "")
+            return token, msg, user_info, gr.update(visible=True)
+        return "", msg, "로그인하지 않음", gr.update(visible=False)
     
-    gr.HTML("<div style='height: 20px;'></div>")
-    
-    # 증강 프롬프트 출력
-    gr.HTML("""
-        <div style="
-            font-family: 'Orbitron', monospace;
-            color: #39ff14;
-            font-size: 1rem;
-            letter-spacing: 3px;
-            margin-bottom: 12px;
-            text-transform: uppercase;
-            text-shadow: 0 0 20px rgba(57, 255, 20, 0.5);
-        ">✨ ENHANCED OUTPUT</div>
-    """)
-    enhanced_output = gr.Textbox(
-        label="",
-        placeholder="[ ENHANCED PROMPT WILL APPEAR HERE ]",
-        lines=5,
-        interactive=True,
-        show_label=False,
-        elem_classes=["enhanced-output", "glass-panel"]
-    )
+    def do_register(email, password, confirm):
+        token, msg = register_user(email, password, confirm)
+        if token:
+            user = validate_session(token)
+            return token, msg, f"✅ **{user['email']}**", gr.update(visible=True)
+        return "", msg, "로그인하지 않음", gr.update(visible=False)
     
-    gr.HTML("<div style='height: 40px;'></div>")
-    
-    # 이미지 비교 섹션
-    gr.HTML("""
-        <div style="text-align: center; margin-bottom: 30px;">
-            <div class="section-header">🖼️ VISUAL COMPARISON</div>
-            <div style="
-                width: 100px;
-                height: 2px;
-                background: linear-gradient(90deg, transparent, #ff00ff, transparent);
-                margin: 16px auto 0 auto;
-            "></div>
-        </div>
-    """)
-    
-    with gr.Row():
-        with gr.Column(elem_classes=["image-card"]):
-            gr.HTML("""
-                <div class="compare-label-original">
-                    📌 ORIGINAL PROMPT RESULT
-                </div>
-            """)
-            original_image_output = gr.Image(
-                label="",
-                type="filepath",
-                show_label=False,
-                height=400
-            )
-        
-        with gr.Column(elem_classes=["image-card"]):
-            gr.HTML("""
-                <div class="compare-label-enhanced">
-                    ✨ ENHANCED PROMPT RESULT
-                </div>
-            """)
-            enhanced_image_output = gr.Image(
-                label="",
-                type="filepath",
-                show_label=False,
-                height=400
-            )
-    
-    # 푸터
-    gr.HTML("""
-        <div style="text-align: center; padding: 50px 20px 30px 20px;">
-            <div style="
-                width: 60%;
-                height: 1px;
-                background: linear-gradient(90deg, transparent, rgba(0, 245, 255, 0.3), transparent);
-                margin: 0 auto 30px auto;
-            "></div>
-            <p class="cyber-footer">
-                POWERED BY FIREWORKS AI (GPT-OSS-120B) + FAL AI (NANO-BANANA-PRO)
-            </p>
-            <p class="cyber-footer" style="margin-top: 8px; opacity: 0.6;">
-                [ SYSTEM VERSION 2.0 ] — NEURAL PROMPT ENHANCEMENT PROTOCOL
-            </p>
-        </div>
-    """)
-    
-    # 이벤트 핸들러
-    generate_btn.click(
-        fn=process_comparison,
-        inputs=[prompt_input, fireworks_key_input, fal_key_input, aspect_ratio, resolution],
-        outputs=[status_text, enhanced_output, original_image_output, enhanced_image_output]
-    )
-
+    def do_logout(token):
+        logout_user(token)
+        return "", "로그인하지 않음", gr.update(visible=False), "✅ 로그아웃됨"
+    
+    def load_account_info(token):
+        user = validate_session(token)
+        if not user:
+            return "로그인이 필요합니다.", []
+        info = f"### 👤 계정 정보\n- **이메일**: {user['email']}\n- **등급**: {'관리자' if user['is_admin'] else '일반'}\n- **일일 제한**: {user['daily_limit']}회\n- **오늘 사용량**: {get_daily_usage(user['id'])}회"
+        history = get_user_generations(user["id"])
+        history_data = [[h["original_prompt"][:50]+"...", h["created_at"], h["status"]] for h in history]
+        return info, history_data
+    
+    def load_admin_panel(token):
+        user = validate_session(token)
+        if not user or not user["is_admin"]:
+            return "❌ 관리자 권한이 필요합니다.", gr.update(visible=False), gr.update(visible=False), "", []
+        stats = get_stats()
+        stats_md = f"| 지표 | 값 |\n|---|---|\n| 총 사용자 | {stats['total_users']} |\n| 오늘 활성 | {stats['active_today']} |\n| 총 생성 | {stats['total_generations']} |\n| 오늘 생성 | {stats['generations_today']} |"
+        users = get_all_users()
+        users_data = [[u["id"], u["email"], "✅" if u["is_admin"] else "", "✅" if u["is_active"] else "❌", u["daily_limit"], u["created_at"]] for u in users]
+        return "✅ 관리자 패널", gr.update(visible=True), gr.update(visible=True), stats_md, users_data
+    
+    def toggle_user_active(token, user_id):
+        user = validate_session(token)
+        if not user or not user["is_admin"]:
+            return "❌ 권한 없음"
+        conn = get_db()
+        cursor = conn.cursor()
+        cursor.execute("UPDATE users SET is_active = NOT is_active WHERE id = ?", (int(user_id),))
+        conn.commit()
+        conn.close()
+        return "✅ 상태 변경됨"
+    
+    def change_user_limit(token, user_id, new_limit):
+        user = validate_session(token)
+        if not user or not user["is_admin"]:
+            return "❌ 권한 없음"
+        update_user_limit(int(user_id), int(new_limit))
+        return f"✅ 제한 변경됨: {int(new_limit)}"
+
+    # 이벤트 바인딩
+    login_btn.click(do_login, [login_email, login_password], [session_token, login_status, current_user_display, logout_btn])
+    reg_btn.click(do_register, [reg_email, reg_password, reg_confirm], [session_token, reg_status, current_user_display, logout_btn])
+    logout_btn.click(do_logout, [session_token], [session_token, current_user_display, logout_btn, login_status])
+    generate_btn.click(process_comparison_with_auth, [prompt_input, session_token, fireworks_key_input, fal_key_input, aspect_ratio, resolution], [status_text, enhanced_output, original_image_output, enhanced_image_output, usage_display])
+    refresh_history_btn.click(load_account_info, [session_token], [account_info, history_display])
+    refresh_stats_btn.click(load_admin_panel, [session_token], [admin_auth_status, admin_panel, admin_actions, stats_display, users_table])
+    refresh_users_btn.click(load_admin_panel, [session_token], [admin_auth_status, admin_panel, admin_actions, stats_display, users_table])
+    toggle_active_btn.click(toggle_user_active, [session_token, user_id_input], [admin_action_status])
+    update_limit_btn.click(change_user_limit, [session_token, user_id_input, new_limit], [admin_action_status])
+    
+    # 탭 선택 시 자동 로드
+    admin_tab.select(load_admin_panel, [session_token], [admin_auth_status, admin_panel, admin_actions, stats_display, users_table])
+    account_tab.select(load_account_info, [session_token], [account_info, history_display])
 
 if __name__ == "__main__":
     demo.launch()