Update app.py

app.py

@@ -3,7 +3,17 @@ import os, re, json
 
 app = Flask(__name__)
 
-# ────────────────────────── 1. CURATED CATEGORIES ──────────────────────────
+# ────────────────────────── 1. CONFIGURATION ──────────────────────────
+DB_FILE = "favorite_sites.json"  # Database file for user-saved sites
+
+# 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/ginigen/perflexity-clone",
@@ -61,20 +71,67 @@ CATEGORIES = {
     ],
 }
 
-# ─────────────── 2. URL HELPERS ────────────────
-def direct_url(hf_url: str) -> str:
+# ────────────────────────── 3. DATABASE FUNCTIONS ──────────────────────────
+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)
+
+def load_db():
+    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_db(lst):
+    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
+
+# ────────────────────────── 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()
+    name = name.replace('.', '-').replace('_', '-').lower()
     return f"https://{owner}-{name}.hf.space"
 
-def screenshot_url(hf_url: str) -> str:
-    return f"https://image.thum.io/get/fullpage/{direct_url(hf_url)}"
+def screenshot_url(url):
+    return f"https://image.thum.io/get/fullpage/{url}"
 
-# ─────────────── 3. API FOR TABS ───────────────
+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', '')
@@ -84,105 +141,418 @@ def api_category():
             "title": url.split('/')[-1],
             "owner": url.split('/')[-2] if '/spaces/' in url else '',
             "iframe": direct_url(url),
-            "shot":   screenshot_url(url),
-            "hf":     url
+            "shot": screenshot_url(url),
+            "hf": url
         } for url in urls
     ])
 
-# ─────────────── 4. ROUTES ───────────────
+@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:
+            # 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"})
+    
+    data = load_db()
+    if url in data:
+        return jsonify({"success": False, "message": "URL already exists"})
+    
+    data.insert(0, url)
+    save_db(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"})
+    
+    data = load_db()
+    try:
+        idx = data.index(old)
+        data[idx] = new
+        save_db(data)
+        return jsonify({"success": True, "message": "URL updated successfully"})
+    except ValueError:
+        return jsonify({"success": False, "message": "URL not found"})
+
+@app.route('/api/url/delete', methods=['POST'])
+def delete_url():
+    url = request.form.get('url', '')
+    data = load_db()
+    try:
+        data.remove(url)
+        save_db(data)
+        return jsonify({"success": True, "message": "URL deleted successfully"})
+    except ValueError:
+        return jsonify({"success": False, "message": "URL not found"})
+
+# ────────────────────────── 6. MAIN ROUTES ──────────────────────────
 @app.route('/')
 def home():
-    return render_template('index.html', cats=list(CATEGORIES.keys()))
-
-# ─────────────── 5. CREATE TEMPLATE ───────────────
-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>Dynamic HF Spaces Gallery</title>
+    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}
-.grid{display:grid;grid-template-columns:repeat(3,1fr);gap:14px;padding:0 16px 60px}
-@media(max-width:800px){.grid{grid-template-columns:1fr}}
-.card{background:#fff;border-radius:12px;box-shadow:0 2px 8px rgba(0,0,0,.08);overflow:hidden;height:420px;display:flex;flex-direction:column;position:relative}
-.frame{flex:1;position:relative;overflow:hidden}
-.frame iframe,.frame img{position:absolute;width:166.667%;height:166.667%;transform:scale(.6);transform-origin:top left;border:0}
-.err{display:none;align-items:center;justify-content:center;width:100%;height:100%;background:#fafafa;text-align:center;padding:10px;color:#666;font-size:.9rem}
-.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}
+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;}
+.grid{display:grid;grid-template-columns:repeat(3,1fr);gap:14px;padding:0 16px 60px;}
+@media(max-width:800px){.grid{grid-template-columns:1fr;}}
+.card{background:#fff;border-radius:12px;box-shadow:0 2px 8px rgba(0,0,0,.08);overflow:hidden;height:420px;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;}
+.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>
 <div class="tabs" id="tabs"></div>
-<div class="grid" id="grid"></div>
+<div id="content"></div>
 
 <script>
-const cats={{cats|tojson}};
-const tabs=document.getElementById('tabs');
-const grid=document.getElementById('grid');
-let active="";
-
-/* ---------- card builder (dynamic + fallback) ---------- */
-function createCard(sp){
-  const card=document.createElement('div');card.className='card';
-
-  const frameBox=document.createElement('div');frameBox.className='frame';
-  const ifr=document.createElement('iframe');
-  ifr.src=sp.iframe; ifr.loading='lazy';
-  frameBox.appendChild(ifr);
-
-  const err=document.createElement('div');err.className='err';
-  err.innerHTML=`Preview blocked.<br><a href="${sp.hf}" target="_blank">Open on HF ↗</a>`;
-  frameBox.appendChild(err);
-
-  let loaded=false;
-  ifr.onload = ()=>{ loaded=true; };          // success
-  ifr.onerror= fallback;                      // immediate fail
-  setTimeout(()=>{ if(!loaded) fallback(); },10000); // watchdog
-
-  function fallback(){
-    ifr.remove();
-    const img=new Image(); img.src=sp.shot; img.loading='lazy';
-    img.onerror=()=>{ err.style.display='flex'; };
-    frameBox.prepend(img);
+// 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();
+}
+
+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);
+  });
+}
 
