12345

.idea/.gitignore

@@ -0,0 +1,8 @@
+# 默认忽略的文件
+/shelf/
+/workspace.xml
+# 基于编辑器的 HTTP 客户端请求
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,34 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="Eslint" enabled="true" level="WARNING" enabled_by_default="true" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="20">
+            <item index="0" class="java.lang.String" itemvalue="httpx" />
+            <item index="1" class="java.lang.String" itemvalue="inquirer" />
+            <item index="2" class="java.lang.String" itemvalue="better_proxy" />
+            <item index="3" class="java.lang.String" itemvalue="curl_cffi" />
+            <item index="4" class="java.lang.String" itemvalue="art" />
+            <item index="5" class="java.lang.String" itemvalue="pydantic" />
+            <item index="6" class="java.lang.String" itemvalue="paddleocr" />
+            <item index="7" class="java.lang.String" itemvalue="PyYAML" />
+            <item index="8" class="java.lang.String" itemvalue="aiofiles" />
+            <item index="9" class="java.lang.String" itemvalue="rich" />
+            <item index="10" class="java.lang.String" itemvalue="numpy" />
+            <item index="11" class="java.lang.String" itemvalue="loguru" />
+            <item index="12" class="java.lang.String" itemvalue="imap_tools" />
+            <item index="13" class="java.lang.String" itemvalue="paddlepaddle" />
+            <item index="14" class="java.lang.String" itemvalue="names" />
+            <item index="15" class="java.lang.String" itemvalue="tortoise-orm" />
+            <item index="16" class="java.lang.String" itemvalue="colorama" />
+            <item index="17" class="java.lang.String" itemvalue="pytz" />
+            <item index="18" class="java.lang.String" itemvalue="urllib3" />
+            <item index="19" class="java.lang.String" itemvalue="aiocsv" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="PPX" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/panel.iml" filepath="$PROJECT_DIR$/.idea/panel.iml" />
+    </modules>
+  </component>
+</project>

.idea/panel.iml

@@ -0,0 +1,23 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="Flask">
+    <option name="enabled" value="true" />
+  </component>
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="PyDocumentationSettings">
+    <option name="format" value="PLAIN" />
+    <option name="myDocStringFormat" value="Plain" />
+  </component>
+  <component name="TemplatesService">
+    <option name="TEMPLATE_CONFIGURATION" value="Jinja2" />
+    <option name="TEMPLATE_FOLDERS">
+      <list>
+        <option value="$MODULE_DIR$/templates" />
+      </list>
+    </option>
+  </component>
+</module>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

Dockerfile

@@ -0,0 +1,23 @@
+# 使用官方 Python 镜像作为基础镜像
+FROM python:3.9-slim
+
+# 设置工作目录
+WORKDIR /app
+
+# 将当前目录下的所有文件复制到工作目录
+COPY . .
+
+# 安装项目依赖
+RUN pip install Flask python-dotenv huggingface_hub requests gunicorn flask-socketio python-engineio python-socketio eventlet
+
+# 开放应用程序的端口
+EXPOSE 5000
+
+# 设置环境变量（可选，如果需要传递 Docker 环境中的环境变量）
+# ENV USERNAME=your_username
+# ENV PASSWORD=your_password
+# ENV HF_TOKENS=token1,token2,token3
+# ENV API_KEY=your_apikey
+
+# 定义启动命令
+CMD ["gunicorn", "--worker-class", "eventlet", "--bind", "0.0.0.0:5000", "app:app"]

api.py

