diff --git a/admin/app.py b/admin/app.py new file mode 100644 index 0000000..b5dcc87 --- /dev/null +++ b/admin/app.py @@ -0,0 +1,254 @@ +""" +参数百科网站 - 后台管理系统 +""" + +from flask import Flask, render_template, jsonify, request +from flask_cors import CORS +import json +from pathlib import Path +from datetime import datetime + +app = Flask(__name__) +CORS(app) + +# 数据目录 - 读取 Next.js 项目的数据文件 +DATA_DIR = Path(__file__).parent.parent / 'src' / 'data' + +def load_models(): + """加载模型数据""" + init_file = DATA_DIR / 'initial.ts' + if init_file.exists(): + # 从 TypeScript 文件解析数据 + content = init_file.read_text(encoding='utf-8') + return parse_ts_data(content, 'models') + return [] + +def load_gpus(): + """加载GPU数据""" + init_file = DATA_DIR / 'initial.ts' + if init_file.exists(): + content = init_file.read_text(encoding='utf-8') + return parse_ts_data(content, 'gpus') + return [] + +def load_cpus(): + """加载CPU数据""" + init_file = DATA_DIR / 'initial.ts' + if init_file.exists(): + content = init_file.read_text(encoding='utf-8') + return parse_ts_data(content, 'cpus') + return [] + +def parse_ts_data(content, data_type): + """简单解析 TypeScript 数据""" + import re + + # 找到数据定义的开始位置 + patterns = { + 'models': r'export const models: Model\[\]\s*=\s*\[([\s\S]*?)\n\]', + 'gpus': r'export const gpus: Gpu\[\]\s*=\s*\[([\s\S]*?)\n\]', + 'cpus': r'export const cpus: Cpu\[\]\s*=\s*\[([\s\S]*?)\n\]', + } + + pattern = patterns.get(data_type) + if not pattern: + return [] + + match = re.search(pattern, content) + if not match: + return [] + + data_str = match.group(1) + + # 简单的对象解析 + items = [] + current_item = {} + current_key = None + current_value = [] + in_string = False + in_object = 0 + + lines = data_str.split('\n') + for line in lines: + line = line.strip() + if line == '{': + current_item = {} + in_object += 1 + elif line == '},' or line == '}': + if current_item: + items.append(current_item) + current_item = {} + in_object = max(0, in_object - 1) + elif ':' in line and in_object > 0: + parts = line.split(':', 1) + key = parts[0].strip() + value = parts[1].strip().rstrip(',') + + if value.startswith('"') or value.startswith("'"): + value = value.strip('"\'') + elif value == 'null': + value = None + elif value == 'true': + value = True + elif value == 'false': + value = False + elif value.startswith('['): + # 数组类型 + value = value.strip('[]').replace('"', '').split(',') if value != '[]' else [] + value = [v.strip() for v in value if v.strip()] + else: + try: + value = int(value) if '.' not in value else float(value) + except: + pass + + current_item[key] = value + + return items + +def save_models(models): + """保存模型数据""" + # 这里需要更新 TypeScript 文件 + # 为简化，我们使用 JSON 文件存储变更 + data_dir = Path(__file__).parent.parent / 'data' + data_dir.mkdir(exist_ok=True) + (data_dir / 'models.json').write_text(json.dumps(models, ensure_ascii=False, indent=2), encoding='utf-8') + +def save_gpus(gpus): + """保存GPU数据""" + data_dir = Path(__file__).parent.parent / 'data' + data_dir.mkdir(exist_ok=True) + (data_dir / 'gpus.json').write_text(json.dumps(gpus, ensure_ascii=False, indent=2), encoding='utf-8') + +# ============ 页面路由 ============ + +@app.route('/') +def index(): + return render_template('index.html') + +@app.route('/models') +def models_page(): + return render_template('models.html') + +@app.route('/gpus') +def gpus_page(): + return render_template('gpus.html') + +@app.route('/cpus') +def cpus_page(): + return render_template('cpus.html') + +# ============ API路由 ============ + +@app.route('/api/stats') +def api_stats(): + models = load_models() + gpus = load_gpus() + cpus = load_cpus() + + # 统计 + open_source = sum(1 for m in models if m.get('isOpenSource')) + nvidia_gpus = sum(1 for g in gpus if g.get('manufacturer') == 'NVIDIA') + amd_gpus = sum(1 for g in gpus if g.get('manufacturer') == 'AMD') + + return jsonify({ + 'models_count': len(models), + 'gpus_count': len(gpus), + 'cpus_count': len(cpus), + 'open_source_models': open_source, + 'closed_source_models': len(models) - open_source, + 'nvidia_gpus': nvidia_gpus, + 'amd_gpus': amd_gpus, + }) + +@app.route('/api/models') +def api_models(): + models = load_models() + return jsonify(models) + +@app.route('/api/models', methods=['POST']) +def api_add_model(): + data = request.json + + models = load_models() + + new_model = { + 'name': data.get('name', ''), + 'slug': data.get('slug', data.get('name', '').lower().replace(' ', '-')), + 'organization': data.get('organization', ''), + 'parametersCount': data.get('parametersCount'), + 'architecture': data.get('architecture'), + 'contextLength': data.get('contextLength'), + 'isOpenSource': data.get('isOpenSource', False), + 'license': data.get('license'), + 'benchmarkMmlu': data.get('benchmarkMmlu'), + 'benchmarkHumaneval': data.get('benchmarkHumaneval'), + 'minVramFp16': data.get('minVramFp16'), + 'minVramInt8': data.get('minVramInt8'), + 'minVramInt4': data.get('minVramInt4'), + } + + models.append(new_model) + save_models(models) + + return jsonify({'success': True, 'model': new_model}) + +@app.route('/api/gpus') +def api_gpus(): + gpus = load_gpus() + return jsonify(gpus) + +@app.route('/api/gpus', methods=['POST']) +def api_add_gpu(): + data = request.json + + gpus = load_gpus() + + new_gpu = { + 'name': data.get('name', ''), + 'slug': data.get('slug', data.get('name', '').lower().replace(' ', '-')), + 'manufacturer': data.get('manufacturer', 'NVIDIA'), + 'architecture': data.get('architecture'), + 'cudaCores': data.get('cudaCores'), + 'tensorCores': data.get('tensorCores'), + 'memoryGb': data.get('memoryGb'), + 'memoryType': data.get('memoryType'), + 'memoryBandwidthGbps': data.get('memoryBandwidthGbps'), + 'fp32Tflops': data.get('fp32Tflops'), + 'fp16Tflops': data.get('fp16Tflops'), + 'tdpWatts': data.get('tdpWatts'), + 'priceUsd': data.get('priceUsd'), + 'recommendedFor': data.get('recommendedFor', []), + } + + gpus.append(new_gpu) + save_gpus(gpus) + + return jsonify({'success': True, 'gpu': new_gpu}) + +@app.route('/api/cpus') +def api_cpus(): + cpus = load_cpus() + return jsonify(cpus) + +@app.route('/api/export/models') +def api_export_models(): + """导出模型数据为 JSON""" + models = load_models() + return jsonify(models) + +@app.route('/api/export/gpus') +def api_export_gpus(): + """导出GPU数据为 JSON""" + gpus = load_gpus() + return jsonify(gpus) + +if __name__ == '__main__': + print("=" * 50) + print("参数百科网站 - 后台管理系统") + print("=" * 50) + print(f"访问地址: http://localhost:19006") + print(f"前台地址: http://localhost:3000") + print("=" * 50) + + app.run(host='0.0.0.0', port=19006, debug=True) \ No newline at end of file diff --git a/admin/requirements.txt b/admin/requirements.txt new file mode 100644 index 0000000..2a6bcb3 --- /dev/null +++ b/admin/requirements.txt @@ -0,0 +1,2 @@ +flask>=2.3.0 +flask-cors>=4.0.0 \ No newline at end of file diff --git a/admin/templates/cpus.html b/admin/templates/cpus.html new file mode 100644 index 0000000..bdd58e9 --- /dev/null +++ b/admin/templates/cpus.html @@ -0,0 +1,94 @@ + + + + + + CPU管理 - 参数百科后台 + + + + +
+ + +
+

