Create app.py

app.py

@@ -0,0 +1,792 @@
+from flask import Flask, render_template, request, redirect, url_for, jsonify, session
+import requests
+import os
+import json
+from datetime import timedelta
+
+app = Flask(__name__)
+app.secret_key = os.urandom(24)  # 세션 암호화를 위한 비밀 키
+app.permanent_session_lifetime = timedelta(days=7)  # 세션 유지 기간 설정
+
+# 허깅페이스 URL 목록
+HUGGINGFACE_URLS = [
+    "https://huggingface.co/spaces/ginipick/Tech_Hangman_Game",
+    "https://huggingface.co/spaces/openfree/deepseek_r1_API",
+    "https://huggingface.co/spaces/ginipick/open_Deep-Research",
+    "https://huggingface.co/spaces/aiqmaster/open-deep-research",
+    "https://huggingface.co/spaces/seawolf2357/DeepSeek-R1-32b-search",
+    "https://huggingface.co/spaces/ginigen/LLaDA",
+    "https://huggingface.co/spaces/VIDraft/PHI4-Multimodal",
+    "https://huggingface.co/spaces/ginigen/Ovis2-8B",
+    "https://huggingface.co/spaces/ginigen/Graph-Mind",
+    "https://huggingface.co/spaces/ginigen/Workflow-Canvas",
+    "https://huggingface.co/spaces/ginigen/Design",
+    "https://huggingface.co/spaces/ginigen/Diagram",
+    "https://huggingface.co/spaces/ginigen/Mockup",
+    "https://huggingface.co/spaces/ginigen/Infographic",
+    "https://huggingface.co/spaces/ginigen/Flowchart",
+    "https://huggingface.co/spaces/aiqcamp/FLUX-Vision",
+    "https://huggingface.co/spaces/ginigen/VoiceClone-TTS",
+    "https://huggingface.co/spaces/openfree/Perceptron-Network",
+    "https://huggingface.co/spaces/openfree/Article-Generator",
+]
+
+# URL에서 모델/스페이스 정보 추출
+def extract_model_info(url):
+    parts = url.split('/')
+    if len(parts) < 6:
+        return None
+    
+    if parts[3] == 'spaces' or parts[3] == 'models':
+        return {
+            'type': parts[3],
+            'owner': parts[4],
+            'repo': parts[5],
+            'full_id': f"{parts[4]}/{parts[5]}"
+        }
+    elif len(parts) >= 5:
+        # 다른 형식의 URL
+        return {
+            'type': 'models',  # 기본값
+            'owner': parts[3],
+            'repo': parts[4],
+            'full_id': f"{parts[3]}/{parts[4]}"
+        }
+    
+    return None
+
+# URL의 마지막 부분을 제목으로 추출
+def extract_title(url):
+    parts = url.split("/")
+    title = parts[-1] if parts else ""
+    return title.replace("_", " ").replace("-", " ")
+
+# 허깅페이스 인증 확인
+def validate_token(token):
+    headers = {"Authorization": f"Bearer {token}"}
+    try:
+        response = requests.get("https://huggingface.co/api/whoami", headers=headers)
+        return response.ok, response.json() if response.ok else None
+    except Exception as e:
+        print(f"토큰 검증 오류: {e}")
+        return False, None
+
+# 좋아요 목록 가져오기
+def get_liked_repos(token):
+    headers = {"Authorization": f"Bearer {token}"}
+    endpoints = [
+        "/api/me/likes",
+        "/api/me/liked-repos",
+        "/api/me/favorites"
+    ]
+    
+    liked_models = {}
+    
+    for endpoint in endpoints:
+        try:
+            response = requests.get(f"https://huggingface.co{endpoint}", headers=headers)
+            if response.ok:
+                data = response.json()
+                if isinstance(data, list):
+                    for model in data:
+                        if isinstance(model, dict):
+                            if 'owner' in model and 'name' in model:
+                                liked_models[f"{model['owner']}/{model['name']}"] = True
+                            elif 'id' in model:
+                                liked_models[model['id']] = True
+                elif isinstance(data, dict):
+                    for key in data:
+                        liked_models[key] = True
+                
+                print(f"좋아요 목록 가져오기 성공 (엔드포인트: {endpoint})")
+                return liked_models
+        except Exception as e:
+            print(f"좋아요 목록 가져오기 오류 (엔드포인트: {endpoint}): {e}")
+    
+    return liked_models
+
+# 좋아요 토글 API 호출
+def toggle_like(token, type_name, owner, repo, is_liked):
+    headers = {"Authorization": f"Bearer {token}"}
+    normalized_type = "spaces" if type_name == "spaces" else "models"
+    method = "DELETE" if is_liked else "POST"
+    url = f"https://huggingface.co/api/{normalized_type}/{owner}/{repo}/like"
+    
+    try:
+        if method == "DELETE":
+            response = requests.delete(url, headers=headers)
+        else:
+            response = requests.post(url, headers=headers)
+        
+        return response.ok, response.status_code
+    except Exception as e:
+        print(f"좋아요 토글 오류: {e}")
+        return False, 500
+
+# 홈페이지 라우트
+@app.route('/')
+def home():
+    return render_template('index.html')
+
+# 로그인 처리 API
+@app.route('/api/login', methods=['POST'])
+def login():
+    token = request.form.get('token', '')
+    
+    if not token:
+        return jsonify({'success': False, 'message': '토���을 입력해주세요.'})
+    
+    is_valid, user_info = validate_token(token)
+    
+    if not is_valid or not user_info:
+        return jsonify({'success': False, 'message': '유효하지 않은 토큰입니다.'})
+    
+    # 사용자 이름 찾기
+    username = None
+    if 'name' in user_info:
+        username = user_info['name']
+    elif 'user' in user_info and 'username' in user_info['user']:
+        username = user_info['user']['username']
+    elif 'username' in user_info:
+        username = user_info['username']
+    else:
+        username = '인증된 사용자'
+    
+    # 좋아요 목록 가져오기
+    liked_models = get_liked_repos(token)
+    
+    # 세션에 저장
+    session['token'] = token
+    session['username'] = username
+    session['liked_models'] = liked_models
+    
+    return jsonify({
+        'success': True, 
+        'username': username,
+        'liked_models': liked_models
+    })
+
+# 로그아웃 처리 API
+@app.route('/api/logout', methods=['POST'])
+def logout():
+    session.pop('token', None)
+    session.pop('username', None)
+    session.pop('liked_models', None)
+    return jsonify({'success': True})
+
+# URL 목록 가져오기 API
+@app.route('/api/urls', methods=['GET'])
+def get_urls():
+    search_query = request.args.get('search', '').lower()
+    show_only_liked = request.args.get('liked', 'false').lower() == 'true'
+    
+    liked_models = session.get('liked_models', {})
+    
+    results = []
+    for url in HUGGINGFACE_URLS:
+        title = extract_title(url)
+        model_info = extract_model_info(url)
+        
+        if not model_info:
+            continue
+        
+        is_liked = model_info['full_id'] in liked_models
+        
+        # 필터링 적용
+        if show_only_liked and not is_liked:
+            continue
+        
+        if search_query and search_query not in url.lower() and search_query not in title.lower():
+            continue
+        
+        results.append({
+            'url': url,
+            'title': title,
+            'model_info': model_info,
+            'is_liked': is_liked
+        })
+    
+    return jsonify(results)
+
+# 좋아요 토글 API
+@app.route('/api/toggle-like', methods=['POST'])
+def api_toggle_like():
+    if 'token' not in session:
+        return jsonify({'success': False, 'message': '로그인이 필요합니다.'})
+    
+    data = request.json
+    url = data.get('url')
+    current_liked = data.get('currentLiked', False)
+    
+    if not url:
+        return jsonify({'success': False, 'message': 'URL이 필요합니다.'})
+    
+    model_info = extract_model_info(url)
+    if not model_info:
+        return jsonify({'success': False, 'message': '유효하지 않은 URL입니다.'})
+    
+    success, status_code = toggle_like(
+        session['token'], 
+        model_info['type'], 
+        model_info['owner'], 
+        model_info['repo'], 
+        current_liked
+    )
+    
+    liked_models = session.get('liked_models', {})
+    
+    if success:
+        # 세션의 좋아요 상태 업데이트
+        if current_liked:
+            if model_info['full_id'] in liked_models:
+                del liked_models[model_info['full_id']]
+        else:
+            liked_models[model_info['full_id']] = True
+        
+        session['liked_models'] = liked_models
+        
+        return jsonify({
+            'success': True, 
+            'new_state': not current_liked,
+            'message': f"{'좋아요 취소' if current_liked else '좋아요'} 성공"
+        })
+    else:
+        return jsonify({
+            'success': False, 
+            'message': f"API 오류 (상태 코드: {status_code})"
+        })
+
+# 세션 상태 확인 API
+@app.route('/api/session-status', methods=['GET'])
+def session_status():
+    return jsonify({
+        'logged_in': 'token' in session,
+        'username': session.get('username'),
+        'liked_count': len(session.get('liked_models', {}))
+    })
+
+if __name__ == '__main__':
+    # templates 폴더 생성
+    os.makedirs('templates', exist_ok=True)
+    
+    # index.html 파일 생성
+    with open('templates/index.html', 'w', encoding='utf-8') as f:
+        f.write('''
+<!DOCTYPE html>
+<html lang="ko">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>Hugging Face URL 카드 리스트</title>
+  <style>
+    body {
+      font-family: Arial, sans-serif;
+      line-height: 1.6;
+      margin: 0;
+      padding: 0;
+      color: #333;
+    }
+    
+    .container {
+      max-width: 1200px;
+      margin: 0 auto;
+      padding: 1rem;
+    }
+    
+    .header {
+      background-color: #f0f0f0;
+      padding: 1rem;
+      border-radius: 5px;
+      margin-bottom: 1rem;
+    }
+    
+    .user-controls {
+      display: flex;
+      justify-content: space-between;
+      align-items: center;
+      flex-wrap: wrap;
+    }
+    
+    .filter-controls {
+      background-color: #f8f9fa;
+      padding: 1rem;
+      border-radius: 5px;
+      margin-bottom: 1rem;
+    }
+    
+    input[type="password"],
+    input[type="text"] {
+      padding: 0.5rem;
+      border: 1px solid #ccc;
+      border-radius: 4px;
+      margin-right: 5px;
+    }
+    
+    button {
+      padding: 0.5rem 1rem;
+      background-color: #4CAF50;
+      color: white;
+      border: none;
+      border-radius: 4px;
+      cursor: pointer;
+    }
+    
+    button:hover {
+      background-color: #45a049;
+    }
+    
+    button.logout {
+      background-color: #f44336;
+    }
+    
+    button.logout:hover {
+      background-color: #d32f2f;
+    }
+    
+    .token-help {
+      margin-top: 0.5rem;
+      font-size: 0.8rem;
+      color: #666;
+    }
+    
+    .token-help a {
+      color: #4CAF50;
+      text-decoration: none;
+    }
+    
+    .token-help a:hover {
+      text-decoration: underline;
+    }
+    
+    .cards-container {
+      display: flex;
+      flex-wrap: wrap;
+      gap: 1rem;
+    }
+    
+    .card {
+      border: 1px solid #ccc;
+      border-radius: 5px;
+      padding: 1rem;
+      width: 300px;
+      box-shadow: 2px 2px 8px rgba(0,0,0,0.1);
+      position: relative;
+      background-color: #f9f9f9;
+    }
+    
+    .card a {
+      text-decoration: none;
+      color: #333;
+      word-break: break-all;
+    }
+    
+    .like-button {
+      position: absolute;
+      top: 1rem;
+      right: 1rem;
+      border: none;
+      background: transparent;
+      font-size: 1.5rem;
+      cursor: pointer;
+      transition: color 0.2s;
+    }
+    
+    .like-button.liked {
+      color: red;
+    }
+    
+    .like-button.not-liked {
+      color: white;
+      -webkit-text-stroke: 1px #333;
+    }
+    
+    .status-message {
+      padding: 1rem;
+      border-radius: 4px;
+      margin-bottom: 1rem;
+      display: none;
+    }
+    
+    .success {
+      background-color: #dff0d8;
+      color: #3c763d;
+    }
+    
+    .error {
+      background-color: #f2dede;
+      color: #a94442;
+    }
+    
+    .loading {
+      position: fixed;
+      top: 0;
+      left: 0;
+      right: 0;
+      bottom: 0;
+      background-color: rgba(255, 255, 255, 0.7);
+      display: none;
+      justify-content: center;
+      align-items: center;
+      z-index: 1000;
+      font-size: 1.5rem;
+    }
+    
+    .login-section {
+      margin-top: 1rem;
+    }
+    
+    .logged-in-section {
+      display: none;
+      margin-top: 1rem;
+    }
+    
+    @media (max-width: 768px) {
+      .user-controls {
+        flex-direction: column;
+        align-items: flex-start;
+      }
+      
+      .user-controls > div {
+        margin-bottom: 1rem;
+      }
+      
+      .card {
+        width: 100%;
+      }
+    }
+  </style>
+</head>
+<body>
+  <div class="container">
+    <div class="header">
+      <div class="user-controls">
+        <div>
+          <span>허깅페이스 계정: </span>
+          <span id="currentUser">로그인되지 않음</span>
+        </div>
+        
+        <div id="loginSection" class="login-section">
+          <input type="password" id="tokenInput" placeholder="허깅페이스 API 토큰 입력" />
+          <button id="loginButton">인증하기</button>
+          <div class="token-help">
+            API 토큰은 <a href="https://huggingface.co/settings/tokens" target="_blank">허깅페이스 토큰 페이지</a>에서 생성할 수 있습니다.
+          </div>
+        </div>
+        
+        <div id="loggedInSection" class="logged-in-section">
+          <button id="logoutButton" class="logout">로그아웃</button>
+        </div>
+      </div>
+    </div>
+    
+    <div class="filter-controls">
+      <label>
+        <input type="checkbox" id="showOnlyLiked" />
+        내가 좋아요한 URL만 보기
+      </label>
+      <label style="margin-left: 1rem;">
+        <input type="text" id="searchInput" placeholder="URL 또는 제목으로 검색" style="width: 250px;" />
+      </label>
+    </div>
+    
+    <div id="statusMessage" class="status-message"></div>
+    
+    <div id="loadingIndicator" class="loading">처리 중...</div>
+    
+    <div id="cardsContainer" class="cards-container"></div>
+  </div>
+  
+  <script>
+    // DOM 요소 참조
+    const elements = {
+      tokenInput: document.getElementById('tokenInput'),
+      loginButton: document.getElementById('loginButton'),
+      logoutButton: document.getElementById('logoutButton'),
+      currentUser: document.getElementById('currentUser'),
+      cardsContainer: document.getElementById('cardsContainer'),
+      loadingIndicator: document.getElementById('loadingIndicator'),
+      statusMessage: document.getElementById('statusMessage'),
+      showOnlyLiked: document.getElementById('showOnlyLiked'),
+      searchInput: document.getElementById('searchInput'),
+      loginSection: document.getElementById('loginSection'),
+      loggedInSection: document.getElementById('loggedInSection')
+    };
+    
+    // 애플리케이션 상태
+    const state = {
+      username: null,
+      likedModels: {},
+      isLoading: false
+    };
+    
+    // 로딩 상태 표시 함수
+    function setLoading(isLoading) {
+      state.isLoading = isLoading;
+      elements.loadingIndicator.style.display = isLoading ? 'flex' : 'none';
+    }
+    
+    // 상태 메시지 표시 함수
+    function showMessage(message, isError = false) {
+      elements.statusMessage.textContent = message;
+      elements.statusMessage.className = `status-message ${isError ? 'error' : 'success'}`;
+      elements.statusMessage.style.display = 'block';
+      
+      // 3초 후 메시지 사라짐
+      setTimeout(() => {
+        elements.statusMessage.style.display = 'none';
+      }, 3000);
+    }
+    
+    // API 오류 처리 함수
+    async function handleApiResponse(response) {
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`API 오류 (${response.status}): ${errorText}`);
+      }
+      return response.json();
+    }
+    
+    // 세션 상태 확인
+    async function checkSessionStatus() {
+      try {
+        const response = await fetch('/api/session-status');
+        const data = await handleApiResponse(response);
+        
+        if (data.logged_in) {
+          state.username = data.username;
+          elements.currentUser.textContent = data.username;
+          elements.loginSection.style.display = 'none';
+          elements.loggedInSection.style.display = 'block';
+          
+          // URL 목록 로드
+          loadUrls();
+        }
+      } catch (error) {
+        console.error('세션 상태 확인 오류:', error);
+      }
+    }
+    
+    // 로그인 처리
+    async function login(token) {
+      if (!token.trim()) {
+        showMessage('토큰을 입력해주세요.', true);
+        return;
+      }
+      
+      setLoading(true);
+      
+      try {
+        const formData = new FormData();
+        formData.append('token', token);
+        
+        const response = await fetch('/api/login', {
+          method: 'POST',
+          body: formData
+        });
+        
+        const data = await handleApiResponse(response);
+        
+        if (data.success) {
+          state.username = data.username;
+          state.likedModels = data.liked_models || {};
+          
+          elements.currentUser.textContent = state.username;
+          elements.loginSection.style.display = 'none';
+          elements.loggedInSection.style.display = 'block';
+          
+          showMessage(`${state.username}님으로 로그인되었습니다.`);
+          
+          // URL 목록 로드
+          loadUrls();
+        } else {
+          showMessage(data.message || '로그인에 실패했습니다.', true);
+        }
+      } catch (error) {
+        console.error('로그인 오류:', error);
+        showMessage(`로그인 오류: ${error.message}`, true);
+      } finally {
+        setLoading(false);
+      }
+    }
+    
+    // 로그아웃 처리
+    async function logout() {
+      setLoading(true);
+      
+      try {
+        const response = await fetch('/api/logout', {
+          method: 'POST'
+        });
+        
+        const data = await handleApiResponse(response);
+        
+        if (data.success) {
+          state.username = null;
+          state.likedModels = {};
+          
+          elements.currentUser.textContent = '로그인되지 않음';
+          elements.tokenInput.value = '';
+          elements.loginSection.style.display = 'block';
+          elements.loggedInSection.style.display = 'none';
+          
+          showMessage('로그아웃되었습니다.');
+          
+          // 카드 초기화
+          elements.cardsContainer.innerHTML = '';
+        }
+      } catch (error) {
+        console.error('로그아웃 오류:', error);
+        showMessage(`로그아웃 오류: ${error.message}`, true);
+      } finally {
+        setLoading(false);
+      }
+    }
+    
+    // URL 목록 로드
+    async function loadUrls() {
+      setLoading(true);
+      
+      try {
+        const showOnlyLiked = elements.showOnlyLiked.checked;
+        const searchText = elements.searchInput.value;
+        
+        const response = await fetch(`/api/urls?liked=${showOnlyLiked}&search=${encodeURIComponent(searchText)}`);
+        const urls = await handleApiResponse(response);
+        
+        renderCards(urls);
+      } catch (error) {
+        console.error('URL 목록 로드 오류:', error);
+        showMessage(`URL 로드 오류: ${error.message}`, true);
+      } finally {
+        setLoading(false);
+      }
+    }
+    
+    // 좋아요 토글
+    async function toggleLike(url, button, currentLiked) {
+      if (!state.username) {
+        showMessage('좋아요를 하려면 허깅페이스 API 토큰으로 인증이 필요합니다.', true);
+        return;
+      }
+      
+      setLoading(true);
+      
+      try {
+        const response = await fetch('/api/toggle-like', {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json'
+          },
+          body: JSON.stringify({
+            url: url,
+            currentLiked: currentLiked
+          })
+        });
+        
+        const data = await handleApiResponse(response);
+        
+        if (data.success) {
+          if (currentLiked) {
+            button.classList.remove('liked');
+            button.classList.add('not-liked');
+          } else {
+            button.classList.add('liked');
+            button.classList.remove('not-liked');
+          }
+          
+          showMessage(data.message);
+          
+          // 필터링된 경우 목록 다시 로드
+          if (elements.showOnlyLiked.checked) {
+            loadUrls();
+          }
+        } else {
+          showMessage(data.message, true);
+        }
+      } catch (error) {
+        console.error('좋아요 토글 오류:', error);
+        showMessage(`좋아요 처리 오류: ${error.message}`, true);
+      } finally {
+        setLoading(false);
+      }
+    }
+    
+    // 카드 렌더링
+    function renderCards(urls) {
+      elements.cardsContainer.innerHTML = '';
+      
+      if (!urls || urls.length === 0) {
+        const noResultsMsg = document.createElement('p');
+        noResultsMsg.textContent = '표시할 URL이 없습니다.';
+        noResultsMsg.style.padding = '1rem';
+        noResultsMsg.style.fontStyle = 'italic';
+        elements.cardsContainer.appendChild(noResultsMsg);
+        return;
+      }
+      
+      urls.forEach(item => {
+        const { url, title, is_liked } = item;
+        
+        // 카드 생성
+        const card = document.createElement('div');
+        card.className = 'card';
+        
+        // 제목
+        const titleEl = document.createElement('h3');
+        titleEl.textContent = title;
+        card.appendChild(titleEl);
+        
+        // URL 링크
+        const linkEl = document.createElement('a');
+        linkEl.href = url;
+        linkEl.textContent = url;
+        linkEl.target = '_blank';
+        card.appendChild(linkEl);
+        
+        // 좋아요 버튼 (♥ 아이콘)
+        const likeBtn = document.createElement('button');
+        likeBtn.className = `like-button ${is_liked ? 'liked' : 'not-liked'}`;
+        likeBtn.textContent = '♥';
+        
+        likeBtn.addEventListener('click', function(e) {
+          e.preventDefault();
+          toggleLike(url, likeBtn, is_liked);
+        });
+        card.appendChild(likeBtn);
+        
+        // 카드 추가
+        elements.cardsContainer.appendChild(card);
+      });
+    }
+    
+    // 이벤트 리스너 설정
+    elements.loginButton.addEventListener('click', () => {
+      login(elements.tokenInput.value);
+    });
+    
+    elements.logoutButton.addEventListener('click', logout);
+    
+    // 엔터 키로 로그인 가능하게
+    elements.tokenInput.addEventListener('keypress', (event) => {
+      if (event.key === 'Enter') {
+        login(elements.tokenInput.value);
+      }
+    });
+    
+    // 필터링 이벤트 리스너
+    elements.showOnlyLiked.addEventListener('change', loadUrls);
+    elements.searchInput.addEventListener('input', () => {
+      // 입력 지연 처리 (타이핑할 때마다 API 호출 방지)
+      clearTimeout(state.searchTimeout);
+      state.searchTimeout = setTimeout(loadUrls, 300);
+    });
+    
+    // 초기화
+    checkSessionStatus();
+  </script>
+</body>
+</html>
+''')
+    
+    # 서버 실행
+    app.run(debug=True)