app_hf.py

"""
3D 模型优化服务 - HF Space 版本
Gradio + HF Hub API | 含用户建议反馈
"""
import os
import json
import uuid
from datetime import datetime

import gradio as gr
from huggingface_hub import HfApi, list_repo_files

from feedback_util import fetch_public_feedback, submit_feedback

# === 配置 ===
REPO_ID = "wangyiyi666/model-optimizer-queue"
REPO_TYPE = "dataset"
HF_TOKEN = os.environ.get("HF_TOKEN", "")
SUPPORTED_FORMATS = [".glb", ".gltf", ".fbx", ".obj"]

api = HfApi(token=HF_TOKEN)


def log(msg):
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    print(f"[{timestamp}] [SPACE] {msg}")


def upload_model(file, progress=gr.Progress()):
    """处理用户上传的模型文件"""
    if file is None:
        return "### ⚠️ 请先选择文件", ""

    filename = os.path.basename(file.name if hasattr(file, "name") else file)
    ext = os.path.splitext(filename)[1].lower()
    log(f"用户上传: {filename}")

    if ext not in SUPPORTED_FORMATS:
        log(f"格式不支持: {ext}")
        return f"### ❌ 不支持的格式: `{ext}`\n\n支持的格式: {', '.join(SUPPORTED_FORMATS)}", ""

    task_id = uuid.uuid4().hex
    target_name = f"{task_id}{ext}"

    try:
        progress(0.3, desc="正在上传模型文件...")
        file_path = file.name if hasattr(file, "name") else file
        api.upload_file(
            path_or_fileobj=file_path,
            path_in_repo=f"inbox/{target_name}",
            repo_id=REPO_ID,
            repo_type=REPO_TYPE,
        )

        progress(0.7, desc="正在创建任务...")
        name_no_ext = os.path.splitext(filename)[0]
        meta = json.dumps({
            "task_id": task_id,
            "filename": filename,
            "name_no_ext": name_no_ext,
            "status": "pending",
            "created": str(datetime.now()),
        })
        api.upload_file(
            path_or_fileobj=meta.encode(),
            path_in_repo=f"inbox/{task_id}.json",
            repo_id=REPO_ID,
            repo_type=REPO_TYPE,
        )

        progress(1.0, desc="上传完成!")
        log(f"上传成功: {target_name}, 任务ID: {task_id}")

        file_size = os.path.getsize(file_path)
        size_str = f"{file_size / 1024:.1f} KB" if file_size < 1024 * 1024 else f"{file_size / 1024 / 1024:.1f} MB"

        return (
            f"### ✅ **上传成功！**\n\n"
            f"| 项目 | 信息 |\n"
            f"|------|------|\n"
            f"| 📋 任务ID | `{task_id}` |\n"
            f"| 📁 文件名 | {filename} |\n"
            f"| 📦 文件大小 | {size_str} |\n"
            f"| ⏱️ 状态 | 等待优化处理 |\n\n"
            f"> ⚠️ **务必保存好任务ID，这是您下载优化结果的唯一凭证！**"
        ), task_id

    except Exception as e:
        log(f"上传失败: {e}")
        return f"### ❌ 上传失败\n\n```\n{str(e)}\n```", ""