CPU管理

+ + +
+ + + + + + + + + + + + + + + + +
CPU名称厂商架构核心/线程频率缓存功耗价格
加载中...
+
+
+
+ + + + \ No newline at end of file diff --git a/admin/templates/gpus.html b/admin/templates/gpus.html new file mode 100644 index 0000000..0744699 --- /dev/null +++ b/admin/templates/gpus.html @@ -0,0 +1,252 @@ + + + + + + GPU管理 - 参数百科后台 + + + + +
+ + +
+
+

GPU管理

+ +
+ + +
+
+ + +
+
+ + +
+ + + + + + + + + + + + + + + + + +
GPU名称厂商架构显存CUDA核心FP16算力功耗价格操作
加载中...
+
+
+
+ + + + + + + \ No newline at end of file diff --git a/admin/templates/index.html b/admin/templates/index.html new file mode 100644 index 0000000..f139b3c --- /dev/null +++ b/admin/templates/index.html @@ -0,0 +1,202 @@ + + + + + + 参数百科 - 后台管理 + + + + + +
+ + + + +
+ +
+
+
+
+

模型总数

+

-

+
+
+ +
+
+
+ 开源: 0 + 闭源: 0 +
+
+ +
+
+
+

GPU总数

+

-

+
+
+ +
+
+
+ NVIDIA: 0 + AMD: 0 +
+
+ +
+
+
+

CPU总数

+

-

+
+
+ +
+
+
+ +
+
+
+

数据完整性

+

良好

+
+
+ +
+
+
+
+ + +
+
+

+ + 快速操作 +

+
+ + + 添加新模型 + + + + 添加新GPU + + + + 添加新CPU + +
+
+ +
+

+ + 数据导出 +

+
+ + + 导出模型数据 (JSON) + + + + 导出GPU数据 (JSON) + +
+
+
+ + +
+

+ + 最新模型 +

+
+

加载中...

+
+
+
+
+ + + + \ No newline at end of file diff --git a/admin/templates/models.html b/admin/templates/models.html new file mode 100644 index 0000000..805c776 --- /dev/null +++ b/admin/templates/models.html @@ -0,0 +1,247 @@ + + + + + + 模型管理 - 参数百科后台 + + + + +
+ + +
+
+

模型管理

+ +
+ + +
+
+ + +
+
+ + +
+ + + + + + + + + + + + + + + +
模型名称组织参数量上下文类型MMLU操作
加载中...
+
+
+
+ + + + + + + \ No newline at end of file