Add upload_to_shared: push results to HuggingFriends generated_data/<model>_<variant>

ge_modal_app.py

@@ -438,6 +438,43 @@ def clean_frames(source: str):
     return len(to_del)
 
 
+# ==========================================================================
+# 把结果上传到「共享数据集」的 generated_data/<model>_<variant>/ 下
+# 结构与搭档 wan26_flash 一致：data/<source>/generated_data/<model>_<variant>/<task>/<episode>/1/
+# ⚠️ 目标是 HuggingFriends 共享仓库，secret 里的 HF_TOKEN 需对该 org 有写权限
+# ==========================================================================
+@app.function(
+    image=image,
+    volumes={OUT_DIR: output_vol},
+    secrets=[modal.Secret.from_name("huggingface-secret")],
+    timeout=60 * 60,
+)
+def upload_to_shared(
+    source: str,
+    variant: str,                      # "prefix" 或 "rewrite"
+    model_name: str = "genie",         # 文件夹名 <model_name>_<variant>，对齐搭档 wan26_flash 风格
+    repo_id: str = "HuggingFriends/mllm-as-embodied-world-judge",
+):
+    import os
+    from huggingface_hub import HfApi
+    api = HfApi(token=os.environ["HF_TOKEN"])
+    # 我们 volume 里的目录名是 prompt_prefix / prompt_rewrite
+    src_dir = os.path.join(OUT_DIR, source, f"prompt_{variant}")
+    if not os.path.isdir(src_dir):
+        print(f"⚠️ {src_dir} 不存在，跳过（该 source/variant 还没生成？）")
+        return
+    path_in_repo = f"data/{source}/generated_data/{model_name}_{variant}"
+    api.upload_folder(
+        folder_path=src_dir,
+        path_in_repo=path_in_repo,
+        repo_id=repo_id,
+        repo_type="dataset",
+        ignore_patterns=["**/video/**"],     # 只传 mp4 + prompt，不传抽帧
+        commit_message=f"Add {model_name} {variant} results: {source}",
+    )
+    print(f"✅ {source}/{variant} -> {repo_id}/{path_in_repo}")
+
+
 # ==========================================================================
 # 本地编排入口：modal run ge_modal_app.py
 # ==========================================================================