#!/usr/bin/env python3
# fix_km_folder_view.py - 将知识库管理改为文件夹视图
import re
HTML = r"D:\AI-Ad-Agent\portal\portal.html"
with open(HTML, 'r', encoding='utf-8') as f:
content = f.read()
# 1. 替换文件列表区域为文件夹视图
old_list = ''' '''
new_list = '''
'''
if old_list in content:
content = content.replace(old_list, new_list)
print("[OK] 文件夹视图HTML已替换")
else:
print("[WARN] 未找到文件列表区域，尝试手动查找...")
# 备用：查找特征字符串
if 'km-filter-category' in content:
print("[INFO] 找到 km-filter-category，但格式不匹配，跳过HTML替换")
print("[INFO] 将只更新JS函数")
# 2. 在 前添加文件夹视图JS函数
js_functions = '''
// ===== 知识库文件夹视图 =====
// 文件夹定义
const KM_FOLDERS = [
{ id: 'material', name: '素材', icon: '🖼️', desc: '图片、视频等广告素材', color: '#6366f1' },
{ id: 'case', name: '案例', icon: '📊', desc: '优质广告案例分析', color: '#f59e0b' },
{ id: 'knowledge', name: '知识点', icon: '📚', desc: '文案、流程等知识内容', color: '#10b981' },
];
let _kmCurrentCategory = null; // 当前打开的文件夹分类
let _kmAllFiles = []; // 缓存当前文件列表
// 显示文件夹视图
async function showKFolderView() {
document.getElementById('km-folder-view').style.display = '';
document.getElementById('km-file-view').style.display = 'none';
_kmCurrentCategory = null;
_kmAllFiles = [];
await loadKFolderView();
}
// 加载文件夹视图（显示3个文件夹+文件数量）
async function loadKFolderView() {
const container = document.getElementById('km-folder-list');
container.innerHTML = '加载中...
';
try {
const resp = await fetch(`${KM_API}/list`, {
headers: { 'Authorization': `Bearer ${authToken}` }
});
const data = await resp.json();
const files = data.files || [];
// 统计每个分类的文件数量
const counts = {};
KM_FOLDERS.forEach(f => counts[f.id] = 0);
files.forEach(f => {
if (counts[f.category] !== undefined) counts[f.category]++;
});
// 渲染文件夹卡片
container.innerHTML = KM_FOLDERS.map(f => {
const count = counts[f.id];
return `
${f.icon}
${f.name}
${f.desc}
${count} 个文件
`;
}).join('');
} catch (e) {
container.innerHTML = `加载失败: ${e.message}
`;
}
}
// 打开文件夹（显示该分类下的所有文件）
async function openKFolder(category) {
const folder = KM_FOLDERS.find(f => f.id === category);
if (!folder) return;
_kmCurrentCategory = category;
document.getElementById('km-folder-view').style.display = 'none';
document.getElementById('km-file-view').style.display = '';
document.getElementById('km-current-folder-title').textContent = `${folder.icon} ${folder.name}`;
await loadKFilesByCategory(category);
}
// 按分类加载文件
async function loadKFilesByCategory(category) {
const container = document.getElementById('km-file-list');
container.innerHTML = '加载中...
';
try {
const resp = await fetch(`${KM_API}/list?category=${category}`, {
headers: { 'Authorization': `Bearer ${authToken}` }
});
const data = await resp.json();
_kmAllFiles = data.files || [];
renderKFiles(_kmAllFiles);
} catch (e) {
container.innerHTML = `加载失败: ${e.message}
`;
}
}
// 搜索/过滤当前文件列表
function filterKFiles() {
const keyword = (document.getElementById('km-search-input').value || '').toLowerCase();
if (!_kmAllFiles.length) return;
const filtered = keyword
? _kmAllFiles.filter(f => (f.filename || '').toLowerCase().includes(keyword))
: _kmAllFiles;
renderKFiles(filtered);
}
// 重写 loadKFiles（供上传后刷新使用）
async function loadKFiles() {
if (_kmCurrentCategory) {
await loadKFilesByCategory(_kmCurrentCategory);
} else {
await loadKFolderView();
}
}
'''
# 在 renderKFiles 函数前插入新JS
marker = '// 渲染知识库文件列表\nfunction renderKFiles(files) {'
if marker in content:
content = content.replace(marker, js_functions + '\n' + marker)
print("[OK] 文件夹视图JS函数已添加")
else:
print("[WARN] 未找到 renderKFiles 标记，JS函数可能未添加")
# 3. 修改 switchPanel 中 knowledge_management 的初始化调用
# 将 loadKFiles() 改为 showKFolderView()
old_init = "if (panel === 'knowledge_management') { loadKFiles(); }"
new_init = "if (panel === 'knowledge_management') { showKFolderView(); }"
if old_init in content:
content = content.replace(old_init, new_init)
print("[OK] switchPanel 初始化已更新为 showKFolderView()")
else:
print("[INFO] switchPanel 中未找到 knowledge_management 初始化代码（可能已更新）")
with open(HTML, 'w', encoding='utf-8') as f:
f.write(content)
print("\n✅ 知识库文件夹视图修改完成！")
print(" 刷新浏览器后生效（Ctrl+F5）")