@@ -0,0 +1,133 @@
+from flask import Blueprint, request, jsonify
+from functools import wraps
+from huggingface_hub import HfApi
+from config import API_KEY
+
+api = Blueprint('api', __name__, url_prefix='/api/v1')
+
+def require_api_key(f):
+    @wraps(f)
+    def decorated(*args, **kwargs):
+        auth_header = request.headers.get('Authorization')
+        if not auth_header:
+            return jsonify({'error': 'No Authorization header'}), 401
+        
+        try:
+            scheme, token = auth_header.split()
+            if scheme.lower() != 'bearer':
+                return jsonify({'error': 'Invalid authorization scheme'}), 401
+            if token != API_KEY:
+                return jsonify({'error': 'Invalid API key'}), 401
+        except ValueError:
+            return jsonify({'error': 'Invalid Authorization header format'}), 401
+        
+        return f(*args, **kwargs)
+    return decorated
+
+@api.route('/info/<token>', methods=['GET'])
+@require_api_key
+def list_spaces(token):
+    """列出所有空间"""
+    try:
+        hf_api = HfApi(token=token)
+        # 验证token
+        try:
+            user_info = hf_api.whoami()
+            username = user_info["name"]
+        except Exception:
+            return jsonify({'error': 'Invalid HuggingFace token'}), 401
+
+        spaces = list(hf_api.list_spaces(author=username))
+        space_list = []
+        
+        for space in spaces:
+            try:
+                space_info = hf_api.space_info(repo_id=space.id)
+                space_list.append(space_info.id)
+            except Exception as e:
+                continue
+
+        return jsonify({
+            'spaces': space_list,
+            'total': len(space_list)
+        })
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@api.route('/info/<token>/<path:space_id>', methods=['GET'])
+@require_api_key
+def get_space_info(token, space_id):
+    """获取特定空间信息"""
+    try:
+        hf_api = HfApi(token=token)
+        try:
+            space_info = hf_api.space_info(repo_id=space_id)
+        except Exception:
+            return jsonify({'error': 'Space not found'}), 404
+
+        # 获取运行状态
+        status = "未知状态"
+        if space_info.runtime:
+            status = space_info.runtime.stage if hasattr(space_info.runtime, 'stage') else "未知状态"
+
+        return jsonify({
+            'id': space_info.id,
+            'status': status,
+            'last_modified': space_info.lastModified.isoformat() if space_info.lastModified else None,
+            'created_at': space_info.created_at.isoformat() if space_info.created_at else None,
+            'sdk': space_info.sdk,
+            'tags': space_info.tags,
+            'private': space_info.private
+        })
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@api.route('/action/<token>/<path:space_id>/restart', methods=['POST'])
+@require_api_key
+def restart_space(token, space_id):
+    """重启空间"""
+    try:
+        hf_api = HfApi(token=token)
+        try:
+            hf_api.restart_space(repo_id=space_id)
+            return jsonify({
+                'success': True,
+                'message': f'Space {space_id} restart initiated successfully'
+            })
+        except Exception as e:
+            return jsonify({
+                'success': False,
+                'error': str(e)
+            }), 400
+
+    except Exception as e:
+        return jsonify({
+            'success': False,
+            'error': str(e)
+        }), 500
+
+@api.route('/action/<token>/<path:space_id>/rebuild', methods=['POST'])
+@require_api_key
+def rebuild_space(token, space_id):
+    """重建空间"""
+    try:
+        hf_api = HfApi(token=token)
+        try:
+            hf_api.restart_space(repo_id=space_id, factory_reboot=True)
+            return jsonify({
+                'success': True,
+                'message': f'Space {space_id} rebuild initiated successfully'
+            })
+        except Exception as e:
+            return jsonify({
+                'success': False,
+                'error': str(e)
+            }), 400
+
+    except Exception as e:
+        return jsonify({
+            'success': False,
+            'error': str(e)
+        }), 500

app.py