def check_status(task_id):
    """查询任务状态"""
    if not task_id or len(task_id.strip()) == 0:
        return "### ⚠️ 请输入任务ID", gr.update(visible=False)

    task_id = task_id.strip()
    log(f"查询: {task_id}")

    try:
        files = list_repo_files(REPO_ID, repo_type=REPO_TYPE, token=HF_TOKEN)
        outbox_matches = [f for f in files if f.startswith(f"outbox/{task_id}") and not f.endswith(".json")]
        if outbox_matches:
            result_path = outbox_matches[0]
            ext = os.path.splitext(result_path)[1].upper().lstrip(".")
            log(f"任务 {task_id} 已完成")
            return (
                f"### ✅ 优化完成！\n\n"
                f"| 项目 | 信息 |\n"
                f"|------|------|\n"
                f"| 📋 任务ID | `{task_id}` |\n"
                f"| 📄 格式 | {ext} |\n\n"
                f"> 👇 点击下方「下载模型」按钮获取优化后的文件"
            ), gr.update(visible=True)

        inbox_files = [f for f in files if f.startswith(f"inbox/{task_id}")]
        if inbox_files:
            log(f"任务 {task_id} 仍在队列中")
            return (
                f"### ⏳ 处理中\n\n"
                f"任务 `{task_id}` 正在优化队列中，请稍后再查询。\n\n"
                f"> Worker 每 **30秒** 检查一次新任务，优化完成后即可下载。"
            ), gr.update(visible=False)

        log(f"任务 {task_id} 未找到")
        return f"### ❌ 未找到任务\n\n任务ID `{task_id}` 不存在，请检查是否输入正确。", gr.update(visible=False)

    except Exception as e:
        log(f"查询出错: {e}")
        return f"### ❌ 查询出错\n\n```\n{str(e)}\n```", gr.update(visible=False)


def download_model(task_id):
    """下载模型并记录取件事件"""
    if not task_id or len(task_id.strip()) == 0:
        return None, "### ⚠️ 请先查询任务ID"

    task_id = task_id.strip()
    log(f"用户下载: {task_id}")

    try:
        files = list_repo_files(REPO_ID, repo_type=REPO_TYPE, token=HF_TOKEN)
        outbox_matches = [f for f in files if f.startswith(f"outbox/{task_id}") and not f.endswith(".json")]
        if not outbox_matches:
            return None, "### ❌ 未找到优化结果"

        result_path = outbox_matches[0]
        download_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/{result_path}"

        # 记录下载事件到 HF Dataset
        from datetime import datetime
        record = json.dumps({
            "task_id": task_id,
            "downloaded_at": datetime.now().isoformat(),
            "result_file": result_path,
        })
        try:
            api.upload_file(
                path_or_fileobj=record.encode(),
                path_in_repo=f"outbox/{task_id}.downloaded.json",
                repo_id=REPO_ID,
                repo_type=REPO_TYPE,
            )
            log(f"已记录下载: {task_id}")
        except Exception as e:
            log(f"记录下载失败(不影响下载): {e}")

        return (
            None,
            f"### ✅ 下载链接已生成\n\n"
            f"> 👇 [点击此处下载优化后的模型]({download_url})\n\n"
            f"*下载记录已通知管理端*"
        )

    except Exception as e:
        log(f"下载出错: {e}")
        return None, f"### ❌ 下载出错\n\n```\n{str(e)}\n```"


def delete_task(task_id):
    """删除任务（清理 inbox + outbox 文件）"""
    if not task_id or len(task_id.strip()) == 0:
        return "### ⚠️ 请输入任务ID"

    task_id = task_id.strip()
    log(f"删除任务: {task_id}")

    try:
        files = list_repo_files(REPO_ID, repo_type=REPO_TYPE, token=HF_TOKEN)
        to_delete = [f for f in files if task_id in f and f != ".gitattributes"]

        if not to_delete:
            return f"### ❌ 未找到任务\n\n任务ID `{task_id}` 不存在。"

        deleted = []
        for filepath in to_delete:
            try:
                api.delete_file(filepath, REPO_ID, repo_type=REPO_TYPE)
                deleted.append(filepath)
                log(f"已删除: {filepath}")
            except Exception as e:
                log(f"删除失败 {filepath}: {e}")

        file_list = "\n".join([f"- `{f}`" for f in deleted])
        return f"### 🗑️ 删除成功\n\n已删除 **{len(deleted)}** 个文件：\n\n{file_list}"

    except Exception as e:
        log(f"删除出错: {e}")
        return f"### ❌ 删除出错\n\n```\n{str(e)}\n```"


