from flask import Flask, render_template, request, jsonify, session
import requests
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 목록
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:
        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')

@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 = '인증된 사용자'
    
    # 세션에 저장
    session['token'] = token
    session['username'] = username
    
    return jsonify({
        'success': True, 
        'username': username
    })

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

@app.route('/api/urls', methods=['GET'])
def get_urls():
    results = []
    for url in HUGGINGFACE_URLS:
        title = extract_title(url)
        model_info = extract_model_info(url)
        
        if not model_info:
            continue
        
        results.append({
            'url': url,
            'title': title,
            'model_info': model_info
        })
    
    return jsonify(results)

@app.route('/api/session-status', methods=['GET'])
def session_status():
    return jsonify({
        'logged_in': 'token' 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;
    }
    
    .container {
      max-width: 1200px;
      margin: 0 auto;
      padding: 1rem;
    }
    
    .header {
      background-color: #f0f0f0;
      padding: 1rem;
      border-radius: 5px;
      margin-bottom: 1rem;
    }
    
    .stats-bar {
      background-color: #e9f7ef;
      padding: 1rem;
      border-radius: 5px;
      margin-bottom: 1rem;
      display: flex;
      justify-content: space-between;
      align-items: center;
    }
    
    .stats-bar .liked-count {
      font-weight: bold;
      color: #e74c3c;
    }
    
    .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 #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: #e74c3c;
      background-color: #fff9f9;
    }
    
    .card a {
      text-decoration: none;
      color: #2980b9;
      word-break: break-all;
      display: block;
      margin-top: 0.5rem;
    }
    
    .card a:hover {
      text-decoration: underline;
    }
    
    .card-title {
      margin-top: 0;
      color: #333;
      font-size: 1.2rem;
      padding-right: 30px; /* 좋아요 버튼 공간 확보 */
    }
    
    .like-button {
      position: absolute;
      top: 1rem;
      right: 1rem;
      border: none;
      background: transparent;
      font-size: 1.8rem;
      cursor: pointer;
      transition: all 0.2s;
      z-index: 2;
    }
    
    .like-button:hover {
      transform: scale(1.2);
    }
    
    .like-button.liked {
      color: #e74c3c;
      text-shadow: 0 0 5px rgba(231, 76, 60, 0.3);
    }
    
    .like-button.not-liked {
      color: #ccc;
    }
    
    .like-label {
      position: absolute;
      top: 10px;
      left: 10px;
      background-color: #e74c3c;
      color: white;
      padding: 2px 8px;
      border-radius: 10px;
      font-size: 0.7rem;
      font-weight: bold;
      display: none;
    }
    
    .card.liked .like-label {
      display: block;
    }
    
    .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;
    }
    
    .view-toggle {
      margin-bottom: 1rem;
      display: flex;
      gap: 0.5rem;
    }
    
    .view-toggle button {
      background-color: #f8f9fa;
      color: #333;
      border: 1px solid #ddd;
    }
    
    .view-toggle button.active {
      background-color: #4CAF50;
      color: white;
    }
    
    @media (max-width: 768px) {
      .user-controls {
        flex-direction: column;
        align-items: flex-start;
      }
      
      .user-controls > div {
        margin-bottom: 1rem;
      }
      
      .card {
        width: 100%;
      }
      
      .stats-bar {
        flex-direction: column;
        gap: 0.5rem;
        align-items: flex-start;
      }
    }
  </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="statsBar" class="stats-bar" style="display: none;">
      <div>
        총 <span id="totalCount">0</span>개 중 <span id="likedCount" class="liked-count">0</span>개 좋아요 함
      </div>
      <div class="view-toggle">
        <button id="allViewBtn" class="active">전체 보기</button>
        <button id="likedViewBtn">좋아요만 보기</button>
      </div>
    </div>
    
    <div class="filter-controls">
      <label>
        <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'),
      searchInput: document.getElementById('searchInput'),
      loginSection: document.getElementById('loginSection'),
      loggedInSection: document.getElementById('loggedInSection'),
      statsBar: document.getElementById('statsBar'),
      totalCount: document.getElementById('totalCount'),
      likedCount: document.getElementById('likedCount'),
      allViewBtn: document.getElementById('allViewBtn'),
      likedViewBtn: document.getElementById('likedViewBtn')
    };
    
    // 애플리케이션 상태
    const state = {
      username: null,
      likedURLs: {},
      allURLs: [],
      isLoading: false,
      viewMode: 'all' // 'all' 또는 'liked'
    };
    
    // 로컬 스토리지 키
    function getLikesStorageKey(username) {
      return `hf_local_likes_${username}`;
    }
    
    // 로컬 스토리지에서 좋아요 정보 로드
    function loadLikesFromStorage() {
      if (!state.username) return {};
      
      const key = getLikesStorageKey(state.username);
      const savedLikes = localStorage.getItem(key);
      return savedLikes ? JSON.parse(savedLikes) : {};
    }
    
    // 로컬 스토리지에 좋아요 정보 저장
    function saveLikesToStorage() {
      if (!state.username) return;
      
      const key = getLikesStorageKey(state.username);
      localStorage.setItem(key, JSON.stringify(state.likedURLs));
    }
    
    // 좋아요 통계 업데이트
    function updateLikeStats() {
      const totalCount = state.allURLs.length;
      const likedCount = Object.keys(state.likedURLs).length;
      
      elements.totalCount.textContent = totalCount;
      elements.likedCount.textContent = likedCount;
    }
    
    // 로딩 상태 표시 함수
    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';
          elements.statsBar.style.display = 'flex';
          
          // 로컬 스토리지에서 좋아요 정보 로드
          state.likedURLs = loadLikesFromStorage();
          
          // 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.likedURLs = loadLikesFromStorage();
          
          elements.currentUser.textContent = state.username;
          elements.loginSection.style.display = 'none';
          elements.loggedInSection.style.display = 'block';
          elements.statsBar.style.display = 'flex';
          
          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.likedURLs = {};
          state.allURLs = [];
          
          elements.currentUser.textContent = '로그인되지 않음';
          elements.tokenInput.value = '';
          elements.loginSection.style.display = 'block';
          elements.loggedInSection.style.display = 'none';
          elements.statsBar.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);
        
        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 } = item;
        
        // 좋아요 필터링 (좋아요만 보기 모드)
        if (state.viewMode === 'liked' && !state.likedURLs[url]) {
          return false;
        }
        
        // 검색 필터링
        if (searchText && !url.toLowerCase().includes(searchText) && !title.toLowerCase().includes(searchText)) {
          return false;
        }
        
        return true;
      });
      
      renderCards(filteredUrls);
    }
    
    // 좋아요 토글
    function toggleLike(url, card) {
      if (!state.username) {
        showMessage('좋아요를 하려면 허깅페이스 API 토큰으로 인증이 필요합니다.', true);
        return;
      }
      
      setLoading(true);
      
      try {
        // 현재 좋아요 상태 확인
        const isCurrentlyLiked = state.likedURLs[url] || false;
        
        // 상태 토글
        if (isCurrentlyLiked) {
          delete state.likedURLs[url];
          card.classList.remove('liked');
          const likeBtn = card.querySelector('.like-button');
          if (likeBtn) {
            likeBtn.classList.remove('liked');
            likeBtn.classList.add('not-liked');
          }
          showMessage('좋아요를 취소했습니다.');
        } else {
          state.likedURLs[url] = true;
          card.classList.add('liked');
          const likeBtn = card.querySelector('.like-button');
          if (likeBtn) {
            likeBtn.classList.add('liked');
            likeBtn.classList.remove('not-liked');
          }
          showMessage('좋아요를 추가했습니다.');
        }
        
        // 로컬 스토리지에 저장
        saveLikesToStorage();
        
        // 좋아요 통계 업데이트
        updateLikeStats();
        
        // 좋아요만 보기 모드인 경우 목록 다시 렌더링
        if (state.viewMode === 'liked') {
          filterAndRenderCards();
        }
      } 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 } = item;
        const isLiked = state.likedURLs[url] || false;
        
        // 카드 생성
        const card = document.createElement('div');
        card.className = `card ${isLiked ? 'liked' : ''}`;
        
        // 좋아요 라벨
        const likeLabel = document.createElement('span');
        likeLabel.className = 'like-label';
        likeLabel.textContent = '좋아요';
        card.appendChild(likeLabel);
        
        // 제목
        const titleEl = document.createElement('h3');
        titleEl.className = 'card-title';
        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 ${isLiked ? 'liked' : 'not-liked'}`;
        likeBtn.textContent = '♥';
        likeBtn.title = isLiked ? '좋아요 취소' : '좋아요';
        
        likeBtn.addEventListener('click', function(e) {
          e.preventDefault();
          toggleLike(url, card);
        });
        card.appendChild(likeBtn);
        
        // 카드 추가
        elements.cardsContainer.appendChild(card);
      });
    }
    
    // 모드 변경 함수
    function changeViewMode(mode) {
      state.viewMode = mode;
      
      // 버튼 상태 업데이트
      elements.allViewBtn.classList.toggle('active', mode === 'all');
      elements.likedViewBtn.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.allViewBtn.addEventListener('click', () => changeViewMode('all'));
    elements.likedViewBtn.addEventListener('click', () => changeViewMode('liked'));
    
    // 초기화
    checkSessionStatus();
  </script>
</body>
</html>
''')
    
    # 허깅페이스 스페이스에서는 7860 포트 사용
    app.run(host='0.0.0.0', port=7860)