-  const foot=document.createElement('div');foot.className='foot';
-  foot.innerHTML=`<a href="${sp.iframe}" target="_blank">${sp.title}</a>`;
+// Tab handlers
+function loadCategory(cat) {
+  if(cat === active) return;
+  active = cat;
+  updateTabs();
+  
+  content.innerHTML = '<p style="text-align:center;padding:40px">Loading…</p>';
+  
+  makeRequest('/api/category?name=' + encodeURIComponent(cat), 'GET', null, function(data) {
+    let html = '<div class="grid">';
+    
+    data.forEach(item => {
+      html += `
+        <div class="card">
+          <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>';
+    content.innerHTML = html;
+  });
+}
 
-  card.appendChild(frameBox);
-  card.appendChild(foot);
-  return card;
+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, '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="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="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>';
+    
+    // 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;
+  });
 }
 
-/* ---------- load tab ---------- */
-function load(cat){
-  if(cat===active) return;
-  active=cat;
-  [...tabs.children].forEach(b=>b.classList.toggle('active',b.dataset.c===cat));
-  grid.innerHTML='<p style="grid-column:1/-1;text-align:center;padding:40px">Loading…</p>';
-  fetch('/api/category?name='+encodeURIComponent(cat))
-    .then(r=>r.json()).then(arr=>{
-      grid.innerHTML='';
-      arr.forEach(sp=>grid.appendChild(createCard(sp)));
+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 => {
+      html += `
+        <div class="url-item">
+          <div>${item.url}</div>
+          <div class="url-controls">
+            <button class="btn" onclick="editUrl('${item.url}')">Edit</button>
+            <button class="btn btn-danger" onclick="deleteUrl('${item.url}')">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();
+    }
+  });
+}
+
+function editUrl(url) {
+  const newUrl = prompt('Edit URL:', url);
+  
+  if(!newUrl || newUrl === url) return;
+  
+  const formData = new FormData();
+  formData.append('old', url);
+  formData.append('new', newUrl);
+  
+  makeRequest('/api/url/update', 'POST', formData, function(data) {
+    if(data.success) {
+      loadUrlList();
+    } else {
+      alert(data.message);
+    }
+  });
 }
 
-/* ---------- init tabs ---------- */
-cats.forEach((c,i)=>{
-  const b=document.createElement('button');
-  b.className='tab'; b.textContent=c; b.dataset.c=c;
-  b.onclick=()=>load(c);
+function deleteUrl(url) {
+  if(!confirm('Are you sure you want to delete this URL?')) return;
+  
+  const formData = new FormData();
+  formData.append('url', url);
+  
+  makeRequest('/api/url/delete', 'POST', formData, function(data) {
+    if(data.success) {
+      loadUrlList();
+    } 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); };
   tabs.appendChild(b);
-  if(i===0) load(c);
 });
+
+// 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>''')
+</body>
+</html>''')
+    
+    # Return the rendered template
+    return render_template('index.html', cats=list(CATEGORIES.keys()))
+
+# Initialize database on startup
+init_db()
 
-# ─────────────── 6. RUN ───────────────
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860)