def render_public_feedback():
    items = fetch_public_feedback()
    if not items:
        return "<p style='color:#888;'>暂无公开反馈</p>"

    blocks = []
    for item in items:
        created = (item.get("created") or "")[:19]
        username = item.get("username") or "匿名"
        text = (item.get("text") or "").replace("\n", "<br>")
        img_note = ""
        if item.get("images"):
            img_note = f"<div style='color:#666;font-size:13px;margin-top:6px;'>附带 {len(item['images'])} 张图片</div>"
        blocks.append(
            f"<div style='border:1px solid #eee;border-radius:8px;padding:14px;margin-bottom:12px;background:#fafafa;'>"
            f"<div style='font-weight:600;'>{username} "
            f"<span style='color:#999;font-size:12px;'>{created}</span></div>"
            f"<div style='margin-top:8px;line-height:1.6;'>{text}</div>{img_note}</div>"
        )
    return "".join(blocks)


def handle_feedback_submit(username, text, images, is_public):
    try:
        image_paths = []
        if images:
            if isinstance(images, list):
                image_paths = [item.name if hasattr(item, "name") else item for item in images]
            else:
                image_paths = [images.name if hasattr(images, "name") else images]
        meta = submit_feedback(username, text, image_paths, is_public)
        visibility = "已公开" if meta.get("is_public") else "仅管理员可见"
        return f"### ✅ 反馈提交成功\n\n| 项目 | 信息 |\n|------|------|\n| 反馈ID | `{meta['feedback_id']}` |\n| 展示范围 | {visibility} |"
    except Exception as e:
        log(f"反馈提交失败: {e}")
        return f"### ❌ 提交失败\n\n```\n{str(e)}\n```"


custom_css = """
.main-title { text-align: center; margin-bottom: 0.5em; }
.sub-title { text-align: center; color: #666; font-size: 1.1em; margin-bottom: 1.5em; }
.format-badge {
    display: inline-block; background: #e3f2fd; color: #1565c0;
    padding: 4px 12px; border-radius: 16px; margin: 2px; font-size: 0.9em; font-weight: 500;
}
footer { display: none !important; }
"""

with gr.Blocks(
    title="3D 模型优化服务",
    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="slate"),
    css=custom_css,
) as demo:
    gr.HTML("""
    <div class="main-title"><h1>🛠️ 3D 模型优化服务</h1></div>
    <div class="sub-title">上传 3D 模型，自动优化处理，完成后下载</div>
    <div style="text-align:center;margin-bottom:1.5em;">
        <span class="format-badge">GLB</span><span class="format-badge">GLTF</span>
        <span class="format-badge">FBX</span><span class="format-badge">OBJ</span>
        <span style="margin:0 8px;">→</span><span class="format-badge">GLB 输出</span>
    </div>
    <p style="text-align:center;color:#e65100;">⚠️ 测试阶段，请优先上传 GLB 格式文件</p>
    """)

    with gr.Tabs():
        with gr.Tab("📤 上传模型", id="upload"):
            with gr.Row(equal_height=True):
                with gr.Column(scale=1):
                    file_input = gr.File(
                        label="选择 3D 模型文件",
                        file_types=[".glb", ".gltf", ".fbx", ".obj"],
                        type="filepath",
                        height=200,
                    )
                    upload_btn = gr.Button("🚀 提交优化", variant="primary", size="lg")
                with gr.Column(scale=1):
                    upload_result = gr.Markdown(
                        value="### 📋 等待上传\n\n选择文件后点击「提交优化」按钮。",
                        label="处理结果",
                    )
                    task_id_output = gr.Textbox(label="📋 任务ID（复制保存）", interactive=False)

            upload_btn.click(upload_model, inputs=[file_input], outputs=[upload_result, task_id_output])

        with gr.Tab("🔍 查询结果", id="query"):
            with gr.Row(equal_height=True):
                with gr.Column(scale=1):
                    task_id_input = gr.Textbox(label="输入任务ID", placeholder="粘贴完整任务ID", max_lines=1)
                    with gr.Row():
                        check_btn = gr.Button("🔍 查询状态", variant="primary", size="lg")
                        delete_btn = gr.Button("🗑️ 删除任务", variant="stop", size="lg")
                    download_btn = gr.Button("📥 下载模型", variant="secondary", size="lg", visible=False)
                with gr.Column(scale=1):
                    status_result = gr.Markdown(
                        value="### 📋 等待查询\n\n输入任务ID后点击「查询状态」按钮。",
                        label="任务状态",
                    )

            check_btn.click(check_status, inputs=[task_id_input], outputs=[status_result, download_btn])
            delete_btn.click(delete_task, inputs=[task_id_input], outputs=[status_result])
            download_btn.click(download_model, inputs=[task_id_input], outputs=[download_btn, status_result])

        with gr.Tab("💬 提交反馈", id="feedback"):
            gr.Markdown("欢迎提交使用建议或问题反馈，可附带截图。管理员会在本地管理面板查看全部反馈。")
            feedback_username = gr.Textbox(label="用户名 / 昵称", placeholder="可选，默认匿名")
            feedback_text = gr.Textbox(label="反馈内容", lines=6, placeholder="请描述您的建议或遇到的问题...")
            feedback_images = gr.File(
                label="截图 / 图片（可选，可多选）",
                file_count="multiple",
                file_types=["image"],
                type="filepath",
            )
            feedback_public = gr.Checkbox(
                label="允许对外公开展示（勾选后其他用户可在「公开反馈」页看到文字内容）",
                value=False,
            )
            feedback_submit_btn = gr.Button("提交反馈", variant="primary")
            feedback_result = gr.Markdown()
            feedback_submit_btn.click(
                handle_feedback_submit,
                inputs=[feedback_username, feedback_text, feedback_images, feedback_public],
                outputs=[feedback_result],
            )

        with gr.Tab("📣 公开反馈", id="public_feedback"):
            refresh_public_btn = gr.Button("🔄 刷新公开反馈")
            public_feedback_html = gr.HTML(value=render_public_feedback())
            refresh_public_btn.click(lambda: render_public_feedback(), outputs=[public_feedback_html])

    gr.HTML("<p style='text-align:center;color:#999;margin-top:1em;'>Powered by Blender · Worker 每30秒检查新任务</p>")