@@ -0,0 +1,252 @@
+from flask import Flask, render_template, request, redirect, url_for, session, jsonify
+from flask_socketio import SocketIO, emit, disconnect
+from dotenv import load_dotenv
+from huggingface_hub import HfApi
+from functools import wraps
+from datetime import datetime, timedelta
+from concurrent.futures import ThreadPoolExecutor
+from config import USERNAME, PASSWORD, HF_TOKENS
+from api import api as api_blueprint
+import concurrent.futures
+import threading
+import logging
+import time
+import os
+
+# 配置日志
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# 加载环境变量
+load_dotenv()
+
+app = Flask(__name__)
+app.secret_key = os.urandom(24)
+app.register_blueprint(api_blueprint)
+socketio = SocketIO(app)
+
+# 缓存管理
+class SpaceCache:
+    def __init__(self):
+        self.spaces = {}
+        self.last_update = None
+        self.lock = threading.Lock()
+        self.active_clients = set()  # 跟踪活动的客户端
+
+    def update_all(self, spaces_data):
+        with self.lock:
+            self.spaces = {space['repo_id']: space for space in spaces_data}
+            self.last_update = datetime.now()
+
+    def get_all(self):
+        with self.lock:
+            return list(self.spaces.values()) if self.spaces else []
+
+    def is_expired(self, expire_minutes=5):
+        if not self.last_update:
+            return True
+        return datetime.now() - self.last_update > timedelta(minutes=expire_minutes)
+
+    def add_client(self, client_id):
+        with self.lock:
+            self.active_clients.add(client_id)
+
+    def remove_client(self, client_id):
+        with self.lock:
+            self.active_clients.discard(client_id)
+
+    def has_active_clients(self):
+        with self.lock:
+            return len(self.active_clients) > 0
+
+space_cache = SpaceCache()
+
+# WebSocket 事件处理
+@socketio.on('connect')
+def handle_connect():
+    if 'authenticated' not in session or not session['authenticated']:
+        disconnect()
+        return
+
+    client_id = request.sid
+    space_cache.add_client(client_id)
+    logger.info(f"Client connected: {client_id}")
+
+@socketio.on('disconnect')
+def handle_disconnect():
+    client_id = request.sid
+    space_cache.remove_client(client_id)
+    logger.info(f"Client disconnected: {client_id}")
+
+# 登录验证装饰器
+def login_required(f):
+    @wraps(f)
+    def decorated_function(*args, **kwargs):
+        if 'authenticated' not in session or not session['authenticated']:
+            return redirect(url_for('login'))
+        return f(*args, **kwargs)
+    return decorated_function
+
+def process_single_space(space, hf_api, username, token):
+    try:
+        space_info = hf_api.space_info(repo_id=space.id)
+        space_runtime = space_info.runtime
+        
+        status = "未知状态"
+        if space_runtime:
+            status = space_runtime.stage if hasattr(space_runtime, 'stage') else "未知状态"
+
+        return {
+            "repo_id": space_info.id,
+            "name": space_info.cardData.get('title') or space_info.id.split('/')[-1],
+            "owner": space_info.author,
+            "username": username,
+            "token": token,
+            "url": f"https://{space_info.author}-{space_info.id.split('/')[-1]}.hf.space",
+            "status": status,
+            "last_modified": space_info.lastModified.strftime("%Y-%m-%d %H:%M:%S") if space_info.lastModified else "未知",
+            "created_at": space_info.created_at.strftime("%Y-%m-%d %H:%M:%S") if space_info.created_at else "未知",
+            "sdk": space_info.sdk,
+            "tags": space_info.tags,
+            "private": space_info.private,
+            "app_port": space_info.cardData.get('app_port', '未知')
+        }
+    except Exception as e:
+        logger.error(f"处理 Space {space.id} 时出错: {e}")
+        return None
+
+def get_all_user_spaces():
+    # 检查缓存是否有效
+    if not space_cache.is_expired():
+        logger.info("从缓存获取 Spaces 数据")
+        return space_cache.get_all()
+
+    all_spaces = []
+    with ThreadPoolExecutor(max_workers=10) as executor:
+        for token in HF_TOKENS:
+            try:
+                hf_api = HfApi(token=token)
+                user_info = hf_api.whoami()
+                username = user_info["name"]
+                logger.info(f"获取到用户信息: {username}")
+
+                spaces = list(hf_api.list_spaces(author=username))
+                logger.info(f"获取到 {len(spaces)} 个 Spaces")
+
+                # 并行处理每个space
+                future_to_space = {
+                    executor.submit(process_single_space, space, hf_api, username, token): space 
+                    for space in spaces
+                }
+
+                for future in concurrent.futures.as_completed(future_to_space):
+                    space_data = future.result()
+                    if space_data:
+                        all_spaces.append(space_data)
+
+            except Exception as e:
+                logger.error(f"获取 Spaces 列表失败 (token: {token[:5]}...): {e}")
+                import traceback
+                traceback.print_exc()
+
+    # 按名称排序
+    all_spaces.sort(key=lambda x: x['name'].lower())
+    
+    # 更新缓存
+    space_cache.update_all(all_spaces)
+    
+    logger.info(f"总共获取到 {len(all_spaces)} 个 Spaces")
+    return all_spaces
+
+# 后台更新缓存的函数
+def update_cache_if_needed():
+    """在有活动客户端时更新缓存"""
+    while True:
+        try:
+            if space_cache.has_active_clients() and space_cache.is_expired():
+                logger.info("Updating cache due to active clients")
+                spaces = get_all_user_spaces()
+                space_cache.update_all(spaces)
+                socketio.emit('spaces_updated', {'timestamp': time.time()})
+        except Exception as e:
+            logger.error(f"Cache update failed: {e}")
+        time.sleep(60)  # 每分钟检查一次
+
+# 启动缓存更新线程
+update_thread = threading.Thread(target=update_cache_if_needed, daemon=True)
+update_thread.start()
+
+@app.route("/", methods=["GET", "POST"])
+def login():
+    if 'authenticated' in session and session['authenticated']:
+        return redirect(url_for('dashboard'))
+    
+    if request.method == "POST":
+        username = request.form.get("username")
+        password = request.form.get("password")
+        if username == USERNAME and password == PASSWORD:
+            session['authenticated'] = True
+            return redirect(url_for("dashboard"))
+        else:
+            return render_template("index.html", error="用户名或密码错误")
+    return render_template("index.html", error=None)
+
+@app.route("/logout")
+def logout():
+    session.clear()
+    return redirect(url_for('login'))
+
+@app.route("/dashboard")
+@login_required
+def dashboard():
+    spaces = get_all_user_spaces()
+    logger.info(f"Dashboard 显示 {len(spaces)} 个 Spaces")
+    return render_template("dashboard.html", spaces=spaces)
+
+@app.route("/api/space/<path:repo_id>/status")
+@login_required
+def get_space_status(repo_id):
+    spaces = get_all_user_spaces()
+    space = next((s for s in spaces if s["repo_id"] == repo_id), None)
+    if not space:
+        return jsonify({"error": "Space not found"}), 404
+    return jsonify({
+        "id": repo_id,
+        "status": space["status"]
+    })
+
+def restart_space(repo_id, token):
+    try:
+        hf_api = HfApi(token=token)
+        hf_api.restart_space(repo_id=repo_id)
+        return f"成功重启 Space: {repo_id}"
+    except Exception as e:
+        return f"重启 Space {repo_id} 失败: {e}"
+    
+def rebuild_space(repo_id, token):
+    try:
+        hf_api = HfApi(token=token)
+        hf_api.restart_space(repo_id=repo_id, factory_reboot=True)
+        return f"成功重建 Space: {repo_id}"
+    except Exception as e:
+        return f"重建 Space {repo_id} 失败: {e}"
+
+@app.route("/action/<action_type>/<path:repo_id>")
+@login_required
+def space_action(action_type, repo_id):
+    spaces = get_all_user_spaces()
+    space = next((s for s in spaces if s["repo_id"] == repo_id), None)
+    
+    if not space:
+        return "Space not found", 404
+    
+    if action_type == "restart":
+        message = restart_space(repo_id, space["token"])
+    elif action_type == "rebuild":
+        message = rebuild_space(repo_id, space["token"])
+    else:
+        message = "未知操作"
+    return render_template("action_result.html", message=message)
+
+if __name__ == "__main__":
+    socketio.run(app, host='0.0.0.0', port=5000)

