Update app.py

app.py

@@ -1,78 +1,41 @@
 from flask import Flask, render_template, request, jsonify, session
-import requests
 import os
 from datetime import timedelta
-import json
 
 app = Flask(__name__)
 app.secret_key = os.urandom(24)
 app.permanent_session_lifetime = timedelta(days=7)
 
-# Hugging Face URL 목록
+# Hugging Face URL 목록 - 일부 URL은 미리 좋아요 상태로 설정
+# 'is_liked'를 true로 설정한 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": "https://huggingface.co/spaces/ginipick/Tech_Hangman_Game", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/openfree/deepseek_r1_API", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/ginipick/open_Deep-Research", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/aiqmaster/open-deep-research", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/seawolf2357/DeepSeek-R1-32b-search", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/ginigen/LLaDA", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/VIDraft/PHI4-Multimodal", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/ginigen/Ovis2-8B", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/ginigen/Graph-Mind", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/ginigen/Workflow-Canvas", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/ginigen/Design", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/ginigen/Diagram", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/ginigen/Mockup", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/ginigen/Infographic", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/ginigen/Flowchart", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/aiqcamp/FLUX-Vision", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/ginigen/VoiceClone-TTS", "is_liked": True},
+    {"url": "https://huggingface.co/spaces/openfree/Perceptron-Network", "is_liked": False},
+    {"url": "https://huggingface.co/spaces/openfree/Article-Generator", "is_liked": True},
 ]
 
-# 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:
-        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-v2", headers=headers)
-        if response.ok:
-            return True, response.json()
-    except Exception as e:
-        print(f"토큰 검증 오류: {e}")
-    
-    return False, None
-
 @app.route('/')
 def home():
     return render_template('index.html')
@@ -84,30 +47,16 @@ def login():
     if not token:
         return jsonify({'success': False, 'message': '토큰을 입력해주세요.'})
     
-    is_valid, user_info = validate_token(token)
-    
-    if not is_valid or not user_info:
+    # 간단히 토큰 길이만 검사 (실제로는 더 복잡한 검증 필요)
+    if len(token) < 5:
         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 = '인증된 사용자'
+    # 토큰의 첫 글자를 사용자 이름으로 설정 (테스트용)
+    username = f"사용자_{token[:3]}"
     
     # 세션에 저장
-    session['token'] = token
     session['username'] = username
     