if __name__ == "__main__":
    log("Gradio 前端启动")
    demo.launch()

"""
3D 模型优化服务 - HF Space 版本
Gradio + HF Hub API | 美观 UI
"""
import gradio as gr
from huggingface_hub import HfApi, list_repo_files
import os
import json
import uuid
from datetime import datetime

# === 配置 ===
REPO_ID = "wangyiyi666/model-optimizer-queue"
REPO_TYPE = "dataset"
HF_TOKEN = os.environ.get("HF_TOKEN", "")
SUPPORTED_FORMATS = [".glb", ".gltf", ".fbx", ".obj"]

api = HfApi(token=HF_TOKEN)


def log(msg):
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    print(f"[{timestamp}] [SPACE] {msg}")


def upload_model(file, progress=gr.Progress()):
    """处理用户上传的模型文件"""
    if file is None:
        return "### ⚠️ 请先选择文件", ""

    filename = os.path.basename(file.name if hasattr(file, 'name') else file)
    ext = os.path.splitext(filename)[1].lower()
    log(f"用户上传: {filename}")

    if ext not in SUPPORTED_FORMATS:
        log(f"格式不支持: {ext}")
        return f"### ❌ 不支持的格式: `{ext}`\n\n支持的格式: {', '.join(SUPPORTED_FORMATS)}", ""

    task_id = uuid.uuid4().hex
    target_name = f"{task_id}{ext}"

    try:
        progress(0.3, desc="正在上传模型文件...")
        file_path = file.name if hasattr(file, 'name') else file
        api.upload_file(
            path_or_fileobj=file_path,
            path_in_repo=f"inbox/{target_name}",
            repo_id=REPO_ID,
            repo_type=REPO_TYPE
        )

        progress(0.7, desc="正在创建任务...")
        name_no_ext = os.path.splitext(filename)[0]
        meta = json.dumps({
            "task_id": task_id,
            "filename": filename,
            "name_no_ext": name_no_ext,
            "status": "pending",
            "created": str(datetime.now())
        })
        api.upload_file(
            path_or_fileobj=meta.encode(),
            path_in_repo=f"inbox/{task_id}.json",
            repo_id=REPO_ID,
            repo_type=REPO_TYPE
        )

        progress(1.0, desc="上传完成!")
        log(f"上传成功: {target_name}, 任务ID: {task_id}")

        file_size = os.path.getsize(file_path)
        size_str = f"{file_size / 1024:.1f} KB" if file_size < 1024 * 1024 else f"{file_size / 1024 / 1024:.1f} MB"

        return (
            f"### ✅ **上传成功！**\n\n"
            f"| 项目 | 信息 |\n"
            f"|------|------|\n"
            f"| 📋 任务ID | `{task_id}` |\n"
            f"| 📁 文件名 | {filename} |\n"
            f"| 📦 文件大小 | {size_str} |\n"
            f"| ⏱️ 状态 | 等待优化处理 |\n\n"
            f"> ⚠️ **务必保存好任务ID，这是您下载优化结果的唯一凭证！**"
        ), task_id

    except Exception as e:
        log(f"上传失败: {e}")
        return f"### ❌ 上传失败\n\n```\n{str(e)}\n```", ""


