Upload 5 files

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 7860
+
+# 设置环境变量（可选，如果需要传递 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:7860", "app:app"]

README-example.md

@@ -0,0 +1,152 @@
+# HF Space Manager
+
+HF Space Manager 是一个用于管理 Hugging Face Spaces 的 Web 应用程序。它提供了直观的界面和API 接口，让你能够方便地查看和管理多个账号下的所有 Spaces。
+
+## 最近更新
+
+2024-12-20 space 信息修改为访问 dashboard 时才会后台自动更新
+
+## 功能特点
+
+- 🚀 支持多个 HuggingFace 账号管理
+- 📊 实时显示 Spaces 状态
+- 🔄 支持重启和重建操作
+- 🔑 安全的登录认证
+- 🎯 简洁的用户界面
+- 📱 响应式设计，支持移动端
+- 🔌 RESTful API 支持
+- 💾 数据缓存机制
+
+## 界面预览
+![index](/preview/index.png "index")
+
+![dashboard](/preview/dashboard.png "dashboard")
+
+## 快速开始
+
+### 环境要求
+
+- Python 3.9+
+- Docker (可选)
+
+### 安装
+
+1. 克隆仓库
+```bash
+git clone https://github.com/ssfun/hf-space-manager.git
+cd hf-space-manager
+```
+
+2. 安装依赖
+```bash
+pip install Flask python-dotenv huggingface_hub requests gunicorn
+```
+
+3. 配置环境变量
+创建 `.env` 文件：
+```env
+USERNAME=your_admin_username
+PASSWORD=your_admin_password
+HF_TOKENS=token1,token2,token3
+API_KEY=your_api_key
+```
+
+### 运行
+
+**直接运行：**
+```bash
+python app.py
+```
+
+**使用 Docker：**
+```bash
+docker run -d \
+    --name hfspace-manager \
+    -p 7860:7860 \
+    -e USERNAME=your_username \
+    -e PASSWORD=your_password \
+    -e HF_TOKENS=token1,token2,token3 \
+    -e API_KEY=your_api_key \
+    sfun/hfspace:latest
+```
+
+## API 文档
+
+### 认证
+所有 API 请求都需要在 header 中包含 API key：
+```
+Authorization: Bearer your_api_key
+```
+
+### 端点
+
+#### 1. 获取所有 Spaces
+```bash
+curl -X GET "http://localhost:7860/api/v1/info/hf_token_here" \
+-H "Content-Type: application/json" \
+-H "Authorization: Bearer API_KEY"
+```
+
+#### 2. 获取特定 Space 信息
+```bash
+curl -X GET "http://localhost:7860/api/v1/info/hf_token_here/username/space-name" \
+-H "Content-Type: application/json" \
+-H "Authorization: Bearer API_KEY"
+```
+
+#### 3. 重启 Space
+```bash
+curl -X POST "http://localhost:7860/api/v1/action/hf_token_here/username/space-name/restart" \
+-H "Content-Type: application/json" \
+-H "Authorization: Bearer API_KEY"
+```
+
+#### 4. 重建 Space
+```bash
+curl -X POST "http://localhost:7860/api/v1/action/hf_token_here/username/space-name/restart" \
+-H "Content-Type: application/json" \
+-H "Authorization: Bearer API_KEY"
+```
+
+## 配置说明
+
+### 环境变量
+
+| 变量名 | 说明 | 必填 |
+|--------|------|------|
+| USERNAME | 管理员用户名 | 是 |
+| PASSWORD | 管理员密码 | 是 |
+| HF_TOKENS | HuggingFace tokens（逗号分隔） | 是 |
+| API_KEY | API访问密钥 | 是 |
+
+### 缓存配置
+
+- 访问 dashboard 时后台自动更新
+
+## 开发
+
+### 本地开发
+```bash
+# 安装开发依赖
+pip install Flask python-dotenv huggingface_hub requests gunicorn
+
+# 运行开发服务器
+python app.py
+```
+
+## 贡献
+
+欢迎 Star 和 Fork 后自行修改！
+
+## 许可证
+
+本项目采用 [MIT 许可证](https://opensource.org/license/mit)。
+
+## 作者
+
+[ssfun](https://github.com/ssfun)
+
+## 致谢
+
+- [Hugging Face](https://huggingface.co/) - 提供优秀的 API 和服务
+- [Flask](https://flask.palletsprojects.com/) - Web 框架支持

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=7860)

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")