-    # 사용자가 좋아요한 URL 상태를 세션에 초기화
-    if 'liked_urls' not in session:
-        session['liked_urls'] = {}
-    
     return jsonify({
         'success': True, 
         'username': username
@@ -115,30 +64,20 @@ def login():
 
 @app.route('/api/logout', methods=['POST'])
 def logout():
-    session.pop('token', None)
     session.pop('username', None)
-    session.pop('liked_urls', None)
     return jsonify({'success': True})
 
 @app.route('/api/urls', methods=['GET'])
 def get_urls():
     results = []
-    for url in HUGGINGFACE_URLS:
+    for url_item in HUGGINGFACE_URLS:
+        url = url_item["url"]
+        is_liked = url_item["is_liked"]
         title = extract_title(url)
-        model_info = extract_model_info(url)
-        
-        if not model_info:
-            continue
-        
-        # 사용자의 좋아요 상태 확인
-        is_liked = False
-        if 'liked_urls' in session and url in session['liked_urls']:
-            is_liked = True
         
         results.append({
             'url': url,
             'title': title,
-            'model_info': model_info,
             'is_liked': is_liked
         })
     
@@ -155,38 +94,19 @@ def toggle_like():
     if not url:
         return jsonify({'success': False, 'message': 'URL이 필요합니다.'})
     
-    # 세션에서 좋아요 정보 가져오기
-    liked_urls = session.get('liked_urls', {})
-    
-    # 좋아요 상태 토글
-    if url in liked_urls:
-        del liked_urls[url]
-        is_liked = False
-        message = '좋아요를 취소했습니다.'
-    else:
-        liked_urls[url] = True
-        is_liked = True
-        message = '좋아요를 추가했습니다.'
-    
-    # 세션에 좋아요 정보 업데이트
-    session['liked_urls'] = liked_urls
-    
-    return jsonify({
-        'success': True,
-        'is_liked': is_liked,
-        'message': message
-    })
-
-@app.route('/api/get-likes', methods=['GET'])
-def get_likes():
-    if 'username' not in session:
-        return jsonify({'success': False, 'message': '로그인이 필요합니다.'})
-    
-    liked_urls = session.get('liked_urls', {})
-    return jsonify({
-        'success': True,
-        'liked_urls': liked_urls
-    })
+    # URL 목록에서 해당 URL 찾기
+    for url_item in HUGGINGFACE_URLS:
+        if url_item["url"] == url:
+            # 좋아요 상태 토글
+            url_item["is_liked"] = not url_item["is_liked"]
+            
+            return jsonify({
+                'success': True,
+                'is_liked': url_item["is_liked"],
+                'message': '좋아요를 추가했습니다.' if url_item["is_liked"] else '좋아요를 취소했습니다.'
+            })
+    
+    return jsonify({'success': False, 'message': '해당 URL을 찾을 수 없습니다.'})
 
 @app.route('/api/session-status', methods=['GET'])
 def session_status():
@@ -560,7 +480,6 @@ if __name__ == '__main__':
     // 애플리케이션 상태
     const state = {
       username: null,
-      likedURLs: {},
       allURLs: [],
       isLoading: false,
       viewMode: 'all' // 'all' 또는 'liked'
@@ -596,7 +515,7 @@ if __name__ == '__main__':
     // 좋아요 통계 업데이트
     function updateLikeStats() {
       const totalCount = state.allURLs.length;
-      const likedCount = Object.keys(state.likedURLs).length;
+      const likedCount = state.allURLs.filter(item => item.is_liked).length;
       
       elements.totalUrlCount.textContent = totalCount;
       elements.likedUrlCount.textContent = likedCount;
@@ -615,9 +534,6 @@ if __name__ == '__main__':
           elements.loggedInSection.style.display = 'block';
           elements.likeStatus.style.display = 'block';
           
-          // 좋아요 목록 가져오기
-          await getLikedUrls();
-          
           // URL 목록 로드
           loadUrls();
         }
@@ -626,23 +542,6 @@ if __name__ == '__main__':
       }
     }
     
-    // 좋아요 URL 가져오기
-    async function getLikedUrls() {
-      try {
-        const response = await fetch('/api/get-likes');
-        const data = await handleApiResponse(response);
-        
-        if (data.success) {
-          state.likedURLs = data.liked_urls || {};
-          return true;
-        }
-      } catch (error) {
-        console.error('좋아요 목록 가져오기 오류:', error);
-      }
-      
-      return false;
-    }
-    
     // 로그인 처리
     async function login(token) {
       if (!token.trim()) {
@@ -673,9 +572,6 @@ if __name__ == '__main__':
           
           showMessage(`${state.username}님으로 로그인되었습니다.`);
           
-          // 좋아요 목록 가져오기
-          await getLikedUrls();
-          
           // URL 목록 로드
           loadUrls();
         } else {
@@ -702,7 +598,6 @@ if __name__ == '__main__':
         
         if (data.success) {
           state.username = null;
-          state.likedURLs = {};
           state.allURLs = [];
           
           elements.currentUser.textContent = '로그인되지 않음';
@@ -735,13 +630,6 @@ if __name__ == '__main__':
         // URL 및 좋아요 상태 저장
         state.allURLs = urls;
         
-        // 서버에서 받은 좋아요 상태 업데이트
-        urls.forEach(item => {
-          if (item.is_liked) {
-            state.likedURLs[item.url] = true;
-          }
-        });
-        
         // 필터링 및 렌더링
         filterAndRenderCards();
         
@@ -761,10 +649,10 @@ if __name__ == '__main__':
       
       // 필터링 적용
       const filteredUrls = state.allURLs.filter(item => {
-        const { url, title } = item;
+        const { url, title, is_liked } = item;
         
         // 좋아요 필터링 (좋아요만 보기 모드)
-        if (state.viewMode === 'liked' && !state.likedURLs[url]) {
+        if (state.viewMode === 'liked' && !is_liked) {
           return false;
         }
         
@@ -800,23 +688,34 @@ if __name__ == '__main__':
         const data = await handleApiResponse(response);
         
         if (data.success) {
-          // 좋아요 상태 업데이트
-          if (data.is_liked) {
-            state.likedURLs[url] = true;
-            card.classList.add('liked');
-            const likeBtn = card.querySelector('.like-button');
-            if (likeBtn) likeBtn.classList.add('liked');
-            const likeBadge = document.createElement('div');
-            likeBadge.className = 'like-badge';
-            likeBadge.textContent = '좋아요';
-            card.appendChild(likeBadge);
-          } else {
-            delete state.likedURLs[url];
-            card.classList.remove('liked');
-            const likeBtn = card.querySelector('.like-button');
-            if (likeBtn) likeBtn.classList.remove('liked');
-            const likeBadge = card.querySelector('.like-badge');
-            if (likeBadge) card.removeChild(likeBadge);
+          // 상태 객체에서 URL 찾기
+          const urlItem = state.allURLs.find(item => item.url === url);
+          if (urlItem) {
+            // 좋아요 상태 업데이트
+            urlItem.is_liked = data.is_liked;
+            
+            // 카드 UI 업데이트
+            if (data.is_liked) {
+              card.classList.add('liked');
+              const likeBtn = card.querySelector('.like-button');
+              if (likeBtn) likeBtn.classList.add('liked');
+              
+              // 좋아요 배지 추가
+              if (!card.querySelector('.like-badge')) {
+                const likeBadge = document.createElement('div');
+                likeBadge.className = 'like-badge';
+                likeBadge.textContent = '좋아요';
+                card.appendChild(likeBadge);
+              }
+            } else {
+              card.classList.remove('liked');
+              const likeBtn = card.querySelector('.like-button');
+              if (likeBtn) likeBtn.classList.remove('liked');
+              
+              // 좋아요 배지 제거
+              const likeBadge = card.querySelector('.like-badge');
+              if (likeBadge) card.removeChild(likeBadge);
+            }
           }
           
           showMessage(data.message);
@@ -853,12 +752,11 @@ if __name__ == '__main__':
       }
       
       urls.forEach(item => {
-        const { url, title } = item;
-        const isLiked = state.likedURLs[url] || false;
+        const { url, title, is_liked } = item;
         
         // 카드 생성
         const card = document.createElement('div');
-        card.className = `card ${isLiked ? 'liked' : ''}`;
+        card.className = `card ${is_liked ? 'liked' : ''}`;
         
         // 카드 헤더
         const cardHeader = document.createElement('div');
@@ -874,18 +772,16 @@ if __name__ == '__main__':
         
         // URL 링크
         const linkEl = document.createElement('a');
-
-
-linkEl.href = url;
+        linkEl.href = url;
         linkEl.textContent = url;
         linkEl.target = '_blank';
         card.appendChild(linkEl);
         
         // 좋아요 버튼
         const likeBtn = document.createElement('button');
-        likeBtn.className = `like-button ${isLiked ? 'liked' : ''}`;
+        likeBtn.className = `like-button ${is_liked ? 'liked' : ''}`;
         likeBtn.innerHTML = '♥';
-        likeBtn.title = isLiked ? '좋아요 취소' : '좋아요';
+        likeBtn.title = is_liked ? '좋아요 취소' : '좋아요';
         
         likeBtn.addEventListener('click', (e) => {
           e.preventDefault();
@@ -895,7 +791,7 @@ linkEl.href = url;
         card.appendChild(likeBtn);
         
         // 좋아요 배지 (좋아요 상태일 때만)
-        if (isLiked) {
+        if (is_liked) {
           const likeBadge = document.createElement('div');
           likeBadge.className = 'like-badge';
           likeBadge.textContent = '좋아요';
@@ -919,7 +815,7 @@ linkEl.href = url;
       filterAndRenderCards();
     }
     
-    // 이벤트 리스너 설정
+// 이벤트 리스너 설정
     elements.loginButton.addEventListener('click', () => {
       login(elements.tokenInput.value);
     });
@@ -952,4 +848,4 @@ linkEl.href = url;
 ''')
     
     # 허깅페이스 스페이스에서는 7860 포트 사용
-    app.run(host='0.0.0.0', port=7860)        
+    app.run(host='0.0.0.0', port=7860)