Create app.py

app.py

@@ -0,0 +1,368 @@
+import os, re, json, sqlite3
+from flask import Flask, render_template, request, jsonify
+
+app = Flask(__name__)
+
+# ────────────────────────── 경로를 절대 경로로 지정 ──────────────────────────
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+DB_FILE = os.path.join(BASE_DIR, "favorite_sites.json")
+SQLITE_DB = os.path.join(BASE_DIR, "favorite_sites.db")
+
+# 이하 기존 코드 동일
+BLOCKED_DOMAINS = [
+    "naver.com", "daum.net", "google.com",
+    "facebook.com", "instagram.com", "kakao.com",
+    "ycombinator.com"
+]
+
+CATEGORIES = {
+    "Productivity": [
+        "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",
+    ],
+}
+
+def init_db():
+    if not os.path.exists(DB_FILE):
+        with open(DB_FILE, "w", encoding="utf-8") as f:
+            json.dump([], f, ensure_ascii=False)
+    
+    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()
+    
+    # JSON에 있는 URL을 (초기화 시) 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"""
+    try:
+        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"""
+    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"""
+    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:
+        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:
+            success = False
+    except sqlite3.IntegrityError:
+        success = False
+    conn.close()
+    return success
+
+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:
+        success = False
+    conn.close()
+    return success
+
+def load_db():
+    """Load URLs: 우선 SQLite에서 불러온 뒤, 없으면 JSON 불러오기"""
+    urls = load_db_sqlite()
+    if not urls:
+        urls = load_json()
+        for url in urls:
+            add_url_to_sqlite(url)
+    return urls
+
+def save_db(lst):
+    """Save URLs to both SQLite and JSON"""
+    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()
+    return save_json(lst)
+
+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 screenshot_url(url):
+    return f"https://image.thum.io/get/fullpage/{url}"
+
+def process_url_for_preview(url):
+    if any(d for d in BLOCKED_DOMAINS if d in url):
+        return screenshot_url(url), "snapshot"
+    if "vibe-coding-tetris" in url or "World-of-Tank-GAME" in url or "Minesweeper-Game" in url:
+        return screenshot_url(url), "snapshot"
+    try:
+        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"
+    return url, "iframe"
+
+@app.route('/api/category')
+def api_category():
+    cat = request.args.get('name', '')
+    urls = CATEGORIES.get(cat, [])
+    return jsonify([
+        {
+            "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
+    ])
+
+@app.route('/api/favorites')
+def api_favorites():
+    urls = load_db()
+    page = int(request.args.get('page', 1))
+    per_page = int(request.args.get('per_page', 9))
+    
+    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:
+            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"})
+    
+    conn = sqlite3.connect(SQLITE_DB)
+    cursor = conn.cursor()
+    try:
+        cursor.execute("INSERT INTO urls (url) VALUES (?)", (url,))
+        conn.commit()
+        success = True
+    except sqlite3.IntegrityError:
+        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"})
+
+@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"})
+    
+    if not update_url_in_sqlite(old, new):
+        return jsonify({"success": False, "message": "URL not found or new URL already exists"})
+    
+    data = load_json()
+    try:
+        idx = data.index(old)
+        data[idx] = new
+        save_json(data)
+    except ValueError:
+        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', '')
+    
+    if not delete_url_from_sqlite(url):
+        return jsonify({"success": False, "message": "URL not found"})
+    
+    data = load_json()
+    try:
+        data.remove(url)
+        save_json(data)
+    except ValueError:
+        pass
+    
+    return jsonify({"success": True, "message": "URL deleted successfully"})
+
+@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 템플릿은 기존 코드 그대로 유지 -->
+<!-- ... -->
+''')
+    return render_template('index.html', cats=list(CATEGORIES.keys()))
+
+init_db()
+
+def ensure_db_consistency():
+    urls = load_db_sqlite()
+    save_json(urls)
+
+@app.before_request
+def before_request_func():
+    if not hasattr(app, '_got_first_request'):
+        ensure_db_consistency()
+        app._got_first_request = True
+
+if __name__ == '__main__':
+    print("Initializing database...")
+    init_db()
+    db_path = os.path.abspath(SQLITE_DB)
+    print(f"SQLite DB path: {db_path}")
+    if os.path.exists(db_path):
+        print(f"Database file exists, size: {os.path.getsize(db_path)} bytes")
+    else:
+        print("Warning: Database file does not exist after initialization!")
+    
+    app.run(host='0.0.0.0', port=7860)