def check_status(task_id):
    """查询任务状态"""
    if not task_id or len(task_id.strip()) == 0:
        return "### ⚠️ 请输入任务ID"

    task_id = task_id.strip()
    log(f"查询: {task_id}")

    try:
        files = list_repo_files(REPO_ID, repo_type=REPO_TYPE, token=HF_TOKEN)

        # 检查 outbox 是否有结果: outbox/{task_id}.xxx（保持原始格式）
        outbox_matches = [f for f in files if f.startswith(f"outbox/{task_id}") and not f.endswith('.json')]
        if outbox_matches:
            result_path = outbox_matches[0]
            ext = os.path.splitext(result_path)[1].upper().lstrip('.')
            download_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/{result_path}"
            log(f"任务 {task_id} 已完成")
            return (
                f"### ✅ 优化完成！\n\n"
                f"| 项目 | 信息 |\n"
                f"|------|------|\n"
                f"| 📋 任务ID | `{task_id}` |\n"
                f"| 📄 格式 | {ext} |\n\n"
                f"> 👇 [点击此处下载优化后的模型]({download_url})"
            )

        # 检查 inbox 是否还在排队
        inbox_files = [f for f in files if f.startswith(f"inbox/{task_id}")]
        if inbox_files:
            log(f"任务 {task_id} 仍在队列中")
            return (
                f"### ⏳ 处理中\n\n"
                f"任务 `{task_id}` 正在优化队列中，请稍后再查询。\n\n"
                f"> Worker 每 **30秒** 检查一次新任务，优化完成后即可下载。"
            )

        log(f"任务 {task_id} 未找到")
        return (
            f"### ❌ 未找到任务\n\n"
            f"任务ID `{task_id}` 不存在，请检查是否输入正确。"
        )

    except Exception as e:
        log(f"查询出错: {e}")
        return f"### ❌ 查询出错\n\n```\n{str(e)}\n```"


def delete_task(task_id):
    """删除任务（清理 inbox + outbox 文件）"""
    if not task_id or len(task_id.strip()) == 0:
        return "### ⚠️ 请输入任务ID"

    task_id = task_id.strip()
    log(f"删除任务: {task_id}")

    try:
        files = list_repo_files(REPO_ID, repo_type=REPO_TYPE, token=HF_TOKEN)
        to_delete = [f for f in files if task_id in f and f != '.gitattributes']

        if not to_delete:
            return f"### ❌ 未找到任务\n\n任务ID `{task_id}` 不存在。"

        deleted = []
        for filepath in to_delete:
            try:
                api.delete_file(filepath, REPO_ID, repo_type=REPO_TYPE)
                deleted.append(filepath)
                log(f"已删除: {filepath}")
            except Exception as e:
                log(f"删除失败 {filepath}: {e}")

        file_list = "\n".join([f"- `{f}`" for f in deleted])
        return f"### 🗑️ 删除成功\n\n已删除 **{len(deleted)}** 个文件：\n\n{file_list}"

    except Exception as e:
        log(f"删除出错: {e}")
        return f"### ❌ 删除出错\n\n```\n{str(e)}\n```"