config.py

@@ -0,0 +1,12 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+USERNAME = os.getenv("USERNAME")
+PASSWORD = os.getenv("PASSWORD")
+HF_TOKENS = os.getenv("HF_TOKENS", "").split(",")
+API_KEY = os.getenv('API_KEY')
+
+if not USERNAME or not PASSWORD or not HF_TOKENS or not API_KEY:
+    raise Exception("请在 .env 文件中配置 USERNAME、PASSWORD、HF_TOKENS 和 API_KEY")

templates/action_result.html

@@ -0,0 +1,97 @@
+<!DOCTYPE html>
+<html lang="zh">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>操作结果 - HF Space Manager</title>
+    <style>
+        * {
+            margin: 0;
+            padding: 0;
+            box-sizing: border-box;
+            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
+        }
+
+        body {
+            background-color: #f5f5f7;
+            min-height: 100vh;
+            display: flex;
+            flex-direction: column;
+        }
+
+        .container {
+            flex: 1;
+            display: flex;
+            flex-direction: column;
+            justify-content: center;
+            align-items: center;
+            padding: 2rem;
+        }
+
+        .result-card {
+            background: white;
+            padding: 2rem;
+            border-radius: 18px;
+            box-shadow: 0 4px 24px rgba(0, 0, 0, 0.1);
+            text-align: center;
+            max-width: 600px;
+            width: 100%;
+        }
+
+        .message {
+            color: #1d1d1f;
+            margin-bottom: 2rem;
+            font-size: 1.1rem;
+            line-height: 1.5;
+        }
+
+        .back-button {
+            display: inline-block;
+            padding: 0.75rem 1.5rem;
+            background: #0071e3;
+            color: white;
+            text-decoration: none;
+            border-radius: 8px;
+            transition: all 0.3s ease;
+        }
+
+        .back-button:hover {
+            background: #0077ED;
+        }
+
+        .footer {
+            text-align: center;
+            padding: 2rem;
+            color: #86868b;
+            font-size: 0.9rem;
+            background: white;
+            border-top: 1px solid #e5e5e7;
+        }
+
+        .footer a {
+            color: #0071e3;
+            text-decoration: none;
+            transition: color 0.3s ease;
+        }
+
+        .footer a:hover {
+            color: #0077ED;
+            text-decoration: underline;
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <div class="result-card">
+            <div class="message">{{ message }}</div>
+            <a href="/dashboard" class="back-button">返回控制面板</a>
+        </div>
+    </div>
+
+    <footer class="footer">
+        <a href="https://github.com/ssfun/hf-space-manager">HF Space Manager</a> 由 
+        <a href="https://github.com/ssfun">ssfun</a> 构建，源代码遵循
+        <a href="https://opensource.org/license/mit">MIT 协议</a>
+    </footer>
+</body>
+</html>

templates/dashboard.html

@@ -0,0 +1,411 @@
+<!DOCTYPE html>
+<html lang="zh">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>HF Space Manager -控制面板</title>
+    <style>
+        * {
+            margin: 0;
+            padding: 0;
+            box-sizing: border-box;
+            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
+        }
+
+        body {
+            background-color: #f5f5f7;
+            min-height: 100vh;
+            display: flex;
+            flex-direction: column;
+        }
+
+        .header {
+            background: rgba(255, 255, 255, 0.85);
+            backdrop-filter: blur(20px);
+            position: fixed;
+            top: 0;
+            left: 0;
+            right: 0;
+            padding: 1rem 2rem;
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+            box-shadow: 0 1px 0 rgba(0, 0, 0, 0.1);z-index: 1000;
+        }
+
+        .header h1 {
+            font-size: 1.5rem;
+            color: #1d1d1f;
+            font-weight: 600;
+        }
+
+        .logout {
+            background: none;
+            border: none;
+            color: #0071e3;
+            font-size: 1rem;
+            cursor: pointer;
+            padding: 8px 16px;
+            border-radius: 8px;
+            transition: all 0.3s ease;text-decoration: none;
+        }
+
+        .logout:hover {
+            background: rgba(0, 113, 227, 0.1);
+        }
+
+        .container {
+            flex: 1;
+            width: 100%;
+            max-width: 1200px;
+            margin: 100px auto 0;
+            padding: 0 2rem 2rem;
+        }
+
+        .loading-overlay {
+            position: fixed;
+            top: 0;
+            left: 0;
+            right: 0;
+            bottom: 0;
+            background: rgba(255, 255, 255, 0.8);
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            z-index: 9999;
+        }
+
+        .loading-spinner {
+            width: 50px;
+            height: 50px;
+            border: 5px solid #f3f3f3;
+            border-top: 5px solid #0071e3;
+            border-radius: 50%;
+            animation: spin 1s linear infinite;
+        }
+
+        @keyframes spin {
+            0% { transform: rotate(0deg); }
+            100% { transform: rotate(360deg); }
+        }
+
+        .owner-section {
+            margin-bottom: 2rem;background: white;
+            border-radius: 18px;
+            padding: 1.5rem;
+            box-shadow: 0 4px 24px rgba(0, 0, 0, 0.1);
+        }
+
+        .owner-name {
+            font-size: 1.3rem;
+            font-weight: 600;
+            color: #1d1d1f;
+            margin-bottom: 1rem;
+            padding-bottom: 0.5rem;
+            border-bottom: 1px solid #e5e5e7;
+            display: flex;
+            justify-content: space-between;align-items: center;
+        }
+
+        .space-count {
+            font-size: 0.9rem;
+            color: #86868b;}
+
+        .space-status-count {
+            margin-left: 10px;
+        }
+
+        .space-grid {
+            display: grid;
+            grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
+            gap: 1.5rem;
+            width: 100%;
+        }
+
+        .space-card {
+            background: white;
+            border-radius: 12px;
+            padding: 1.5rem;
+            box-shadow: 0 2px 10px rgba(0, 0, 0, 0.05);
+            transition: all 0.3s ease;
+            display: flex;
+            flex-direction: column;}
+
+        .space-card:hover {
+            transform: translateY(-3px);
+            box-shadow: 0 4px 20px rgba(0, 0, 0, 0.1);
+        }
+
+        .space-name {
+            font-size: 1.1rem;
+            font-weight: 500;
+            color: #1d1d1f;
+            margin-bottom: 1rem;
+        }
+
+        .space-info {
+            font-size: 0.9rem;
+            color: #86868b;
+            margin-bottom: 1rem;
+            flex-grow: 1;
+        }
+
+        .space-info p {
+            margin-bottom: 0.5rem;
+        }
+
+        .status-badge {
+            display: inline-block;
+            padding: 2px 8px;
+            border-radius: 4px;
+            font-weight: 500;}
+
+        .status-BUILDING {
+            background-color: #ff9500;
+            color: white;
+        }
+
+        .status-RUNNING {
+            background-color: #34c759;
+            color: white;
+        }
+
+        .status-SLEEPING {
+            background-color: #007aff;
+            color: white;
+        }
+
+        .status-STOPPED {
+            background-color: #8e8e93;
+            color: white;
+        }
+
+        .status-FAILED {
+            background-color: #ff3b30;
+            color: white;
+        }
+
+        .status-BUILD_ERROR {
+            background-color: #ff3b30;
+            color: white;
+        }
+
+        .status-UNKNOWN {
+            background-color: #8e8e93;
+            color: white;
+        }
+
+        .action-buttons {
+            display: grid;
+            grid-template-columns: repeat(3, 1fr);
+            gap: 0.5rem;
+            margin-top: auto;
+        }
+
+        .action-button {
+            padding: 8px 12px;
+            border-radius: 8px;
+            border: none;
+            font-size: 0.9rem;
+            cursor: pointer;
+            transition: all 0.3s ease;text-align: center;
+            text-decoration: none;
+        }
+
+        .view {
+            background: #e5e5e7;
+            color: #1d1d1f;
+        }
+
+        .view:hover {
+            background: #d5d5d7;
+        }
+
+        .restart {
+            background: #0071e3;
+            color: white;
+        }
+
+        .restart:hover {
+            background: #0077ED;
+        }
+
+        .rebuild {
+            background: #f5f5f7;
+            color: #1d1d1f;
+        }
+
+        .rebuild:hover {
+            background: #e5e5e7;
+        }
+
+        .footer {
+            text-align: center;
+            padding: 2rem;
+            color: #86868b;
+            font-size: 0.9rem;
+            background: white;
+            border-top: 1px solid #e5e5e7;
+        }
+
+        .footer a {
+            color: #0071e3;
+            text-decoration: none;transition: color 0.3s ease;
+        }
+
+        .footer a:hover {
+            color: #0077ED;
+            text-decoration: underline;
+        }
+
+        @media (max-width: 768px) {
+            .container {
+                padding: 0 1rem;
+            }
+
+            .owner-name {
+                flex-direction: column;
+                align-items: flex-start;
+            }
+
+            .space-count {
+                margin-top: 0.5rem;
+            }
+        }
+    </style>
+</head>
+<body>
+    <div id="loading" class="loading-overlay">
+        <div class="loading-spinner"></div>
+    </div>
+
+    <header class="header">
+        <h1>HF Space Manager</h1>
+        <a href="/logout" class="logout">退出登录</a>
+    </header>
+
+    <div class="container">
+        {% if spaces %}
+            {% set grouped_spaces = {} %}
+            {% for space in spaces %}
+                {% if space.owner not in grouped_spaces %}
+                    {% set _ = grouped_spaces.update({space.owner: []}) %}
+                {% endif %}
+                {% set _ = grouped_spaces[space.owner].append(space) %}
+            {% endfor %}
+
+            {% for owner, owner_spaces in grouped_spaces.items() %}
+                {% set sorted_spaces = owner_spaces %}
+                {% set running_count = sorted_spaces | selectattr('status','equalto','RUNNING') | list | length %}
+                {% set building_count = sorted_spaces | selectattr('status','equalto','BUILDING') | list | length %}
+                {% set sleeping_count = sorted_spaces | selectattr('status','equalto','SLEEPING') | list | length %}
+                {% set stopped_count = sorted_spaces | selectattr('status','equalto','STOPPED') | list | length %}
+                {% set failed_count = sorted_spaces | selectattr('status','equalto','BUILD_ERROR') | list | length %}
+                <div class="owner-section">
+                    <div class="owner-name">
+                        <span>{{ owner }}</span>
+                <span class="space-count">
+                            总数: {{ sorted_spaces | length }}
+                            <span class="space-status-count">运行:{{ running_count }}</span>
+                            <span class="space-status-count">休眠:{{ sleeping_count }}</span>
+                            <span class="space-status-count">停止:{{ stopped_count }}</span>
+                            <span class="space-status-count">失败:{{ failed_count }}</span>
+                        </span>
+                    </div><div class="space-grid">
+                        {% for space in sorted_spaces %}
+                <div class="space-card" data-space-id="{{ space.repo_id }}"><div class="space-name">{{ space.name }}</div>
+                            <div class="space-info">
+                                <p>ID: {{ space.repo_id }}</p>
+                                <p>状态: <span class="status-badge status-{{ space.status }}">{{ space.status }}</span></p><p>创建时间: {{ space.created_at }}</p>
+                                <p>最后修改: {{ space.last_modified }}</p>
+                <p>SDK: {{ space.sdk }}</p><p>App端口: {{ space.app_port }}</p>
+                {% if space.tags %}
+                                <p>标签:{% for tag in space.tags %}<span class="tag">{{ tag }}</span>
+                                    {% endfor %}
+                                </p>
+                                {% endif %}<p>私有: {{ '是' if space.private else '否' }}</p>
+                            </div><div class="action-buttons">
+                                <a href="{{ space.url }}" target="_blank" class="action-button view">查看</a>
+                                <button onclick="confirmAction('restart', '{{ space.repo_id }}')" class="action-button restart">重启</button>
+                                <button onclick="confirmAction('rebuild', '{{ space.repo_id }}')" class="action-button rebuild">重建</button>
+                            </div>
+                        </div>
+                        {% endfor %}
+                    </div>
+                </div>
+            {% endfor %}
+        {% else %}
+            <div class="owner-section">
+                <p style="text-align: center; color: #86868b;">没有找到任何 Spaces。请确保你的账户中有创建的Spaces，并且提供的 token 有正确的权限。</p>
+            </div>
+        {% endif %}
+    </div>
+
+    <footer class="footer">
+        <a href="https://github.com/ssfun/hf-space-manager">HF Space Manager</a> 由 
+        <a href="https://github.com/ssfun">ssfun</a> 构建，源代码遵循
+        <a href="https://opensource.org/license/mit">MIT 协议</a>
+    </footer>
+
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.0.1/socket.io.js"></script>
+    <script>
+        // 连接 WebSocket
+        const socket = io();
+
+        socket.on('connect', () => {
+            console.log('Connected to server');
+        });
+
+        socket.on('disconnect', () => {
+            console.log('Disconnected from server');
+        });
+
+        // 当收到缓存更新通知时刷新数据
+        socket.on('spaces_updated', (data) => {
+            updateSpaceStatuses();
+        });
+
+        // 页面可见性变化处理
+        document.addEventListener('visibilitychange', function() {
+            if (document.hidden) {
+                // 页面隐藏时断开连接
+                socket.disconnect();
+            } else {
+                // 页面可见时重新连接
+                socket.connect();
+            }
+        });
+        
+        // 页面加载完成后隐藏加载动画
+        window.addEventListener('load', function() {
+            document.getElementById('loading').style.display = 'none';
+        });
+
+        // 定期更新状态
+        function updateSpaceStatuses() {document.querySelectorAll('.space-card').forEach(card => {
+                const spaceId = card.dataset.spaceId;
+                fetch(`/api/space/${spaceId}/status`)
+                    .then(response => response.json())
+                    .then(data => {
+                        const statusElement = card.querySelector('.status-badge');
+                        if (statusElement && data.status) {
+                            statusElement.className = `status-badge status-${data.status}`;statusElement.textContent = data.status;
+                        }
+                    })
+                    .catch(error => console.error('Error updating status:', error));
+            });
+        }
+
+        // 在进行操作时显示加载动画
+        function confirmAction(action, spaceId) {
+            var actionText = action === 'restart' ? '重启' : '重建';
+            if (confirm(`确定要${actionText} "${spaceId}" 吗？`)) {document.getElementById('loading').style.display = 'flex';
+                window.location.href = `/action/${action}/${spaceId}`;
+            }
+        }
+
+        // 每30秒更新一次状态
+        setInterval(updateSpaceStatuses, 30000);
+    </script>
+</body>
+</html>

templates/index.html

@@ -0,0 +1,135 @@
+<!DOCTYPE html>
+<html lang="zh">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>登录 - HF Space Manager</title>
+    <style>
+        * {
+            margin: 0;
+            padding: 0;
+            box-sizing: border-box;
+            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
+        }
+
+        body {
+            background-color: #f5f5f7;
+            min-height: 100vh;
+            display: flex;
+            flex-direction: column;
+        }
+
+        .login-container {
+            flex: 1;
+            display: flex;
+            flex-direction: column;
+            justify-content: center;
+            align-items: center;
+            padding: 2rem;
+        }
+
+        .logo {
+            font-size: 2rem;
+            font-weight: 600;
+            color: #1d1d1f;
+            margin-bottom: 2rem;
+        }
+
+        form {
+            background: white;
+            padding: 2rem;
+            border-radius: 18px;
+            box-shadow: 0 4px 24px rgba(0, 0, 0, 0.1);
+            width: 100%;
+            max-width: 400px;}
+
+        .form-group {
+            margin-bottom: 1.5rem;
+        }
+
+        label {
+            display: block;
+            margin-bottom: 0.5rem;
+            color: #1d1d1f;
+            font-weight: 500;
+        }
+
+        input {
+            width: 100%;
+            padding: 0.75rem;
+            border: 1px solid #e5e5e7;
+            border-radius: 8px;
+            font-size: 1rem;
+            transition: all 0.3s ease;
+        }
+
+        input:focus {
+            outline: none;
+            border-color: #0071e3;
+            box-shadow: 0 0 0 3px rgba(0, 113, 227, 0.1);
+        }
+
+        button {
+            width: 100%;
+            padding: 0.75rem;
+            background: #0071e3;
+            color: white;
+            border: none;
+            border-radius: 8px;
+            font-size: 1rem;
+            cursor: pointer;
+            transition: all 0.3s ease;
+        }
+
+        button:hover {
+            background: #0077ED;
+        }
+
+        .error-message {
+            color: #ff3b30;
+            margin-bottom: 1rem;text-align: center;
+        }
+
+        .footer {
+            text-align: center;
+            padding: 2rem;
+            color: #86868b;font-size: 0.9rem;
+            background: white;
+            border-top: 1px solid #e5e5e7;
+        }
+
+        .footer a {
+            color: #0071e3;
+            text-decoration: none;transition: color 0.3s ease;
+        }
+
+        .footer a:hover {
+            color: #0077ED;text-decoration: underline;
+        }
+    </style>
+</head>
+<body>
+    <div class="login-container">
+        <div class="logo">HF Space Manager</div>
+        <form method="post">
+            <div class="form-group">
+                <label for="username">用户名</label>
+                <input type="text" id="username" name="username" required>
+            </div>
+            <div class="form-group">
+                <label for="password">密码</label>
+                <input type="password" id="password" name="password" required>
+            </div>
+            {% if error %}
+            <div class="error-message">{{ error }}</div>
+            {% endif %}
+            <button type="submit">登录</button>
+        </form>
+    </div>
+
+    <footer class="footer">
+        <a href="https://github.com/ssfun/hf-space-manager">HF Space Manager</a> 由<a href="https://github.com/ssfun">ssfun</a> 构建，源代码遵循
+        <a href="https://opensource.org/license/mit">MIT 协议</a>
+    </footer>
+</body>
+</html>