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: 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: success = False except sqlite3.IntegrityError: # New URL already exists 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(): """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() # 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 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: 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, []) 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(): # Load URLs from SQLite database 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: # 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 }) # `/api/url/add` 경로 함수 수정 @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"}) @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''' Web Gallery

🌟 Web Gallery Manager

🚀 Free AI Spaces & Website Gallery ✨ Save and manage your favorite sites! Supports LIVE and Static preview modes.

Your desired URL will be permanently saved when recorded in the favorite_sites.json file.

badge

''') # Return the rendered template return render_template('index.html', cats=list(CATEGORIES.keys())) # 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)