ZeroGPU-Leader / app.py
seawolf2357's picture
Update app.py
5423956 verified
raw
history blame
29.9 kB
from flask import Flask, render_template, request, jsonify, session
import requests
from bs4 import BeautifulSoup
import os
from datetime import timedelta
import logging
import time
# ๋กœ๊น… ์„ค์ •
logging.basicConfig(level=logging.INFO,
format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)
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:
logger.error(f"ํ† ํฐ ๊ฒ€์ฆ ์˜ค๋ฅ˜: {e}")
return False, None
# ์›น ์Šคํฌ๋ž˜ํ•‘์œผ๋กœ ์ข‹์•„์š” ์ƒํƒœ ํ™•์ธ
def check_like_status_by_scraping(url, token):
headers = {
"Authorization": f"Bearer {token}",
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
}
try:
# ํŽ˜์ด์ง€ ์š”์ฒญ
response = requests.get(url, headers=headers)
if not response.ok:
logger.warning(f"ํŽ˜์ด์ง€ ์š”์ฒญ ์‹คํŒจ: {url}, ์ƒํƒœ ์ฝ”๋“œ: {response.status_code}")
return False
# HTML ํŒŒ์‹ฑ
soup = BeautifulSoup(response.text, 'html.parser')
# ์ข‹์•„์š” ๋ฒ„ํŠผ ์ฐพ๊ธฐ (๋‹ค์–‘ํ•œ ์„ ํƒ์ž ์‹œ๋„)
like_button = None
selectors = [
'.like-button-container button',
'button[aria-label="Like"]',
'button.like-button',
'button[data-testid="like-button"]',
'button.heart-button'
]
for selector in selectors:
like_button = soup.select_one(selector)
if like_button:
break
if not like_button:
logger.warning(f"์ข‹์•„์š” ๋ฒ„ํŠผ์„ ์ฐพ์„ ์ˆ˜ ์—†์Œ: {url}")
return False
# ์ข‹์•„์š” ์ƒํƒœ ํ™•์ธ (ํด๋ž˜์Šค, aria-pressed ์†์„ฑ ๋“ฑ์œผ๋กœ ํ™•์ธ)
is_liked = False
# ํด๋ž˜์Šค๋กœ ํ™•์ธ
if 'liked' in like_button.get('class', []) or 'active' in like_button.get('class', []):
is_liked = True
# aria-pressed ์†์„ฑ์œผ๋กœ ํ™•์ธ
elif like_button.get('aria-pressed') == 'true':
is_liked = True
# ๋‚ด๋ถ€ ํ…์ŠคํŠธ ๋˜๋Š” ์•„์ด์ฝ˜์œผ๋กœ ํ™•์ธ
elif like_button.find('span', class_='liked') or like_button.find('svg', class_='liked'):
is_liked = True
logger.info(f"์Šคํฌ๋ž˜ํ•‘ ๊ฒฐ๊ณผ: {url} - ์ข‹์•„์š” {is_liked}")
return is_liked
except Exception as e:
logger.error(f"์Šคํฌ๋ž˜ํ•‘ ์˜ค๋ฅ˜ ({url}): {e}")
return False
# ์ „์ฒด URL ๋ชฉ๋ก์˜ ์ข‹์•„์š” ์ƒํƒœ ์Šคํฌ๋ž˜ํ•‘
def scrape_all_like_status(token):
like_status = {}
for url in HUGGINGFACE_URLS:
try:
# ๊ณผ๋„ํ•œ ์š”์ฒญ ๋ฐฉ์ง€๋ฅผ ์œ„ํ•œ ์ง€์—ฐ
time.sleep(1)
is_liked = check_like_status_by_scraping(url, token)
like_status[url] = is_liked
logger.info(f"์ข‹์•„์š” ์ƒํƒœ ํ™•์ธ: {url} - {is_liked}")
except Exception as e:
logger.error(f"URL ์ฒ˜๋ฆฌ ์ค‘ ์˜ค๋ฅ˜: {url} - {e}")
like_status[url] = False
return like_status
@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
# ์›น ์Šคํฌ๋ž˜ํ•‘์œผ๋กœ ์ข‹์•„์š” ์ƒํƒœ ํ™•์ธ
# ์ฐธ๊ณ : ์ด ์ž‘์—…์ด ์‹œ๊ฐ„์ด ์˜ค๋ž˜ ๊ฑธ๋ฆด ์ˆ˜ ์žˆ์œผ๋ฏ€๋กœ ๋น„๋™๊ธฐ๋กœ ์ฒ˜๋ฆฌํ•˜๋Š” ๊ฒƒ์ด ์ข‹์Šต๋‹ˆ๋‹ค
# ํ˜„์žฌ๋Š” ์˜ˆ์‹œ๋กœ ๋™๊ธฐ ๋ฐฉ์‹์œผ๋กœ ๊ตฌํ˜„ํ–ˆ์Šต๋‹ˆ๋‹ค
try:
like_status = scrape_all_like_status(token)
session['like_status'] = like_status
except Exception as e:
logger.error(f"์ข‹์•„์š” ์ƒํƒœ ์Šคํฌ๋ž˜ํ•‘ ์ค‘ ์˜ค๋ฅ˜: {e}")
session['like_status'] = {}
return jsonify({
'success': True,
'username': username
})
@app.route('/api/logout', methods=['POST'])
def logout():
session.pop('token', None)
session.pop('username', None)
session.pop('like_status', None)
return jsonify({'success': True})
@app.route('/api/urls', methods=['GET'])
def get_urls():
like_status = session.get('like_status', {})
results = []
for url in HUGGINGFACE_URLS:
title = extract_title(url)
model_info = extract_model_info(url)
if not model_info:
continue
# ์ข‹์•„์š” ์ƒํƒœ ํ™•์ธ
is_liked = like_status.get(url, False)
results.append({
'url': url,
'title': title,
'model_info': model_info,
'is_liked': is_liked
})
return jsonify(results)
@app.route('/api/toggle-like', methods=['POST'])
def toggle_like():
if 'token' not in session:
return jsonify({'success': False, 'message': '๋กœ๊ทธ์ธ์ด ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.'})
data = request.json
url = data.get('url')
if not url:
return jsonify({'success': False, 'message': 'URL์ด ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.'})
# ํ˜„์žฌ ์ข‹์•„์š” ์ƒํƒœ ํ™•์ธ
like_status = session.get('like_status', {})
current_status = like_status.get(url, False)
# ์‹ค์ œ๋กœ๋Š” ์—ฌ๊ธฐ์„œ ํ—ˆ๊น…ํŽ˜์ด์Šค API๋ฅผ ํ˜ธ์ถœํ•˜์—ฌ ์ข‹์•„์š” ์ƒํƒœ๋ฅผ ๋ณ€๊ฒฝํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค
# ํ˜„์žฌ๋Š” ์˜ˆ์‹œ๋กœ ์„ธ์…˜์—๋งŒ ์ƒํƒœ๋ฅผ ์ €์žฅํ•ฉ๋‹ˆ๋‹ค
like_status[url] = not current_status
session['like_status'] = like_status
return jsonify({
'success': True,
'is_liked': like_status[url],
'message': '์ข‹์•„์š”๋ฅผ ์ถ”๊ฐ€ํ–ˆ์Šต๋‹ˆ๋‹ค.' if like_status[url] else '์ข‹์•„์š”๋ฅผ ์ทจ์†Œํ–ˆ์Šต๋‹ˆ๋‹ค.'
})
@app.route('/api/refresh-likes', methods=['POST'])
def refresh_likes():
if 'token' not in session:
return jsonify({'success': False, 'message': '๋กœ๊ทธ์ธ์ด ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.'})
try:
# ์›น ์Šคํฌ๋ž˜ํ•‘์œผ๋กœ ์ข‹์•„์š” ์ƒํƒœ ์ƒˆ๋กœ๊ณ ์นจ
like_status = scrape_all_like_status(session['token'])
session['like_status'] = like_status
return jsonify({
'success': True,
'message': '์ข‹์•„์š” ์ƒํƒœ๊ฐ€ ์ƒˆ๋กœ๊ณ ์นจ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.',
'like_status': like_status
})
except Exception as e:
logger.error(f"์ข‹์•„์š” ์ƒํƒœ ์ƒˆ๋กœ๊ณ ์นจ ์ค‘ ์˜ค๋ฅ˜: {e}")
return jsonify({
'success': False,
'message': f'์ข‹์•„์š” ์ƒํƒœ ์ƒˆ๋กœ๊ณ ์นจ ์ค‘ ์˜ค๋ฅ˜: {str(e)}'
})
@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.refresh {
background-color: #2196F3;
}
button.refresh:hover {
background-color: #0b7dda;
}
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;
}
.note {
padding: 0.5rem;
background-color: #fffde7;
border-left: 3px solid #ffd600;
margin-bottom: 1rem;
font-size: 0.9rem;
}
@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="refreshButton" class="refresh">์ƒˆ๋กœ๊ณ ์นจ</button>
<button id="logoutButton" class="logout">๋กœ๊ทธ์•„์›ƒ</button>
</div>
</div>
</div>
<div class="note">
<p><strong>์ฐธ๊ณ :</strong> ์ด ํŽ˜์ด์ง€๋Š” ์›น ์Šคํฌ๋ž˜ํ•‘ ๋ฐฉ์‹์œผ๋กœ ์ข‹์•„์š” ์ƒํƒœ๋ฅผ ๊ฐ€์ ธ์˜ต๋‹ˆ๋‹ค. ์ข‹์•„์š” ์ƒํƒœ๊ฐ€ ์ •ํ™•ํ•˜์ง€ ์•Š๊ฑฐ๋‚˜ ์ง€์—ฐ๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. '์ƒˆ๋กœ๊ณ ์นจ' ๋ฒ„ํŠผ์„ ํด๋ฆญํ•˜์—ฌ ์ตœ์‹  ์ƒํƒœ๋ฅผ ๊ฐ€์ ธ์˜ฌ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.</p>
</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'),
refreshButton: document.getElementById('refreshButton'),
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);
}
}
// ์ข‹์•„์š” ์ƒํƒœ ์ƒˆ๋กœ๊ณ ์นจ
async function refreshLikes() {
setLoading(true);
try {
const response = await fetch('/api/refresh-likes', {
method: 'POST'
});
const data = await handleApiResponse(response);
if (data.success) {
// URL ๋ชฉ๋ก ๋‹ค์‹œ ๋กœ๋“œ
loadUrls();
showMessage('์ข‹์•„์š” ์ƒํƒœ๊ฐ€ ์ƒˆ๋กœ๊ณ ์นจ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.');
} else {
showMessage(data.message || '์ข‹์•„์š” ์ƒํƒœ ์ƒˆ๋กœ๊ณ ์นจ์— ์‹คํŒจํ–ˆ์Šต๋‹ˆ๋‹ค.', true);
}
} 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 data = await handleApiResponse(response);
state.allURLs = data;
updateLikeStats();
renderCards();
} catch (error) {
console.error('URL ๋ชฉ๋ก ๋กœ๋“œ ์˜ค๋ฅ˜:', error);
showMessage(`URL ๋ชฉ๋ก ๋กœ๋“œ ์˜ค๋ฅ˜: ${error.message}`, true);
} finally {
setLoading(false);
}
}
// ์ข‹์•„์š” ํ† ๊ธ€
async function toggleLike(url) {
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 urlObj = state.allURLs.find(item => item.url === url);
if (urlObj) {
urlObj.is_liked = data.is_liked;
updateLikeStats();
renderCards();
}
showMessage(data.message);
} else {
showMessage(data.message || '์ข‹์•„์š” ์ƒํƒœ ๋ณ€๊ฒฝ์— ์‹คํŒจํ–ˆ์Šต๋‹ˆ๋‹ค.', true);
}
} catch (error) {
console.error('์ข‹์•„์š” ํ† ๊ธ€ ์˜ค๋ฅ˜:', error);
showMessage(`์ข‹์•„์š” ํ† ๊ธ€ ์˜ค๋ฅ˜: ${error.message}`, true);
}
}
// ์นด๋“œ ๋ Œ๋”๋ง
function renderCards() {
elements.cardsContainer.innerHTML = '';
let urlsToShow = state.allURLs;
// ๊ฒ€์ƒ‰์–ด๋กœ ํ•„ํ„ฐ๋ง
const searchTerm = elements.searchInput.value.trim().toLowerCase();
if (searchTerm) {
urlsToShow = urlsToShow.filter(item =>
item.url.toLowerCase().includes(searchTerm) ||
item.title.toLowerCase().includes(searchTerm)
);
}
// ๋ณด๊ธฐ ๋ชจ๋“œ๋กœ ํ•„ํ„ฐ๋ง (์ „์ฒด ๋˜๋Š” ์ข‹์•„์š”๋งŒ)
if (state.viewMode === 'liked') {
urlsToShow = urlsToShow.filter(item => item.is_liked);
}
if (urlsToShow.length === 0) {
const emptyMessage = document.createElement('div');
emptyMessage.textContent = 'ํ‘œ์‹œํ•  URL์ด ์—†์Šต๋‹ˆ๋‹ค.';
emptyMessage.style.padding = '1rem';
emptyMessage.style.width = '100%';
emptyMessage.style.textAlign = 'center';
elements.cardsContainer.appendChild(emptyMessage);
return;
}
// ์นด๋“œ ์ƒ์„ฑ
urlsToShow.forEach(item => {
const card = document.createElement('div');
card.className = `card ${item.is_liked ? 'liked' : ''}`;
if (item.is_liked) {
const badge = document.createElement('div');
badge.className = 'like-badge';
badge.textContent = '์ข‹์•„์š”';
card.appendChild(badge);
}
const cardContent = `
<div class="card-header">
<h3 class="card-title">${item.title}</h3>
<button class="like-button ${item.is_liked ? 'liked' : ''}" data-url="${item.url}">
โค
</button>
</div>
<div class="card-body">
<a href="${item.url}" target="_blank">${item.url}</a>
<div style="margin-top: 0.5rem;">
<span>์†Œ์œ ์ž: ${item.model_info.owner}</span>
</div>
<div>
<span>์ €์žฅ์†Œ: ${item.model_info.repo}</span>
</div>
<div>
<span>์œ ํ˜•: ${item.model_info.type}</span>
</div>
</div>
`;
card.innerHTML = cardContent;
elements.cardsContainer.appendChild(card);
// ์ข‹์•„์š” ๋ฒ„ํŠผ ์ด๋ฒคํŠธ ์ถ”๊ฐ€
const likeButton = card.querySelector('.like-button');
likeButton.addEventListener('click', (e) => {
e.preventDefault();
const url = e.target.dataset.url;
toggleLike(url);
});
});
}
// ์ด๋ฒคํŠธ ๋ฆฌ์Šค๋„ˆ ๋“ฑ๋ก
function registerEventListeners() {
// ๋กœ๊ทธ์ธ ๋ฒ„ํŠผ
elements.loginButton.addEventListener('click', () => {
login(elements.tokenInput.value);
});
// ์—”ํ„ฐ ํ‚ค๋กœ ๋กœ๊ทธ์ธ
elements.tokenInput.addEventListener('keydown', (e) => {
if (e.key === 'Enter') {
login(elements.tokenInput.value);
}
});
// ๋กœ๊ทธ์•„์›ƒ ๋ฒ„ํŠผ
elements.logoutButton.addEventListener('click', logout);
// ์ƒˆ๋กœ๊ณ ์นจ ๋ฒ„ํŠผ
elements.refreshButton.addEventListener('click', refreshLikes);
// ๊ฒ€์ƒ‰ ์ž…๋ ฅ ํ•„๋“œ
elements.searchInput.addEventListener('input', renderCards);
// ํ•„ํ„ฐ ๋ฒ„ํŠผ - ์ „์ฒด ๋ณด๊ธฐ
elements.allUrlsBtn.addEventListener('click', () => {
elements.allUrlsBtn.classList.add('active');
elements.likedUrlsBtn.classList.remove('active');
state.viewMode = 'all';
renderCards();
});
// ํ•„ํ„ฐ ๋ฒ„ํŠผ - ์ข‹์•„์š”๋งŒ ๋ณด๊ธฐ
elements.likedUrlsBtn.addEventListener('click', () => {
elements.likedUrlsBtn.classList.add('active');
elements.allUrlsBtn.classList.remove('active');
state.viewMode = 'liked';
renderCards();
});
}
// ์ดˆ๊ธฐํ™” ํ•จ์ˆ˜
function init() {
registerEventListeners();
checkSessionStatus();
}
// ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜ ์ดˆ๊ธฐํ™”
init();
</script>
</body>
</html>
'''
)
app.run(debug=True, host='0.0.0.0', port=7860)