Update app.py

app.py

@@ -1,539 +1,800 @@
-import gradio as gr
-import requests
-import io
-import random
-import os
-import time
-from PIL import Image
-import json
-
-# Get API token from environment variable 
-HF_TOKEN = os.getenv("HF_TOKEN")
-if not HF_TOKEN:
-    raise ValueError("HF_TOKEN environment variable is not set")
-
-def query(
-    prompt,
-    model,
-    custom_lora,
-    negative_prompt="",       # ← 기존 is_negative=False → negative_prompt="" 로 변경
-    steps=35,
-    cfg_scale=7,
-    sampler="DPM++ 2M Karras",
-    seed=-1,
-    strength=0.7,
-    width=1024,
-    height=1024
-):
-    print("Starting query function...")
-    
-    if not prompt:
-        raise gr.Error("Prompt cannot be empty")
-
-    # Set headers with API token
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    
-    # Generate a unique key for tracking
-    key = random.randint(0, 999)
-    
-    # Enhance prompt
-    prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
-    print(f'Generation {key}: {prompt}')
+from flask import Flask, render_template, request, jsonify
+import os, re, json, sqlite3
 
+app = Flask(__name__)
+
+# ────────────────────────── 1. CONFIGURATION ──────────────────────────
+DB_FILE = "favorite_sites.json"  # JSON file for backward compatibility
+SQLITE_DB = "favorite_sites.db"  # SQLite database for persistence
+
+# Domains that commonly block iframes
+BLOCKED_DOMAINS = [
+    "naver.com", "daum.net", "google.com",
+    "facebook.com", "instagram.com", "kakao.com",
+    "ycombinator.com"
+]
+
+# ────────────────────────── 2. CURATED CATEGORIES ──────────────────────────
+CATEGORIES = {
+    "Productivity": [
+        "https://huggingface.co/spaces/ginipick/AI-BOOK",
+        "https://huggingface.co/spaces/ginigen/perflexity-clone",
+        "https://huggingface.co/spaces/ginipick/IDEA-DESIGN",
+        "https://huggingface.co/spaces/VIDraft/mouse-webgen",
+        "https://huggingface.co/spaces/openfree/Vibe-Game",
+        "https://huggingface.co/spaces/openfree/Game-Gallery",
+        "https://huggingface.co/spaces/aiqtech/Contributors-Leaderboard",
+        "https://huggingface.co/spaces/fantaxy/Model-Leaderboard",
+        "https://huggingface.co/spaces/fantaxy/Space-Leaderboard",
+        "https://huggingface.co/spaces/openfree/Korean-Leaderboard",
+    ],
+    "Multimodal": [
+        "https://huggingface.co/spaces/openfree/DreamO-video",
+        "https://huggingface.co/spaces/Heartsync/NSFW-Uncensored-photo",
+        "https://huggingface.co/spaces/Heartsync/NSFW-Uncensored",
+        "https://huggingface.co/spaces/fantaxy/Sound-AI-SFX",
+        "https://huggingface.co/spaces/ginigen/SFX-Sound-magic",
+        "https://huggingface.co/spaces/ginigen/VoiceClone-TTS",
+        "https://huggingface.co/spaces/aiqcamp/MCP-kokoro",
+        "https://huggingface.co/spaces/aiqcamp/ENGLISH-Speaking-Scoring",
+    ],
+    "Professional": [
+        "https://huggingface.co/spaces/ginigen/blogger",
+        "https://huggingface.co/spaces/VIDraft/money-radar",
+        "https://huggingface.co/spaces/immunobiotech/drug-discovery",
+        "https://huggingface.co/spaces/immunobiotech/Gemini-MICHELIN",
+        "https://huggingface.co/spaces/Heartsync/Papers-Leaderboard",
+        "https://huggingface.co/spaces/VIDraft/PapersImpact",
+        "https://huggingface.co/spaces/ginipick/AgentX-Papers",
+        "https://huggingface.co/spaces/openfree/Cycle-Navigator",
+    ],
+    "Image": [
+        "https://huggingface.co/spaces/ginigen/interior-design",
+        "https://huggingface.co/spaces/ginigen/Workflow-Canvas",
+        "https://huggingface.co/spaces/ginigen/Multi-LoRAgen",
+        "https://huggingface.co/spaces/ginigen/Every-Text",
+        "https://huggingface.co/spaces/ginigen/text3d-r1",
+        "https://huggingface.co/spaces/ginipick/FLUXllama",
+        "https://huggingface.co/spaces/Heartsync/FLUX-Vision",
+        "https://huggingface.co/spaces/ginigen/VisualCloze",
+        "https://huggingface.co/spaces/seawolf2357/Ghibli-Multilingual-Text-rendering",
+        "https://huggingface.co/spaces/ginigen/Ghibli-Meme-Studio",
+        "https://huggingface.co/spaces/VIDraft/Open-Meme-Studio",
+        "https://huggingface.co/spaces/ginigen/3D-LLAMA",
+    ],
+    "LLM / VLM": [
+        "https://huggingface.co/spaces/VIDraft/Gemma-3-R1984-4B",
+        "https://huggingface.co/spaces/VIDraft/Gemma-3-R1984-12B",
+        "https://huggingface.co/spaces/ginigen/Mistral-Perflexity",
+        "https://huggingface.co/spaces/aiqcamp/gemini-2.5-flash-preview",
+        "https://huggingface.co/spaces/openfree/qwen3-30b-a3b-research",
+        "https://huggingface.co/spaces/openfree/qwen3-235b-a22b-research",
+        "https://huggingface.co/spaces/openfree/Llama-4-Maverick-17B-Research",
+    ],
+}
+
+# ────────────────────────── 3. DATABASE FUNCTIONS ──────────────────────────
+def init_db():
+    # Initialize JSON file if it doesn't exist
+    if not os.path.exists(DB_FILE):
+        with open(DB_FILE, "w", encoding="utf-8") as f:
+            json.dump([], f, ensure_ascii=False)
+    
+    # Initialize SQLite database
+    conn = sqlite3.connect(SQLITE_DB)
+    cursor = conn.cursor()
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS urls (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        url TEXT UNIQUE NOT NULL,
+        date_added TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+    )
+    ''')
+    conn.commit()
+    
+    # If we have data in JSON but not in SQLite (first run with new SQLite DB),
+    # migrate the data from JSON to SQLite
+    json_urls = load_json()
+    if json_urls:
+        db_urls = load_db_sqlite()
+        for url in json_urls:
+            if url not in db_urls:
+                add_url_to_sqlite(url)
+    
+    conn.close()
+
+def load_json():
+    """Load URLs from JSON file (for backward compatibility)"""
     try:
-        # Set API URL based on model selection
-        if custom_lora.strip():
-            API_URL = f"https://api-inference.huggingface.co/models/{custom_lora.strip()}"
+        with open(DB_FILE, "r", encoding="utf-8") as f:
+            raw = json.load(f)
+        return raw if isinstance(raw, list) else []
+    except Exception:
+        return []
+
+def save_json(lst):
+    """Save URLs to JSON file (for backward compatibility)"""
+    try:
+        with open(DB_FILE, "w", encoding="utf-8") as f:
+            json.dump(lst, f, ensure_ascii=False, indent=2)
+        return True
+    except Exception:
+        return False
+
+def load_db_sqlite():
+    """Load URLs from SQLite database"""
+    conn = sqlite3.connect(SQLITE_DB)
+    cursor = conn.cursor()
+    cursor.execute("SELECT url FROM urls ORDER BY date_added DESC")
+    urls = [row[0] for row in cursor.fetchall()]
+    conn.close()
+    return urls
+
+def add_url_to_sqlite(url):
+    """Add a URL to SQLite database"""
+    conn = sqlite3.connect(SQLITE_DB)
+    cursor = conn.cursor()
+    try:
+        cursor.execute("INSERT INTO urls (url) VALUES (?)", (url,))
+        conn.commit()
+        success = True
+    except sqlite3.IntegrityError:
+        # URL already exists
+        success = False
+    conn.close()
+    return success
+
+def update_url_in_sqlite(old_url, new_url):
+    """Update a URL in SQLite database"""
+    conn = sqlite3.connect(SQLITE_DB)
+    cursor = conn.cursor()
+    try:
+        cursor.execute("UPDATE urls SET url = ? WHERE url = ?", (new_url, old_url))
+        if cursor.rowcount > 0:
+            conn.commit()
+            success = True
         else:
-            if model == 'Stable Diffusion XL':
-                API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-            elif model == 'FLUX.1 [Dev]':
-                API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
-            elif model == 'FLUX.1 [Schnell]':
-                API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
-            elif model == 'Flux Logo Design':
-                API_URL = "https://api-inference.huggingface.co/models/Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design"
-                prompt = f"wablogo, logo, Minimalist, {prompt}"
-            elif model == 'Flux Uncensored':
-                API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-uncensored"
-            elif model == 'Flux Uncensored V2':
-                API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-Uncensored-V2"
-            elif model == 'Flux Tarot Cards':
-                API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA"
-                prompt = f"Tarot card, {prompt}"
-            elif model == 'Pixel Art Sprites':
-                API_URL = "https://api-inference.huggingface.co/models/sWizad/pokemon-trainer-sprites-pixelart-flux"
-                prompt = f"a pixel image, {prompt}"
-            elif model == '3D Sketchfab':
-                API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA"
-                prompt = f"3D Sketchfab, {prompt}"
-            elif model == 'Retro Comic Flux':
-                API_URL = "https://api-inference.huggingface.co/models/renderartist/retrocomicflux"
-                prompt = f"c0m1c, comic book panel, {prompt}"
-            elif model == 'Caricature':
-                API_URL = "https://api-inference.huggingface.co/models/TheAwakenOne/caricature"
-                prompt = f"CCTUR3, {prompt}"
-            elif model == 'Huggieverse':
-                API_URL = "https://api-inference.huggingface.co/models/Chunte/flux-lora-Huggieverse"
-                prompt = f"HGGRE, {prompt}"
-            elif model == 'Propaganda Poster':
-                API_URL = "https://api-inference.huggingface.co/models/AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion"
-                prompt = f"propaganda poster, {prompt}"
-            elif model == 'Flux Game Assets V2':
-                API_URL = "https://api-inference.huggingface.co/models/gokaygokay/Flux-Game-Assets-LoRA-v2"
-                prompt = f"wbgmsst, white background, {prompt}"
-            elif model == 'SoftPasty Flux':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/softpasty-flux-dev"
-                prompt = f"araminta_illus illustration style, {prompt}"
-            elif model == 'Flux Stickers':
-                API_URL = "https://api-inference.huggingface.co/models/diabolic6045/Flux_Sticker_Lora"
-                prompt = f"5t1cker 5ty1e, {prompt}"
-            elif model == 'Flux Animex V2':
-                API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Animex-v2-LoRA"
-                prompt = f"Animex, {prompt}"
-            elif model == 'Flux Animeo V1':
-                API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Animeo-v1-LoRA"
-                prompt = f"Animeo, {prompt}"
-            elif model == 'Movie Board':
-                API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux.1-Dev-Movie-Boards-LoRA"
-                prompt = f"movieboard, {prompt}"
-            elif model == 'Purple Dreamy':
-                API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Purple-Dreamy-Flux-LoRA"
-                prompt = f"Purple Dreamy, {prompt}"
-            elif model == 'PS1 Style Flux':
-                API_URL = "https://api-inference.huggingface.co/models/veryVANYA/ps1-style-flux"
-                prompt = f"ps1 game screenshot, {prompt}"
-            elif model == 'Softserve Anime':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/softserve_anime"
-                prompt = f"sftsrv style illustration, {prompt}"
-            elif model == 'Flux Tarot v1':
-                API_URL = "https://api-inference.huggingface.co/models/multimodalart/flux-tarot-v1"
-                prompt = f"in the style of TOK a trtcrd tarot style, {prompt}"
-            elif model == 'Half Illustration':
-                API_URL = "https://api-inference.huggingface.co/models/davisbro/half_illustration"
-                prompt = f"in the style of TOK, {prompt}"
-            elif model == 'OpenDalle v1.1':
-                API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalleV1.1"
-            elif model == 'Flux Ghibsky Illustration':
-                API_URL = "https://api-inference.huggingface.co/models/aleksa-codes/flux-ghibsky-illustration"
-                prompt = f"GHIBSKY style, {prompt}"
-            elif model == 'Flux Koda':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/flux-koda"
-                prompt = f"flmft style, {prompt}"
-            elif model == 'Soviet Diffusion XL':
-                API_URL = "https://api-inference.huggingface.co/models/openskyml/soviet-diffusion-xl"
-                prompt = f"soviet poster, {prompt}"
-            elif model == 'Flux Realism LoRA':
-                API_URL = "https://api-inference.huggingface.co/models/XLabs-AI/flux-RealismLora"
-            elif model == 'Frosting Lane Flux':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/frosting_lane_flux"
-                prompt = f"frstingln illustration, {prompt}"
-            elif model == 'Phantasma Anime':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/phantasma-anime"
-            elif model == 'Boreal':
-                API_URL = "https://api-inference.huggingface.co/models/kudzueye/Boreal"
-                prompt = f"photo, {prompt}"
-            elif model == 'How2Draw':
-                API_URL = "https://api-inference.huggingface.co/models/glif/how2draw"
-                prompt = f"How2Draw, {prompt}"
-            elif model == 'Flux AestheticAnime':
-                API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/FLUX-AestheticAnime"
-            elif model == 'Fashion Hut Modeling LoRA':
-                API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Fashion-Hut-Modeling-LoRA"
-                prompt = f"Modeling of, {prompt}"
-            elif model == 'Flux SyntheticAnime':
-                API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/FLUX-SyntheticAnime"
-                prompt = f"1980s anime screengrab, VHS quality, syntheticanime, {prompt}"
-            elif model == 'Flux Midjourney Anime':
-                API_URL = "https://api-inference.huggingface.co/models/brushpenbob/flux-midjourney-anime"
-                prompt = f"egmid, {prompt}"
-            elif model == 'Coloring Book Generator':
-                API_URL = "https://api-inference.huggingface.co/models/robert123231/coloringbookgenerator"
-            elif model == 'Collage Flux':
-                API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Castor-Collage-Dim-Flux-LoRA"
-                prompt = f"collage, {prompt}"
-            elif model == 'Flux Product Ad Backdrop':
-                API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux-Product-Ad-Backdrop"
-                prompt = f"Product Ad, {prompt}"
-            elif model == 'Product Design':
-                API_URL = "https://api-inference.huggingface.co/models/multimodalart/product-design"
-                prompt = f"product designed by prdsgn, {prompt}"
-            elif model == '90s Anime Art':
-                API_URL = "https://api-inference.huggingface.co/models/glif/90s-anime-art"
-            elif model == 'Brain Melt Acid Art':
-                API_URL = "https://api-inference.huggingface.co/models/glif/Brain-Melt-Acid-Art"
-                prompt = f"maximalism, in an acid surrealism style, {prompt}"
-            elif model == 'Lustly Flux Uncensored v1':
-                API_URL = "https://api-inference.huggingface.co/models/lustlyai/Flux_Lustly.ai_Uncensored_nsfw_v1"
-            elif model == 'NSFW Master Flux':
-                API_URL = "https://api-inference.huggingface.co/models/Keltezaa/NSFW_MASTER_FLUX"
-                prompt = f"NSFW, {prompt}"
-            elif model == 'Flux Outfit Generator':
-                API_URL = "https://api-inference.huggingface.co/models/tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator"
-            elif model == 'Midjourney':
-                API_URL = "https://api-inference.huggingface.co/models/Jovie/Midjourney"
-            elif model == 'DreamPhotoGASM':
-                API_URL = "https://api-inference.huggingface.co/models/Yntec/DreamPhotoGASM"
-            elif model == 'Flux Super Realism LoRA':
-                API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Super-Realism-LoRA"
-            elif model == 'Stable Diffusion 2-1':
-                API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1-base"
-            elif model == 'Stable Diffusion 3.5 Large':
-                API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
-            elif model == 'Stable Diffusion 3.5 Large Turbo':
-                API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large-turbo"
-            elif model == 'Stable Diffusion 3 Medium':
-                API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3-medium-diffusers"
-                prompt = f"A, {prompt}"
-            elif model == 'Duchaiten Real3D NSFW XL':
-                API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/duchaiten-real3d-nsfw-xl"
-            elif model == 'Pixel Art XL':
-                API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
-                prompt = f"pixel art, {prompt}"
-            elif model == 'Character Design':
-                API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/character-design"
-                prompt = f"Character Design, {prompt}"
-            elif model == 'Sketched Out Manga':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/sketchedoutmanga"
-                prompt = f"daiton, {prompt}"
-            elif model == 'Archfey Anime':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/archfey_anime"
-            elif model == 'Lofi Cuties':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/lofi-cuties"
-            elif model == 'YiffyMix':
-                API_URL = "https://api-inference.huggingface.co/models/Yntec/YiffyMix"
-            elif model == 'Analog Madness Realistic v7':
-                API_URL = "https://api-inference.huggingface.co/models/digiplay/AnalogMadness-realistic-model-v7"
-            elif model == 'Selfie Photography':
-                API_URL = "https://api-inference.huggingface.co/models/artificialguybr/selfiephotographyredmond-selfie-photography-lora-for-sdxl"
-                prompt = f"instagram model, discord profile picture, {prompt}"
-            elif model == 'Filmgrain':
-                API_URL = "https://api-inference.huggingface.co/models/artificialguybr/filmgrain-redmond-filmgrain-lora-for-sdxl"
-                prompt = f"Film Grain, FilmGrainAF, {prompt}"
-            elif model == 'Leonardo AI Style Illustration':
-                API_URL = "https://api-inference.huggingface.co/models/goofyai/Leonardo_Ai_Style_Illustration"
-                prompt = f"leonardo style, illustration, vector art, {prompt}"
-            elif model == 'Cyborg Style XL':
-                API_URL = "https://api-inference.huggingface.co/models/goofyai/cyborg_style_xl"
-                prompt = f"cyborg style, {prompt}"
-            elif model == 'Little Tinies':
-                API_URL = "https://api-inference.huggingface.co/models/alvdansen/littletinies"
-            elif model == 'NSFW XL':
-                API_URL = "https://api-inference.huggingface.co/models/Dremmar/nsfw-xl"
-            elif model == 'Analog Redmond':
-                API_URL = "https://api-inference.huggingface.co/models/artificialguybr/analogredmond"
-                prompt = f"timeless style, {prompt}"
-            elif model == 'Pixel Art Redmond':
-                API_URL = "https://api-inference.huggingface.co/models/artificialguybr/PixelArtRedmond"
-                prompt = f"Pixel Art, {prompt}"
-            elif model == 'Ascii Art':
-                API_URL = "https://api-inference.huggingface.co/models/CiroN2022/ascii-art"
-                prompt = f"ascii art, {prompt}"
-            elif model == 'Analog':
-                API_URL = "https://api-inference.huggingface.co/models/Yntec/Analog"
-            elif model == 'Maple Syrup':
-                API_URL = "https://api-inference.huggingface.co/models/Yntec/MapleSyrup"
-            elif model == 'Perfect Lewd Fantasy':
-                API_URL = "https://api-inference.huggingface.co/models/digiplay/perfectLewdFantasy_v1.01"
-            elif model == 'AbsoluteReality 1.8.1':
-                API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
-            elif model == 'Disney':
-                API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
-                prompt = f"Disney style, {prompt}"
-            elif model == 'Redmond SDXL':
-                API_URL = "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
-            elif model == 'epiCPhotoGasm':
-                API_URL = "https://api-inference.huggingface.co/models/Yntec/epiCPhotoGasm"
-            else:
-                API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-
-        # Prepare payload in Hugging Face Inference API style
-        # (negative_prompt, steps, cfg_scale, seed, strength 등은 parameters 안에 배치)
-        payload = {
-            "inputs": prompt,
-            "parameters": {
-                "negative_prompt": negative_prompt,
-                "num_inference_steps": steps,
-                "guidance_scale": cfg_scale,
-                "width": width,
-                "height": height,
-                "strength": strength,
-                # seed를 지원하는 모델/엔드포인트에 따라 무시될 수도 있음
-                "seed": seed if seed != -1 else random.randint(1, 1000000000),
-            },
-            # 모델이 로딩 중일 경우 기다리도록 설정
-            "options": {"wait_for_model": True}
-        }
+            success = False
+    except sqlite3.IntegrityError:
+        # New URL already exists
+        success = False
+    conn.close()
+    return success
 
-        # Improved retry logic with exponential backoff
-        max_retries = 3
-        current_retry = 0
-        backoff_factor = 2  # Exponential backoff
-        
-        while current_retry < max_retries:
-            try:
-                response = requests.post(API_URL, headers=headers, json=payload, timeout=180)
-                
-                # 디버깅용 정보 출력
-                print("Response Content-Type:", response.headers.get("Content-Type"))
-                print("Response Text (snippet):", response.text[:500])
-
-                response.raise_for_status()  # HTTP 에러 코드 시 예외 발생
-                image = Image.open(io.BytesIO(response.content))
-                
-                print(f'Generation {key} completed successfully')
-                return image
-                
-            except (requests.exceptions.Timeout,
-                    requests.exceptions.ConnectionError, 
-                    requests.exceptions.HTTPError,
-                    requests.exceptions.RequestException) as e:
-                current_retry += 1
-                if current_retry < max_retries:
-                    wait_time = backoff_factor ** current_retry  # Exponential backoff
-                    print(f"Network error occurred: {str(e)}. Retrying in {wait_time} seconds... (Attempt {current_retry + 1}/{max_retries})")
-                    time.sleep(wait_time)
-                    continue
-                else:
-                    # Detailed error message based on exception type
-                    if isinstance(e, requests.exceptions.Timeout):
-                        error_msg = f"Request timed out after {max_retries} attempts. The model might be busy, please try again later."
-                    elif isinstance(e, requests.exceptions.ConnectionError):
-                        error_msg = f"Connection error after {max_retries} attempts. Please check your network connection."
-                    elif isinstance(e, requests.exceptions.HTTPError):
-                        status_code = e.response.status_code if hasattr(e, 'response') and e.response is not None else "unknown"
-                        error_msg = f"HTTP error (status code: {status_code}) after {max_retries} attempts."
-                    else:
-                        error_msg = f"Request failed after {max_retries} attempts: {str(e)}"
-                    
-                    raise gr.Error(error_msg)
-        
-    except Exception as e:
-        error_message = f"Unexpected error: {str(e)}"
-        if isinstance(e, requests.exceptions.RequestException) and hasattr(e, 'response') and e.response is not None:
-            if e.response.status_code == 401:
-                error_message = "Invalid API token. Please check your Hugging Face API token."
-            elif e.response.status_code == 403:
-                error_message = "Access denied. Please check your API token permissions."
-            elif e.response.status_code == 503:
-                error_message = "Model is currently loading. Please try again in a few moments."
-        raise gr.Error(error_message)
-
-
-def generate_grid(prompt, selected_models, custom_lora, negative_prompt, steps, cfg_scale, seed, strength, width, height, progress=gr.Progress()):
-    if len(selected_models) > 4:
-        raise gr.Error("Please select up to 4 models")
-    if len(selected_models) == 0:
-        raise gr.Error("Please select at least 1 model")
-    
-    # Initialize image array
-    images = [None] * 4
-    total_models = len(selected_models[:4])
-    
-    def update_gallery():
-        # Only include non-None images for gallery update
-        return [img for img in images if img is not None]
-    
-    # Create placeholder for failed models
-    placeholder_image = None
-    
-    # Generate image for each model
-    for idx, model_name in enumerate(selected_models[:4]):
-        try:
-            progress((idx + 1) / total_models, f"Generating image for {model_name}...")
-            img = query(prompt, model_name, custom_lora, negative_prompt, steps, cfg_scale, seed, strength, width, height)
-            images[idx] = img
-            
-            # If this is the first successful generation, save as placeholder for failed models
-            if placeholder_image is None:
-                placeholder_image = img
-                
-            # Update gallery after each successful generation
-            yield update_gallery()
-        except Exception as e:
-            print(f"Error generating image for {model_name}: {str(e)}")
-            # Keep the slot as None and continue with next model
-            continue
-    
-    # Fill empty slots with a placeholder (either the last successful image or a blank image)
-    if placeholder_image:
-        for i in range(len(images)):
-            if images[i] is None:
-                # Create a copy of placeholder to avoid reference issues
-                images[i] = placeholder_image.copy() 
+def delete_url_from_sqlite(url):
+    """Delete a URL from SQLite database"""
+    conn = sqlite3.connect(SQLITE_DB)
+    cursor = conn.cursor()
+    cursor.execute("DELETE FROM urls WHERE url = ?", (url,))
+    if cursor.rowcount > 0:
+        conn.commit()
+        success = True
     else:
-        # If all models failed, create a blank image with error text
-        for i in range(len(images)):
-            blank_img = Image.new('RGB', (width, height), color=(240, 240, 240))
-            images[i] = blank_img
+        success = False
+    conn.close()
+    return success
+
+def load_db():
+    """Primary function to load URLs - prioritizes SQLite DB but falls back to JSON"""
+    urls = load_db_sqlite()
+    if not urls:
+        # If SQLite DB is empty, try loading from JSON
+        urls = load_json()
+        # If we found URLs in JSON, migrate them to SQLite
+        for url in urls:
+            add_url_to_sqlite(url)
+    return urls
+
+def save_db(lst):
+    """Save URLs to both SQLite and JSON"""
+    # Get existing URLs from SQLite for comparison
+    existing_urls = load_db_sqlite()
+    
+    # Clear all URLs from SQLite and add the new list
+    conn = sqlite3.connect(SQLITE_DB)
+    cursor = conn.cursor()
+    cursor.execute("DELETE FROM urls")
+    for url in lst:
+        cursor.execute("INSERT INTO urls (url) VALUES (?)", (url,))
+    conn.commit()
+    conn.close()
     
-    progress(1.0, "Generation complete!")
-    yield update_gallery()
+    # Also save to JSON for backward compatibility
+    return save_json(lst)
 
+# ────────────────────────── 4. URL HELPERS ──────────────────────────
+def direct_url(hf_url):
+    m = re.match(r"https?://huggingface\.co/spaces/([^/]+)/([^/?#]+)", hf_url)
+    if not m:
+        return hf_url
+    owner, name = m.groups()
+    owner = owner.lower()
+    name = name.replace('.', '-').replace('_', '-').lower()
+    return f"https://{owner}-{name}.hf.space"
 
-def check_network_connectivity():
-    """Utility function to check network connectivity to the Hugging Face API"""
+def screenshot_url(url):
+    return f"https://image.thum.io/get/fullpage/{url}"
+
+def process_url_for_preview(url):
+    """Returns (preview_url, mode)"""
+    # Handle blocked domains first
+    if any(d for d in BLOCKED_DOMAINS if d in url):
+        return screenshot_url(url), "snapshot"
+    
+    # Special case handling for problematic URLs
+    if "vibe-coding-tetris" in url or "World-of-Tank-GAME" in url or "Minesweeper-Game" in url:
+        return screenshot_url(url), "snapshot"
+    
+    # General HF space handling
     try:
-        response = requests.get("https://api-inference.huggingface.co", timeout=5)
-        if response.status_code == 200:
-            return True
-        return False
-    except:
-        return False
+        if "huggingface.co/spaces" in url:
+            parts = url.rstrip("/").split("/")
+            if len(parts) >= 5:
+                owner = parts[-2]
+                name = parts[-1]
+                embed_url = f"https://huggingface.co/spaces/{owner}/{name}/embed"
+                return embed_url, "iframe"
+    except Exception:
+        return screenshot_url(url), "snapshot"
+    
+    # Default handling
+    return url, "iframe"
+
+# ────────────────────────── 5. API ROUTES ──────────────────────────
+@app.route('/api/category')
+def api_category():
+    cat = request.args.get('name', '')
+    urls = CATEGORIES.get(cat, [])
+    
+    # Add pagination for categories as well
+    page = int(request.args.get('page', 1))
+    per_page = int(request.args.get('per_page', 4))  # Changed to 4 per page
+    
+    total_pages = max(1, (len(urls) + per_page - 1) // per_page)
+    start = (page - 1) * per_page
+    end = min(start + per_page, len(urls))
+    
+    urls_page = urls[start:end]
+    
+    items = [
+        {
+            "title": url.split('/')[-1],
+            "owner": url.split('/')[-2] if '/spaces/' in url else '',
+            "iframe": direct_url(url),
+            "shot": screenshot_url(url),
+            "hf": url
+        } for url in urls_page
+    ]
+    
+    return jsonify({
+        "items": items,
+        "page": page,
+        "total_pages": total_pages
+    })
+
+@app.route('/api/favorites')
+def api_favorites():
+    # Load URLs from SQLite database
+    urls = load_db()
+    
+    page = int(request.args.get('page', 1))
+    per_page = int(request.args.get('per_page', 4))  # Changed to 4 per page
+    
+    total_pages = max(1, (len(urls) + per_page - 1) // per_page)
+    start = (page - 1) * per_page
+    end = min(start + per_page, len(urls))
+    
+    urls_page = urls[start:end]
+    
+    result = []
+    for url in urls_page:
+        try:
+            preview_url, mode = process_url_for_preview(url)
+            result.append({
+                "title": url.split('/')[-1],
+                "url": url,
+                "preview_url": preview_url,
+                "mode": mode
+            })
+        except Exception:
+            # Fallback to screenshot mode
+            result.append({
+                "title": url.split('/')[-1],
+                "url": url,
+                "preview_url": screenshot_url(url),
+                "mode": "snapshot"
+            })
+    
+    return jsonify({
+        "items": result,
+        "page": page,
+        "total_pages": total_pages
+    })
 
+@app.route('/api/url/add', methods=['POST'])
+def add_url():
+    url = request.form.get('url', '').strip()
+    if not url:
+        return jsonify({"success": False, "message": "URL is required"})
+    
+    # SQLite에 추가 시도
+    conn = sqlite3.connect(SQLITE_DB)
+    cursor = conn.cursor()
+    try:
+        cursor.execute("INSERT INTO urls (url) VALUES (?)", (url,))
+        conn.commit()
+        success = True
+    except sqlite3.IntegrityError:
+        # URL이 이미 존재하는 경우
+        success = False
+    except Exception as e:
+        print(f"SQLite error: {str(e)}")
+        success = False
+    finally:
+        conn.close()
+    
+    if not success:
+        return jsonify({"success": False, "message": "URL already exists or could not be added"})
+    
+    # JSON 파일에도 추가 (백업용)
+    data = load_json()
+    if url not in data:
+        data.insert(0, url)
+        save_json(data)
+    
+    return jsonify({"success": True, "message": "URL added successfully"})
 
-css = """
-footer {
-    visibility: hidden;
+@app.route('/api/url/update', methods=['POST'])
+def update_url():
+    old = request.form.get('old', '')
+    new = request.form.get('new', '').strip()
+    
+    if not new:
+        return jsonify({"success": False, "message": "New URL is required"})
+    
+    # Update in SQLite DB
+    if not update_url_in_sqlite(old, new):
+        return jsonify({"success": False, "message": "URL not found or new URL already exists"})
+    
+    # Also update JSON file for backward compatibility
+    data = load_json()
+    try:
+        idx = data.index(old)
+        data[idx] = new
+        save_json(data)
+    except ValueError:
+        # If URL not in JSON, add it
+        data.append(new)
+        save_json(data)
+    
+    return jsonify({"success": True, "message": "URL updated successfully"})
+
+@app.route('/api/url/delete', methods=['POST'])
+def delete_url():
+    url = request.form.get('url', '')
+    
+    # Delete from SQLite DB
+    if not delete_url_from_sqlite(url):
+        return jsonify({"success": False, "message": "URL not found"})
+    
+    # Also update JSON file for backward compatibility
+    data = load_json()
+    try:
+        data.remove(url)
+        save_json(data)
+    except ValueError:
+        pass
+    
+    return jsonify({"success": True, "message": "URL deleted successfully"})
+
+# ────────────────────────── 6. MAIN ROUTES ──────────────────────────
+@app.route('/')
+def home():
+    os.makedirs('templates', exist_ok=True)
+    
+    with open('templates/index.html', 'w', encoding='utf-8') as fp:
+        fp.write(r'''<!DOCTYPE html>
+<html>
+<head>
+<meta charset="utf-8">
+<meta name="viewport" content="width=device-width, initial-scale=1">
+<title>Web Gallery</title>
+<style>
+@import url('https://fonts.googleapis.com/css2?family=Nunito:wght@300;600&display=swap');
+body{margin:0;font-family:Nunito,sans-serif;background:#f6f8fb;}
+.tabs{display:flex;flex-wrap:wrap;gap:8px;padding:16px;}
+.tab{padding:6px 14px;border:none;border-radius:18px;background:#e2e8f0;font-weight:600;cursor:pointer;}
+.tab.active{background:#a78bfa;color:#1a202c;}
+.tab.manage{background:#ff6e91;color:white;}
+.tab.manage.active{background:#ff2d62;color:white;}
+/* Updated grid to show 2x2 layout */
+.grid{display:grid;grid-template-columns:repeat(2,1fr);gap:20px;padding:0 16px 60px;max-width:1200px;margin:0 auto;}
+@media(max-width:800px){.grid{grid-template-columns:1fr;}}
+/* Increased card height for larger display */
+.card{background:#fff;border-radius:12px;box-shadow:0 2px 8px rgba(0,0,0,.08);overflow:hidden;height:540px;display:flex;flex-direction:column;position:relative;}
+.frame{flex:1;position:relative;overflow:hidden;}
+.frame iframe{position:absolute;width:166.667%;height:166.667%;transform:scale(.6);transform-origin:top left;border:0;}
+.frame img{width:100%;height:100%;object-fit:cover;}
+.card-label{position:absolute;top:10px;left:10px;padding:4px 8px;border-radius:4px;font-size:11px;font-weight:bold;z-index:100;text-transform:uppercase;letter-spacing:0.5px;box-shadow:0 2px 4px rgba(0,0,0,0.2);}
+.label-live{background:linear-gradient(135deg, #00c6ff, #0072ff);color:white;}
+.label-static{background:linear-gradient(135deg, #ff9a9e, #fad0c4);color:#333;}
+.foot{height:44px;background:#fafafa;display:flex;align-items:center;justify-content:center;border-top:1px solid #eee;}
+.foot a{font-size:.82rem;font-weight:700;color:#4a6dd8;text-decoration:none;}
+.pagination{display:flex;justify-content:center;margin:20px 0;gap:10px;}
+.pagination button{padding:5px 15px;border:none;border-radius:20px;background:#e2e8f0;cursor:pointer;}
+.pagination button:disabled{opacity:0.5;cursor:not-allowed;}
+.manage-panel{background:white;border-radius:12px;box-shadow:0 2px 8px rgba(0,0,0,.08);margin:16px;padding:20px;}
+.form-group{margin-bottom:15px;}
+.form-group label{display:block;margin-bottom:5px;font-weight:600;}
+.form-control{width:100%;padding:8px;border:1px solid #ddd;border-radius:4px;box-sizing:border-box;}
+.btn{padding:8px 15px;border:none;border-radius:4px;cursor:pointer;font-weight:600;}
+.btn-primary{background:#4a6dd8;color:white;}
+.btn-danger{background:#e53e3e;color:white;}
+.btn-success{background:#38a169;color:white;}
+.status{padding:10px;margin:10px 0;border-radius:4px;display:none;}
+.status.success{display:block;background:#c6f6d5;color:#22543d;}
+.status.error{display:block;background:#fed7d7;color:#822727;}
+.url-list{margin:20px 0;border:1px solid #eee;border-radius:4px;max-height:300px;overflow-y:auto;}
+.url-item{padding:10px;border-bottom:1px solid #eee;display:flex;justify-content:space-between;align-items:center;}
+.url-item:last-child{border-bottom:none;}
+.url-controls{display:flex;gap:5px;}
+</style>
+</head>
+<body>
+<header style="text-align: center; padding: 20px; background: linear-gradient(135deg, #f6f8fb, #e2e8f0); border-bottom: 1px solid #ddd;">
+  <h1 style="margin-bottom: 10px;">🌟 Web Gallery Manager</h1>
+  <p class="description" style="margin-bottom: 15px;">
+    🚀 <strong>Free AI Spaces & Website Gallery</strong> ✨ Save and manage your favorite sites! Supports <span style="background: linear-gradient(135deg, #00c6ff, #0072ff); color: white; padding: 2px 6px; border-radius: 4px; font-size: 12px;">LIVE</span> and <span style="background: linear-gradient(135deg, #ff9a9e, #fad0c4); color: #333; padding: 2px 6px; border-radius: 4px; font-size: 12px;">Static</span> preview modes.
+  </p>
+  <p style="font-size: 0.8rem; color: #888; margin-top: 10px;">
+  Your desired URL will be permanently saved when recorded in the <code>favorite_sites.json</code> file.
+  </p>
+  <p>
+    <a href="https://discord.gg/openfreeai" target="_blank"><img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%230000ff&labelColor=%23800080&logo=discord&logoColor=white&style=for-the-badge" alt="badge"></a>
+  </p>
+</header>
+<div class="tabs" id="tabs"></div>
+<div id="content"></div>
+
+<script>
+// Basic configuration
+const cats = {{cats|tojson}};
+const tabs = document.getElementById('tabs');
+const content = document.getElementById('content');
+let active = "";
+let currentPage = 1;
+
+// Simple utility functions
+function loadHTML(url, callback) {
+  const xhr = new XMLHttpRequest();
+  xhr.open('GET', url, true);
+  xhr.onreadystatechange = function() {
+    if (xhr.readyState === 4 && xhr.status === 200) {
+      callback(xhr.responseText);
+    }
+  };
+  xhr.send();
 }
-"""
-
-with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
-    gr.Markdown("# ZeroWeight Studio")
-    
-    with gr.Row():
-        with gr.Column(scale=2):
-            text_prompt = gr.Textbox(
-                label="Prompt",
-                placeholder="Describe what you want to create...",
-                lines=3
-            )
-            
-            negative_prompt = gr.Textbox(
-                label="Negative Prompt",
-                placeholder="What should not be in the image",
-                value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
-                lines=2
-            )
-
-            custom_lora = gr.Textbox(
-                label="Custom LoRA Path (Optional)",
-                placeholder="e.g., multimodalart/vintage-ads-flux",
-                lines=1
-            )
-
-        with gr.Column(scale=1):
-            with gr.Group():
-                gr.Markdown("### Image Settings")
-                width = gr.Slider(label="Width", value=1024, minimum=512, maximum=1216, step=64)
-                height = gr.Slider(label="Height", value=1024, minimum=512, maximum=1216, step=64)
-                
-            with gr.Group():
-                gr.Markdown("### Generation Parameters")
-                steps = gr.Slider(label="Steps", value=35, minimum=1, maximum=100, step=1)
-                cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=0.5)
-                strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.1)
-                seed = gr.Slider(label="Seed (-1 for random)", value=-1, minimum=-1, maximum=1000000000, step=1)
-
-            with gr.Accordion("Model Selection", open=False):
-                model_search = gr.Textbox(
-                    label="Search Models",
-                    placeholder="Type to filter models...",
-                    lines=1
-                )
-
-                # Set top 4 models as default
-                default_models = [
-                    "FLUX.1 [Schnell]",
-                    "Stable Diffusion 3.5 Large",
-                    "Stable Diffusion 3.5 Large Turbo",
-                    "Midjourney"
-                ]
-
-                # Full model list
-                models_list = [
-                    "FLUX.1 [Schnell]",
-                    "Stable Diffusion 3.5 Large",
-                    "Stable Diffusion 3.5 Large Turbo",
-                    "Stable Diffusion XL",
-                    "FLUX.1 [Dev]",
-                    "Midjourney",
-                    "DreamPhotoGASM",
-                    "Disney",
-                    "Leonardo AI Style Illustration", 
-                    "AbsoluteReality 1.8.1",
-                    "Analog Redmond",
-                    "Stable Diffusion 3 Medium",
-                    "Flux Super Realism LoRA",
-                    "Flux Realism LoRA",
-                    "Selfie Photography",
-                    "Character Design",
-                    "Pixel Art XL",
-                    "3D Sketchfab",
-                    "Flux Animex V2",
-                    "Flux Animeo V1",
-                    "Flux AestheticAnime",
-                    "90s Anime Art",
-                    "Softserve Anime",
-                    "Brain Melt Acid Art",
-                    "Retro Comic Flux",
-                    "Purple Dreamy",
-                    "SoftPasty Flux",
-                    "Flux Logo Design",
-                    "Product Design",
-                    "Propaganda Poster",
-                    "Movie Board",
-                    "Collage Flux"
-                ]
-
-                model = gr.Checkboxgroup(
-                    label="Select Models (Choose up to 4)",
-                    choices=models_list,
-                    value=default_models,
-                    interactive=True
-                )
-
-    with gr.Row():
-        generate_btn = gr.Button("Generate 2x2 Grid", variant="primary", size="lg")
-        
-    # Add network status indicator
-    network_status = gr.Markdown("", elem_id="network_status")
-    
-    # Function to check and update network status
-    def update_network_status():
-        if check_network_connectivity():
-            return "✅ Connected to Hugging Face API"
-        else:
-            return "❌ No connection to Hugging Face API. Please check your network."
-
-    with gr.Row():
-        gallery = gr.Gallery(
-            label="Generated Images",
-            show_label=True,
-            elem_id="gallery",
-            columns=2,
-            rows=2,
-            height="auto",
-            preview=True,
-        )
-
-    # Event handlers
-    generate_btn.click(
-        fn=generate_grid,
-        inputs=[
-            text_prompt,
-            model,
-            custom_lora,
-            negative_prompt,
-            steps,
-            cfg,
-            seed,
-            strength,
-            width,
-            height
-        ],
-        outputs=gallery,
-        show_progress=True
-    )
+
+function makeRequest(url, method, data, callback) {
+  const xhr = new XMLHttpRequest();
+  xhr.open(method, url, true);
+  xhr.onreadystatechange = function() {
+    if (xhr.readyState === 4 && xhr.status === 200) {
+      callback(JSON.parse(xhr.responseText));
+    }
+  };
+  if (method === 'POST') {
+    xhr.send(data);
+  } else {
+    xhr.send();
+  }
+}
+
+function updateTabs() {
+  Array.from(tabs.children).forEach(b => {
+    b.classList.toggle('active', b.dataset.c === active);
+  });
+}
+
+// Tab handlers
+function loadCategory(cat, page) {
+  if(cat === active && currentPage === page) return;
+  active = cat;
+  currentPage = page || 1;
+  updateTabs();
+  
+  content.innerHTML = '<p style="text-align:center;padding:40px">Loading…</p>';
+  
+  makeRequest('/api/category?name=' + encodeURIComponent(cat) + '&page=' + currentPage + '&per_page=4', 'GET', null, function(data) {
+    let html = '<div class="grid">';
+    
+    if(data.items.length === 0) {
+      html += '<p style="grid-column:1/-1;text-align:center;padding:40px">No items in this category.</p>';
+    } else {
+      data.items.forEach(item => {
+        html += `
+          <div class="card">
+            <div class="card-label label-live">LIVE</div>
+            <div class="frame">
+              <iframe src="${item.iframe}" loading="lazy" sandbox="allow-forms allow-modals allow-popups allow-same-origin allow-scripts allow-downloads"></iframe>
+            </div>
+            <div class="foot">
+              <a href="${item.hf}" target="_blank">${item.title}</a>
+            </div>
+          </div>
+        `;
+      });
+    }
+    
+    html += '</div>';
+    
+    // Add pagination
+    html += `
+      <div class="pagination">
+        <button ${currentPage <= 1 ? 'disabled' : ''} onclick="loadCategory('${cat}', ${currentPage-1})">« Previous</button>
+        <span>Page ${currentPage} of ${data.total_pages}</span>
+        <button ${currentPage >= data.total_pages ? 'disabled' : ''} onclick="loadCategory('${cat}', ${currentPage+1})">Next »</button>
+      </div>
+    `;
+    
+    content.innerHTML = html;
+  });
+}
+
+function loadFavorites(page) {
+  if(active === 'Favorites' && currentPage === page) return;
+  active = 'Favorites';
+  currentPage = page || 1;
+  updateTabs();
+  
+  content.innerHTML = '<p style="text-align:center;padding:40px">Loading…</p>';
+  
+  makeRequest('/api/favorites?page=' + currentPage + '&per_page=4', 'GET', null, function(data) {
+    let html = '<div class="grid">';
+    
+    if(data.items.length === 0) {
+      html += '<p style="grid-column:1/-1;text-align:center;padding:40px">No favorites saved yet.</p>';
+    } else {
+      data.items.forEach(item => {
+        if(item.mode === 'snapshot') {
+          html += `
+            <div class="card">
+              <div class="card-label label-static">Static</div>
+              <div class="frame">
+                <img src="${item.preview_url}" loading="lazy">
+              </div>
+              <div class="foot">
+                <a href="${item.url}" target="_blank">${item.title}</a>
+              </div>
+            </div>
+          `;
+        } else {
+          html += `
+            <div class="card">
+              <div class="card-label label-live">LIVE</div>
+              <div class="frame">
+                <iframe src="${item.preview_url}" loading="lazy" sandbox="allow-forms allow-modals allow-popups allow-same-origin allow-scripts allow-downloads"></iframe>
+              </div>
+              <div class="foot">
+                <a href="${item.url}" target="_blank">${item.title}</a>
+              </div>
+            </div>
+          `;
+        }
+      });
+    }
+    
+    html += '</div>';
     
-    def filter_models(search_term):
-        filtered_models = [m for m in models_list if search_term.lower() in m.lower()]
-        return gr.update(choices=filtered_models, value=[])
+    // Add pagination
+    html += `
+      <div class="pagination">
+        <button ${currentPage <= 1 ? 'disabled' : ''} onclick="loadFavorites(${currentPage-1})">« Previous</button>
+        <span>Page ${currentPage} of ${data.total_pages}</span>
+        <button ${currentPage >= data.total_pages ? 'disabled' : ''} onclick="loadFavorites(${currentPage+1})">Next »</button>
+      </div>
+    `;
+    
+    content.innerHTML = html;
+  });
+}
+
+function loadManage() {
+  if(active === 'Manage') return;
+  active = 'Manage';
+  updateTabs();
+  
+  content.innerHTML = `
+    <div class="manage-panel">
+      <h2>Add New URL</h2>
+      <div class="form-group">
+        <label for="new-url">URL</label>
+        <input type="text" id="new-url" class="form-control" placeholder="https://example.com">
+      </div>
+      <button onclick="addUrl()" class="btn btn-primary">Add URL</button>
+      <div id="add-status" class="status"></div>
+      
+      <h2>Manage Saved URLs</h2>
+      <div id="url-list" class="url-list">Loading...</div>
+    </div>
+  `;
+  
+  loadUrlList();
+}
+
+// URL management functions
+function loadUrlList() {
+  makeRequest('/api/favorites?per_page=100', 'GET', null, function(data) {
+    const urlList = document.getElementById('url-list');
+    
+    if(data.items.length === 0) {
+      urlList.innerHTML = '<p style="text-align:center;padding:20px">No URLs saved yet.</p>';
+      return;
+    }
+    
+    let html = '';
+    data.items.forEach(item => {
+      // Escape the URL to prevent JavaScript injection when used in onclick handlers
+      const escapedUrl = item.url.replace(/'/g, "\\'");
+      
+      html += `
+        <div class="url-item">
+          <div>${item.url}</div>
+          <div class="url-controls">
+            <button class="btn" onclick="editUrl('${escapedUrl}')">Edit</button>
+            <button class="btn btn-danger" onclick="deleteUrl('${escapedUrl}')">Delete</button>
+          </div>
+        </div>
+      `;
+    });
+    
+    urlList.innerHTML = html;
+  });
+}
+
+function addUrl() {
+  const url = document.getElementById('new-url').value.trim();
+  
+  if(!url) {
+    showStatus('add-status', 'Please enter a URL', false);
+    return;
+  }
+  
+  const formData = new FormData();
+  formData.append('url', url);
+  
+  makeRequest('/api/url/add', 'POST', formData, function(data) {
+    showStatus('add-status', data.message, data.success);
+    if(data.success) {
+      document.getElementById('new-url').value = '';
+      loadUrlList();
+      // If currently in Favorites tab, reload to see changes immediately
+      if(active === 'Favorites') {
+        loadFavorites(currentPage);
+      }
+    }
+  });
+}
+
+function editUrl(url) {
+  // Decode URL if it was previously escaped
+  const decodedUrl = url.replace(/\\'/g, "'");
+  const newUrl = prompt('Edit URL:', decodedUrl);
+  
+  if(!newUrl || newUrl === decodedUrl) return;
+  
+  const formData = new FormData();
+  formData.append('old', decodedUrl);
+  formData.append('new', newUrl);
+  
+  makeRequest('/api/url/update', 'POST', formData, function(data) {
+    if(data.success) {
+      loadUrlList();
+      // If currently in Favorites tab, reload to see changes immediately
+      if(active === 'Favorites') {
+        loadFavorites(currentPage);
+      }
+    } else {
+      alert(data.message);
+    }
+  });
+}
 
-    model_search.change(filter_models, inputs=model_search, outputs=model)
+function deleteUrl(url) {
+  // Decode URL if it was previously escaped
+  const decodedUrl = url.replace(/\\'/g, "'");
+  if(!confirm('Are you sure you want to delete this URL?')) return;
+  
+  const formData = new FormData();
+  formData.append('url', decodedUrl);
+  
+  makeRequest('/api/url/delete', 'POST', formData, function(data) {
+    if(data.success) {
+      loadUrlList();
+      // If currently in Favorites tab, reload to see changes immediately
+      if(active === 'Favorites') {
+        loadFavorites(currentPage);
+      }
+    } else {
+      alert(data.message);
+    }
+  });
+}
+
+function showStatus(id, message, success) {
+  const status = document.getElementById(id);
+  status.textContent = message;
+  status.className = success ? 'status success' : 'status error';
+  setTimeout(() => {
+    status.className = 'status';
+  }, 3000);
+}
+
+// Create tabs
+// Favorites tab first
+const favTab = document.createElement('button');
+favTab.className = 'tab';
+favTab.textContent = 'Favorites';
+favTab.dataset.c = 'Favorites';
+favTab.onclick = function() { loadFavorites(1); };
+tabs.appendChild(favTab);
+
+// Category tabs
+cats.forEach(c => {
+  const b = document.createElement('button');
+  b.className = 'tab';
+  b.textContent = c;
+  b.dataset.c = c;
+  b.onclick = function() { loadCategory(c, 1); };
+  tabs.appendChild(b);
+});
+
+// Manage tab last
+const manageTab = document.createElement('button');
+manageTab.className = 'tab manage';
+manageTab.textContent = 'Manage';
+manageTab.dataset.c = 'Manage';
+manageTab.onclick = function() { loadManage(); };
+tabs.appendChild(manageTab);
+
+// Start with Favorites tab
+loadFavorites(1);
+</script>
+</body>
+</html>''')
     
-    # Update network status when the app loads
-    dalle.load(fn=update_network_status, outputs=network_status)
+    # Return the rendered template
+    return render_template('index.html', cats=list(CATEGORIES.keys()))
 
-if __name__ == "__main__":
-    dalle.launch(show_api=False, share=False)
+# Initialize database on startup
+init_db()
+
+# Define a function to ensure database consistency
+def ensure_db_consistency():
+    # Make sure we have the latest data in both JSON and SQLite
+    urls = load_db_sqlite()
+    save_json(urls)
+
+# For Flask 2.0+ compatibility
+@app.before_request
+def before_request_func():
+    # Use a flag to run this only once
+    if not hasattr(app, '_got_first_request'):
+        ensure_db_consistency()
+        app._got_first_request = True
+
+if __name__ == '__main__':
+    # 앱 시작 전에 명시적으로 DB 초기화
+    print("Initializing database...")
+    init_db()
+    
+    # 데이터베이스 파일 경로 및 존재 여부 확인
+    db_path = os.path.abspath(SQLITE_DB)
+    print(f"SQLite DB path: {db_path}")
+    if os.path.exists(SQLITE_DB):
+        print(f"Database file exists, size: {os.path.getsize(SQLITE_DB)} bytes")
+    else:
+        print("Warning: Database file does not exist after initialization!")
+    
+    app.run(host='0.0.0.0', port=7860)