add api_key args

app.py

@@ -6,25 +6,26 @@ ROOT = Path(__file__).resolve().parent
 OUTPUT_DIR = ROOT / "output"
 ZIP_PATH = ROOT / "output.zip"
 
-def run_pipeline(model_name_t, model_name_v, result_dir, paper_latex_root, arxiv_url):
+def run_pipeline(model_name_t, model_name_v, result_dir, paper_latex_root, arxiv_url, openai_key, gemini_key):
     # 清理旧输出
     if OUTPUT_DIR.exists():
         shutil.rmtree(OUTPUT_DIR)
     if ZIP_PATH.exists():
         ZIP_PATH.unlink()
 
-    # 构造命令
+    # 构造命令（传递两个 key）
     cmd = [
         "python", "pipeline.py",
         "--model_name_t", model_name_t,
         "--model_name_v", model_name_v,
         "--result_dir", result_dir,
         "--paper_latex_root", paper_latex_root,
-        "--arxiv_url", arxiv_url
+        "--arxiv_url", arxiv_url,
+        "--openai_key", openai_key,
+        "--gemini_key", gemini_key
     ]
     
     try:
-        # 运行 pipeline
         result = subprocess.run(cmd, capture_output=True, text=True, timeout=1800)
         logs = result.stdout + "\n" + result.stderr
     except subprocess.TimeoutExpired:
@@ -32,7 +33,6 @@ def run_pipeline(model_name_t, model_name_v, result_dir, paper_latex_root, arxiv
     except Exception as e:
         return f"❌ Error: {e}", None
 
-    # 检查输出目录
     if not OUTPUT_DIR.exists():
         return "❌ No output generated.", None
 
@@ -45,7 +45,7 @@ def run_pipeline(model_name_t, model_name_v, result_dir, paper_latex_root, arxiv
     
     return logs, ZIP_PATH
 
-# Gradio 界面
+# ===================== Gradio UI =====================
 iface = gr.Interface(
     fn=run_pipeline,
     inputs=[
@@ -54,6 +54,8 @@ iface = gr.Interface(
         gr.Textbox(label="Result Dir", value="output"),
         gr.Textbox(label="Paper LaTeX Root", value="input/latex_proj"),
         gr.Textbox(label="ArXiv URL", value="https://arxiv.org/abs/2505.21497"),
+        gr.Textbox(label="OpenAI API Key", placeholder="sk-...", type="password"),
+        gr.Textbox(label="Gemini API Key", placeholder="AIza...", type="password"),
     ],
     outputs=[
         gr.Textbox(label="Logs"),

pipeline.py

@@ -1,4 +1,3 @@
-# pipeline.py
 import cv2
 import pdb
 import json
@@ -16,30 +15,20 @@ print("Initializing...")
 
 from Paper2Video.src.slide_code_gen import latex_code_gen
 from Paper2Video.src.wei_utils import get_agent_config
-
-# ✅ 用小写包名
 from posterbuilder import build_poster as build_poster
-# 取出 build_poster 里用到的图片目录常量，便于推导 JSON 前缀
 from posterbuilder.build_poster import IMAGES_DIR_NAME
 
-# ----------------- 常用路径 -----------------
 ROOT_DIR = Path(__file__).resolve().parent
 P2V_ASSETS = ROOT_DIR / "Paper2Video" / "assets" / "demo" / "latex_proj"
 P2P_ROOT   = ROOT_DIR / "Paper2Poster"
 PB_ROOT    = ROOT_DIR / "posterbuilder"
-
-# 允许以包名 PosterAgent.* 方式导入（供潜在直接 import 使用）
 sys.path.append(str(P2P_ROOT))
 
-os.environ["GEMINI_API_KEY"] = "AIzaSyA1wVVdlYAVs3FULSmCVD1Noulwrq7zqeo"
-os.environ["OPENAI_API_KEY"] = "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"
-
 def copy_folder(src_dir, dst_dir):
     src_dir = Path(src_dir)
     dst_dir = Path(dst_dir)
     if not src_dir.exists():
         raise FileNotFoundError(f"no such dir: {src_dir}")
-    # 若目标存在，先删掉再复制
     if dst_dir.exists():
         shutil.rmtree(dst_dir)
     shutil.copytree(src_dir, dst_dir)
@@ -58,21 +47,14 @@ def str2list(s):
     return [int(x) for x in s.split(',')]
 
 def run_paper2poster_content_build():
-    """Step 1.5: 
-    1) 把 input/paper.pdf 复制到 Paper2Poster/input/paper/paper.pdf 
-    2) 运行 Paper2Poster 的 PosterAgent.new_pipeline 生成内容
-    3) 将 3 个 JSON 拷贝/改名到 posterbuilder/contents/（并在包根再放一份，兼容读取）
-    """
     print("🧩 Step 1.5: Preparing Paper2Poster inputs & generating poster contents ...")
 
-    # 1) 复制 PDF
     src_pdf = ROOT_DIR / "input" / "paper.pdf"
     dst_pdf = P2P_ROOT / "input" / "paper" / "paper.pdf"
     dst_pdf.parent.mkdir(parents=True, exist_ok=True)
     safe_copy(src_pdf, dst_pdf)
     print(f"   📄 Copied paper: {src_pdf.relative_to(ROOT_DIR)} → {dst_pdf.relative_to(ROOT_DIR)}")
 
-    # 2) 运行 PosterAgent.new_pipeline
     cmd = [
         sys.executable, "-m", "PosterAgent.new_pipeline",
         f'--poster_path={dst_pdf.relative_to(P2P_ROOT)}',
@@ -85,39 +67,28 @@ def run_paper2poster_content_build():
     subprocess.run(cmd, cwd=str(P2P_ROOT), check=True)
     print("   ✅ PosterAgent.new_pipeline finished.")
 
-    # 3) 拷贝三份 JSON
-    #    从 <4o_4o>_images_and_tables 推导前缀 <4o_4o>
-    tag_prefix = IMAGES_DIR_NAME.split("_images_and_tables")[0]  # "<4o_4o>"
-
-    # 源路径
+    tag_prefix = IMAGES_DIR_NAME.split("_images_and_tables")[0]
     src_raw_content = P2P_ROOT / "contents" / f"{tag_prefix}_paper_raw_content.json"
     src_tree_split  = P2P_ROOT / "tree_splits" / f"{tag_prefix}_paper_tree_split_0.json"
     src_images_json = P2P_ROOT / IMAGES_DIR_NAME / "paper_images.json"
 
-    # 目标（contents 下）
     dst_contents_dir = PB_ROOT / "contents"
     dst_raw_content  = dst_contents_dir / "poster_content.json"
     dst_tree_split   = dst_contents_dir / "arrangement.json"
     dst_fig_caption  = dst_contents_dir / "figure_caption.json"
 
-    # 目标（包根，再各放一份，兼容不同读取方式）
     dst_root_raw     = PB_ROOT / "poster_content.json"
     dst_root_tree    = PB_ROOT / "arrangement.json"
     dst_root_figcap  = PB_ROOT / "figure_caption.json"
 
-    # 拷贝
     safe_copy(src_raw_content, dst_raw_content)
     safe_copy(src_tree_split,  dst_tree_split)
     safe_copy(src_images_json, dst_fig_caption)
-    # 兼容：在包根也放一份
     safe_copy(src_raw_content, dst_root_raw)
     safe_copy(src_tree_split,  dst_root_tree)
     safe_copy(src_images_json, dst_root_figcap)
 
-    print("   📦 JSON copied & renamed:")
-    print(f"     - {src_raw_content.relative_to(ROOT_DIR)} → posterbuilder/contents/poster_content.json (+ root)")
-    print(f"     - {src_tree_split.relative_to(ROOT_DIR)} → posterbuilder/contents/arrangement.json (+ root)")
-    print(f"     - {src_images_json.relative_to(ROOT_DIR)} → posterbuilder/contents/figure_caption.json (+ root)")
+    print("   📦 JSON copied & renamed.")
     print("   ✅ Step 1.5 done.\n")
 
 if __name__ == '__main__':
@@ -130,19 +101,22 @@ if __name__ == '__main__':
     parser.add_argument('--if_tree_search', type=bool, default=True)
     parser.add_argument('--beamer_templete_prompt', type=str, default=None)
     parser.add_argument('--stage', type=str, default='["0"]')
-    parser.add_argument('--arxiv_url', type=str, default=None, help='arXiv paper URL (e.g., https://arxiv.org/abs/2401.12345)')
+    parser.add_argument('--arxiv_url', type=str, default=None)
+    parser.add_argument('--openai_key', type=str, required=True, help='Your OpenAI API key')
+    parser.add_argument('--gemini_key', type=str, required=True, help='Your Gemini API key')
 
     args = parser.parse_args()
     print("start")
 
-    # ================
-    # 🧹 清空 output 目录并新建子目录
-    # ================
+    # ✅ 使用传入的 key 设置环境变量
+    os.environ["OPENAI_API_KEY"] = args.openai_key
+    os.environ["GEMINI_API_KEY"] = args.gemini_key
+
+    # 清空 output
     output_dir = ROOT_DIR / "output"
     if output_dir.exists():
         print(f"   🧹 Clearing old output directory: {output_dir.relative_to(ROOT_DIR)}")
         shutil.rmtree(output_dir)
-
     (output_dir / "latex_proj").mkdir(parents=True, exist_ok=True)
     (output_dir / "poster_latex_proj").mkdir(parents=True, exist_ok=True)
     (output_dir / "slide_imgs").mkdir(parents=True, exist_ok=True)