from flask import Flask, render_template, request, jsonify, session
import os
from datetime import timedelta

app = Flask(__name__)
app.secret_key = os.urandom(24)
app.permanent_session_lifetime = timedelta(days=7)

# Hugging Face URL 목록 - 일부 URL은 미리 좋아요 상태로 설정
# 'is_liked'를 true로 설정한 URL은 항상 좋아요 상태로 표시됨
HUGGINGFACE_URLS = [
    {"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_title(url):
    parts = url.split("/")
    title = parts[-1] if parts else ""
    return title.replace("_", " ").replace("-", " ")

@app.route('/')
def home():
    return render_template('index.html')

@app.route('/api/login', methods=['POST'])
def login():
    token = request.form.get('token', '')
    
    if not token:
        return jsonify({'success': False, 'message': '토큰을 입력해주세요.'})
    
    # 간단히 토큰 길이만 검사 (실제로는 더 복잡한 검증 필요)
    if len(token) < 5:
        return jsonify({'success': False, 'message': '유효하지 않은 토큰입니다.'})
    
    # 토큰의 첫 글자를 사용자 이름으로 설정 (테스트용)
    username = f"사용자_{token[:3]}"
    
    # 세션에 저장
    session['username'] = username
    
    return jsonify({
        'success': True, 
        'username': username
    })

@app.route('/api/logout', methods=['POST'])
def logout():
    session.pop('username', None)
    return jsonify({'success': True})

@app.route('/api/urls', methods=['GET'])
def get_urls():
    results = []
    for url_item in HUGGINGFACE_URLS:
        url = url_item["url"]
        is_liked = url_item["is_liked"]
        title = extract_title(url)
        
        results.append({
            'url': url,
            'title': title,
            'is_liked': is_liked
        })
    
    return jsonify(results)

@app.route('/api/toggle-like', methods=['POST'])
def toggle_like():
    if 'username' not in session:
        return jsonify({'success': False, 'message': '로그인이 필요합니다.'})
    
    data = request.json
    url = data.get('url')
    
    if not url:
        return jsonify({'success': False, 'message': 'URL이 필요합니다.'})
    
    # 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():
    return jsonify({
        'logged_in': 'username' in session,
        'username': session.get('username')
    })

if __name__ == '__main__':
    os.makedirs('templates', exist_ok=True)
    
    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;
      background-color: #f4f5f7;
    }
    
    .container {
      max-width: 1200px;
      margin: 0 auto;
      padding: 1rem;
    }
    
    .header {
      background-color: #fff;
      padding: 1rem;
      border-radius: 8px;
      margin-bottom: 1rem;
      box-shadow: 0 2px 4px rgba(0,0,0,0.1);
    }
    
    .user-controls {
      display: flex;
      justify-content: space-between;
      align-items: center;
      flex-wrap: wrap;
    }
    
    .filter-controls {
      background-color: #fff;
      padding: 1rem;
      border-radius: 8px;
      margin-bottom: 1rem;
      box-shadow: 0 2px 4px rgba(0,0,0,0.1);
      display: flex;
      justify-content: space-between;
      align-items: center;
    }
    
    input[type="password"],
    input[type="text"] {
      padding: 0.5rem;
      border: 1px solid #ddd;
      border-radius: 4px;
      margin-right: 5px;
    }
    
    button {
      padding: 0.5rem 1rem;
      background-color: #4CAF50;
      color: white;
      border: none;
      border-radius: 4px;
      cursor: pointer;
      transition: background-color 0.2s;
    }
    
    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 #ddd;
      border-radius: 8px;
      padding: 1rem;
      width: 300px;
      box-shadow: 0 2px 4px rgba(0,0,0,0.1);
      position: relative;
      background-color: #fff;
      transition: all 0.3s ease;
    }
    
    .card:hover {
      transform: translateY(-5px);
      box-shadow: 0 5px 15px rgba(0,0,0,0.1);
    }
    
    .card.liked {
      border-color: #ff4757;
      background-color: #ffebee;
    }
    
    .card-header {
      margin-bottom: 0.5rem;
      padding-right: 40px; /* 좋아요 버튼 공간 */
    }
    
    .card-title {
      font-size: 1.2rem;
      margin: 0 0 0.5rem 0;
      color: #333;
    }
    
    .card a {
      text-decoration: none;
      color: #2980b9;
      word-break: break-all;
      display: block;
      font-size: 0.9rem;
    }
    
    .card a:hover {
      text-decoration: underline;
    }
    
    .like-button {
      position: absolute;
      top: 1rem;
      right: 1rem;
      width: 30px;
      height: 30px;
      display: flex;
      align-items: center;
      justify-content: center;
      border-radius: 50%;
      border: none;
      background: transparent;
      font-size: 1.5rem;
      cursor: pointer;
      transition: all 0.3s ease;
      color: #ddd;
    }
    
    .like-button:hover {
      transform: scale(1.2);
    }
    
    .like-button.liked {
      color: #ff4757;
    }
    
    .like-badge {
      position: absolute;
      top: -5px;
      left: -5px;
      background-color: #ff4757;
      color: white;
      padding: 0.2rem 0.5rem;
      border-radius: 4px;
      font-size: 0.7rem;
      font-weight: bold;
    }
    
    .like-status {
      background-color: #fff;
      padding: 1rem;
      border-radius: 8px;
      margin-bottom: 1rem;
      box-shadow: 0 2px 4px rgba(0,0,0,0.1);
      display: none;
    }
    
    .like-status strong {
      color: #ff4757;
    }
    
    .status-message {
      position: fixed;
      bottom: 20px;
      right: 20px;
      padding: 1rem;
      border-radius: 8px;
      display: none;
      box-shadow: 0 4px 12px rgba(0,0,0,0.15);
      z-index: 1000;
      max-width: 300px;
    }
    
    .success {
      background-color: #4CAF50;
      color: white;
    }
    
    .error {
      background-color: #f44336;
      color: white;
    }
    
    .loading {
      position: fixed;
      top: 0;
      left: 0;
      right: 0;
      bottom: 0;
      background-color: rgba(255, 255, 255, 0.8);
      display: none;
      justify-content: center;
      align-items: center;
      z-index: 1000;
    }
    
    .spinner {
      width: 40px;
      height: 40px;
      border: 4px solid #f3f3f3;
      border-top: 4px solid #3498db;
      border-radius: 50%;
      animation: spin 1s linear infinite;
    }
    
    @keyframes spin {
      0% { transform: rotate(0deg); }
      100% { transform: rotate(360deg); }
    }
    
    .filter-toggle {
      display: flex;
    }
    
    .filter-toggle button {
      margin-right: 0.5rem;
      background-color: #f0f0f0;
      color: #333;
    }
    
    .filter-toggle button.active {
      background-color: #4CAF50;
      color: white;
    }
    
    .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;
      }
      
      .filter-controls {
        flex-direction: column;
      }
      
      .filter-controls > div {
        margin-bottom: 0.5rem;
      }
      
      .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 id="likeStatus" class="like-status">
      <div id="likeStatsText">총 <span id="totalUrlCount">0</span>개 중 <strong><span id="likedUrlCount">0</span>개</strong>의 URL을 좋아요 했습니다.</div>
    </div>
    
    <div class="filter-controls">
      <div>
        <input type="text" id="searchInput" placeholder="URL 또는 제목으로 검색" style="width: 300px;" />
      </div>
      <div class="filter-toggle">
        <button id="allUrlsBtn" class="active">전체 보기</button>
        <button id="likedUrlsBtn">좋아요만 보기</button>
      </div>
    </div>
    
    <div id="statusMessage" class="status-message"></div>
    
    <div id="loadingIndicator" class="loading">
      <div class="spinner"></div>
    </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'),
      searchInput: document.getElementById('searchInput'),
      loginSection: document.getElementById('loginSection'),
      loggedInSection: document.getElementById('loggedInSection'),
      likeStatus: document.getElementById('likeStatus'),
      totalUrlCount: document.getElementById('totalUrlCount'),
      likedUrlCount: document.getElementById('likedUrlCount'),
      allUrlsBtn: document.getElementById('allUrlsBtn'),
      likedUrlsBtn: document.getElementById('likedUrlsBtn')
    };
    
    // 애플리케이션 상태
    const state = {
      username: null,
      allURLs: [],
      isLoading: false,
      viewMode: 'all' // 'all' 또는 'liked'
    };
    
    // 로딩 상태 표시 함수
    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();
    }
    
    // 좋아요 통계 업데이트
    function updateLikeStats() {
      const totalCount = state.allURLs.length;
      const likedCount = state.allURLs.filter(item => item.is_liked).length;
      
      elements.totalUrlCount.textContent = totalCount;
      elements.likedUrlCount.textContent = likedCount;
    }
    
    // 세션 상태 확인
    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';
          elements.likeStatus.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;
          
          elements.currentUser.textContent = state.username;
          elements.loginSection.style.display = 'none';
          elements.loggedInSection.style.display = 'block';
          elements.likeStatus.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.allURLs = [];
          
          elements.currentUser.textContent = '로그인되지 않음';
          elements.tokenInput.value = '';
          elements.loginSection.style.display = 'block';
          elements.loggedInSection.style.display = 'none';
          elements.likeStatus.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 response = await fetch('/api/urls');
        const urls = await handleApiResponse(response);
        
        // URL 및 좋아요 상태 저장
        state.allURLs = urls;
        
        // 필터링 및 렌더링
        filterAndRenderCards();
        
        // 좋아요 통계 업데이트
        updateLikeStats();
      } catch (error) {
        console.error('URL 목록 로드 오류:', error);
        showMessage(`URL 로드 오류: ${error.message}`, true);
      } finally {
        setLoading(false);
      }
    }
    
    // 필터링 및 카드 렌더링
    function filterAndRenderCards() {
      const searchText = elements.searchInput.value.toLowerCase();
      
      // 필터링 적용
      const filteredUrls = state.allURLs.filter(item => {
        const { url, title, is_liked } = item;
        
        // 좋아요 필터링 (좋아요만 보기 모드)
        if (state.viewMode === 'liked' && !is_liked) {
          return false;
        }
        
        // 검색 필터링
        if (searchText && !url.toLowerCase().includes(searchText) && !title.toLowerCase().includes(searchText)) {
          return false;
        }
        
        return true;
      });
      
      renderCards(filteredUrls);
    }
    
    // 좋아요 토글
    async function toggleLike(url, card) {
      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 })
        });
        
        const data = await handleApiResponse(response);
        
        if (data.success) {
          // 상태 객체에서 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);
          
          // 좋아요 통계 업데이트
          updateLikeStats();
          
          // 좋아요만 보기 모드인 경우 목록 다시 필터링
          if (state.viewMode === 'liked') {
            filterAndRenderCards();
          }
        } 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 ${is_liked ? 'liked' : ''}`;
        
        // 카드 헤더
        const cardHeader = document.createElement('div');
        cardHeader.className = 'card-header';
        
        // 제목
        const titleEl = document.createElement('h3');
        titleEl.className = 'card-title';
        titleEl.textContent = title;
        cardHeader.appendChild(titleEl);
        
        card.appendChild(cardHeader);
        
        // 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' : ''}`;
        likeBtn.innerHTML = '♥';
        likeBtn.title = is_liked ? '좋아요 취소' : '좋아요';
        
        likeBtn.addEventListener('click', (e) => {
          e.preventDefault();
          toggleLike(url, card);
        });
        
        card.appendChild(likeBtn);
        
        // 좋아요 배지 (좋아요 상태일 때만)
        if (is_liked) {
          const likeBadge = document.createElement('div');
          likeBadge.className = 'like-badge';
          likeBadge.textContent = '좋아요';
          card.appendChild(likeBadge);
        }
        
        // 카드 추가
        elements.cardsContainer.appendChild(card);
      });
    }
    
    // 뷰 모드 변경
    function changeViewMode(mode) {
      state.viewMode = mode;
      
      // 버튼 활성화 상태 업데이트
      elements.allUrlsBtn.classList.toggle('active', mode === 'all');
      elements.likedUrlsBtn.classList.toggle('active', mode === 'liked');
      
      // 카드 다시 렌더링
      filterAndRenderCards();
    }
    
// 이벤트 리스너 설정
    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.searchInput.addEventListener('input', () => {
      // 디바운싱 (입력 지연 처리)
      clearTimeout(state.searchTimeout);
      state.searchTimeout = setTimeout(filterAndRenderCards, 300);
    });
    
    // 필터 버튼 이벤트 리스너
    elements.allUrlsBtn.addEventListener('click', () => changeViewMode('all'));
    elements.likedUrlsBtn.addEventListener('click', () => changeViewMode('liked'));
    
    // 초기화
    checkSessionStatus();
  </script>
</body>
</html>
''')
    
    # 허깅페이스 스페이스에서는 7860 포트 사용
    app.run(host='0.0.0.0', port=7860)