Upload app.py with huggingface_hub

app.py

@@ -98,9 +98,11 @@ def check_status(task_id):
     try:
         files = list_repo_files(REPO_ID, repo_type=REPO_TYPE, token=HF_TOKEN)
 
-        # 检查 outbox 是否有结果: outbox/{task_id}.glb
-        result_path = f"outbox/{task_id}.glb"
-        if result_path in files:
+        # 检查 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 (
@@ -108,7 +110,7 @@ def check_status(task_id):
                 f"| 项目 | 信息 |\n"
                 f"|------|------|\n"
                 f"| 📋 任务ID | `{task_id}` |\n"
-                f"| 📄 格式 | GLB |\n\n"
+                f"| 📄 格式 | {ext} |\n\n"
                 f"> 👇 [点击此处下载优化后的模型]({download_url})"
             )
 
@@ -212,6 +214,9 @@ with gr.Blocks(
         <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():