Update pipeline.py

pipeline.py

@@ -43,20 +43,26 @@ def safe_copy(src: Path, dst: Path):
 def str2list(s): 
     return [int(x) for x in s.split(',')]
 
-def run_paper2poster_content_build():
+def run_paper2poster_content_build(orientation="landscape"):
     print("🧩 Step 1.5: Preparing Paper2Poster inputs & generating poster contents ...")
 
     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)
+
+    if orientation.lower() == "portrait":
+        width_in, height_in = "36", "48"
+    else:
+        width_in, height_in = "48", "36"
+
     cmd = [
         sys.executable, "-m", "PosterAgent.new_pipeline",
         f'--poster_path={dst_pdf.relative_to(P2P_ROOT)}',
         '--model_name_t=gpt-5',
         '--model_name_v=gpt-5',
-        '--poster_width_inches=48',
-        '--poster_height_inches=36'
+        f'--poster_width_inches={width_in}',
+        f'--poster_height_inches={height_in}'
     ]
     print("   ▶ Running: python -m PosterAgent.new_pipeline ...")
     subprocess.run(cmd, cwd=str(P2P_ROOT), check=True)
@@ -193,6 +199,7 @@ if __name__ == '__main__':
     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')
     parser.add_argument('--logo_dir', type=str, required=True, help='Directory containing uploaded logo image(s)')
+    parser.add_argument('--orientation', type=str, choices=['landscape', 'portrait'], default='landscape', help='Poster layout orientation')
     args = parser.parse_args()
     print("start")
 
@@ -301,7 +308,7 @@ if __name__ == '__main__':
     # Step 1.5: Poster2Poster 内容生成
     # =========================
     try:
-        run_paper2poster_content_build()
+        run_paper2poster_content_build(orientation=args.orientation)
     except Exception as e:
         print(f"❌ Step 1.5 failed: {e}")
 
@@ -310,7 +317,7 @@ if __name__ == '__main__':
     # =========================
     try:
         print("🧩 Step 2: Building poster ...")
-        build_poster()
+        build_poster(orientation=args.orientation)
         print("✅ Step 2 done.")
     except Exception as e:
         print(f"❌ Step 2 failed: {e}")