# === 自定义 CSS ===
custom_css = """
.main-title {
    text-align: center;
    margin-bottom: 0.5em;
}
.sub-title {
    text-align: center;
    color: #666;
    font-size: 1.1em;
    margin-bottom: 1.5em;
}
.format-badge {
    display: inline-block;
    background: #e3f2fd;
    color: #1565c0;
    padding: 4px 12px;
    border-radius: 16px;
    margin: 2px;
    font-size: 0.9em;
    font-weight: 500;
}
footer { display: none !important; }
"""

# === Gradio 界面 ===
with gr.Blocks(
    title="3D 模型优化服务",
    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="slate"),
    css=custom_css
) as demo:

    gr.HTML("""
    <div class="main-title">
        <h1>🛠️ 3D 模型优化服务</h1>
    </div>
    <div class="sub-title">
        上传 3D 模型，自动优化处理，完成后下载
    </div>
    <div style="text-align: center; margin-bottom: 20px;">
        <span class="format-badge">GLB</span>
        <span class="format-badge">GLTF</span>
        <span class="format-badge">FBX</span>
        <span class="format-badge">OBJ</span>
        <span style="margin: 0 8px; color: #999;">→</span>
        <span class="format-badge" style="background: #e8f5e9; color: #2e7d32;">GLB 输出</span>
    </div>
    <div style="text-align: center; margin-bottom: 16px; padding: 8px 16px; background: #fff3cd; border-radius: 8px; color: #856404; font-size: 14px;">
        ⚠️ 测试阶段，请优先上传 GLB 格式文件
    </div>
    """)

    with gr.Tabs():
        with gr.Tab("📤 上传模型", id="upload"):
            with gr.Row(equal_height=True):
                with gr.Column(scale=1):
                    file_input = gr.File(
                        label="选择 3D 模型文件",
                        file_types=[".glb", ".gltf", ".fbx", ".obj"],
                        type="filepath",
                        height=200,
                    )
                    upload_btn = gr.Button(
                        "🚀 提交优化",
                        variant="primary",
                        size="lg",
                    )
                with gr.Column(scale=1):
                    upload_result = gr.Markdown(
                        value="### 📋 等待上传\n\n选择文件后点击「提交优化」按钮。",
                        label="处理结果",
                    )
                    task_id_output = gr.Textbox(
                        label="📋 任务ID（复制保存）",
                        interactive=False,
                    )

            upload_btn.click(
                upload_model,
                inputs=[file_input],
                outputs=[upload_result, task_id_output]
            )

        with gr.Tab("🔍 查询结果", id="query"):
            with gr.Row(equal_height=True):
                with gr.Column(scale=1):
                    task_id_input = gr.Textbox(
                        label="输入任务ID",
                        placeholder="粘贴完整任务ID",
                        max_lines=1,
                    )
                    with gr.Row():
                        check_btn = gr.Button(
                            "🔍 查询状态",
                            variant="primary",
                            size="lg",
                        )
                        delete_btn = gr.Button(
                            "🗑️ 删除任务",
                            variant="stop",
                            size="lg",
                        )
                with gr.Column(scale=1):
                    status_result = gr.Markdown(
                        value="### 📋 等待查询\n\n输入任务ID后点击「查询状态」按钮。",
                        label="任务状态",
                    )

            check_btn.click(
                check_status,
                inputs=[task_id_input],
                outputs=[status_result]
            )
            delete_btn.click(
                delete_task,
                inputs=[task_id_input],
                outputs=[status_result]
            )

    gr.HTML("""
    <div style="text-align: center; margin-top: 20px; padding: 15px; border-top: 1px solid #eee; color: #999; font-size: 0.85em;">
        Powered by Blender · Worker 每30秒检查新任务
    </div>
    """)

if __name__ == "__main__":
    log("Gradio 前端启动")
    demo.launch()