Update gen_poster.py

gen_poster.py

@@ -24,7 +24,7 @@ def load_config():
         "api_keys": {
             "gemini_api_key": api_key
         },
-        "api_base_url": api_base_url
+        "api_base_url": api_base_url  # <--- ✅ 放在这里才对！
     }
     
     if api_key:
@@ -93,9 +93,9 @@ def main():
             continue  # 只处理存在 auto 文件夹的目录
         
         # ✅ 如果 success.txt 已存在，跳过该目录
-        success_flag = os.path.join(auto_path, "success_pr.txt")
+        success_flag = os.path.join(auto_path, "success_poster.txt")
         if os.path.isfile(success_flag):
-            print(f"✅ success_pr.txt exists in {auto_path}, skipping...")
+            print(f"✅ success_poster.txt exists in {auto_path}, skipping...")
             continue
         
         print(f"\n🚀 Processing paper folder: {auto_path}")
@@ -180,86 +180,86 @@ def main():
         # # ===  Refiner  ===                
         # refinement_ppt(input_index=auto_path, prompts=[commenter_prompt, reviser_prompt], model=model_name, max_iterations=3, config=config)
 
-        # # =============================  Poster部分  ================================
+        # =============================  Poster部分  ================================
         
-        # poster_outline_path = os.path.join(auto_path, "poster_outline.txt")
+        poster_outline_path = os.path.join(auto_path, "poster_outline.txt")
         
-        # print(f"📝 Generating poster outline at: {poster_outline_path}")
-        # generate_poster_outline_txt(dag_path=dag_path,poster_outline_path=poster_outline_path,poster_outline_prompt=poster_outline_prompt,model=model_name, config=config)
-        # print (f"✅ Poster outline generated.")
+        print(f"📝 Generating poster outline at: {poster_outline_path}")
+        generate_poster_outline_txt(dag_path=dag_path,poster_outline_path=poster_outline_path,poster_outline_prompt=poster_outline_prompt,model=model_name, config=config)
+        print (f"✅ Poster outline generated.")
 
 
-        # poster_path=os.path.join(auto_path, "poster.html")
+        poster_path=os.path.join(auto_path, "poster.html")
 
 
-        # subdir_name = Path(subdir_path).name
-        # poster_outline_path_modified  = os.path.join(auto_path, "poster_outline_modified.txt")
-        # print(f"📝 Modifying poster outline for paper: {subdir_name}")
-        # modify_poster_outline(poster_outline_path=poster_outline_path,poster_paper_name=subdir_name,modified_poster_outline_path=poster_outline_path_modified)
-        # print (f"✅ Poster outline modified.")
+        subdir_name = Path(subdir_path).name
+        poster_outline_path_modified  = os.path.join(auto_path, "poster_outline_modified.txt")
+        print(f"📝 Modifying poster outline for paper: {subdir_name}")
+        modify_poster_outline(poster_outline_path=poster_outline_path,poster_paper_name=subdir_name,modified_poster_outline_path=poster_outline_path_modified)
+        print (f"✅ Poster outline modified.")
         
-        # modified_poster_logic(poster_outline_path_modified, modified_poster_logic_prompt, model=model_name, config=config)
+        modified_poster_logic(poster_outline_path_modified, modified_poster_logic_prompt, model=model_name, config=config)
 
 
-        # print(f"🖼️ Building poster HTML at: {poster_path}")
-        # build_poster_from_outline(poster_outline_path=poster_outline_path_modified,poster_template_path="./poster_template/poster_template.html",poster_path=poster_path,)
-        # print (f"✅ Poster HTML built.")
+        print(f"🖼️ Building poster HTML at: {poster_path}")
+        build_poster_from_outline(poster_outline_path=poster_outline_path_modified,poster_template_path="./poster_template/poster_template.html",poster_path=poster_path,)
+        print (f"✅ Poster HTML built.")
         
 
-        # print(f"🖊️ Modifying title and authors in poster HTML...")
-        # modify_title_and_author(dag_path=dag_path,poster_path=poster_path)
-        # print (f"✅ Title and authors modified.")
+        print(f"🖊️ Modifying title and authors in poster HTML...")
+        modify_title_and_author(dag_path=dag_path,poster_path=poster_path)
+        print (f"✅ Title and authors modified.")
 
 
-        # poster_final_index = os.path.join(auto_path, "final")
-        # os.makedirs(poster_final_index, exist_ok=True)
+        poster_final_index = os.path.join(auto_path, "final")
+        os.makedirs(poster_final_index, exist_ok=True)
 
-        # poster_final_output_path = os.path.join(poster_final_index, "poster_final.html")
-        # print(f"🖊️ Refining poster HTML with Gemini...")
+        poster_final_output_path = os.path.join(poster_final_index, "poster_final.html")
+        print(f"🖊️ Refining poster HTML with Gemini...")
         
 
-        # out = inject_img_section_to_poster(figure_path="./poster_template/expore_our_work_in_detail.jpg",auto_path=auto_path,poster_path=poster_path)
+        out = inject_img_section_to_poster(figure_path="./poster_template/expore_our_work_in_detail.jpg",auto_path=auto_path,poster_path=poster_path)
 
 
 
 
-        # refinement_poster(input_html_path=poster_path, prompts=poster_refinement_prompt,output_html_path=poster_final_output_path,model=model_name, config=config)
+        refinement_poster(input_html_path=poster_path, prompts=poster_refinement_prompt,output_html_path=poster_final_output_path,model=model_name, config=config)
         
-        # print (f"✅ Poster HTML refined. Final poster at: {poster_final_output_path}")
+        print (f"✅ Poster HTML refined. Final poster at: {poster_final_output_path}")
 
 
-        # =============================  PR部分  ================================
-        pr_template_path="./pr_template.md"
-        basic_information_path = extract_basic_information(dag_path=dag_path, auto_path=auto_path,extract_basic_information_prompt=extract_basic_information_prompt,model=model_name, config=config)
+        # # =============================  PR部分  ================================
+        # pr_template_path="./pr_template.md"
+        # basic_information_path = extract_basic_information(dag_path=dag_path, auto_path=auto_path,extract_basic_information_prompt=extract_basic_information_prompt,model=model_name, config=config)
         
-        initialize_pr_markdown(basic_information_path=basic_information_path,auto_path=auto_path,pr_template_path=pr_template_path)
+        # initialize_pr_markdown(basic_information_path=basic_information_path,auto_path=auto_path,pr_template_path=pr_template_path)
 
-        pr_path=os.path.join(auto_path, "markdown.md")
+        # pr_path=os.path.join(auto_path, "markdown.md")
 
 
-        generate_pr_from_dag(dag_path=dag_path, pr_path=pr_path, generate_pr_prompt=generate_pr_prompt, model=model_name, config=config)
-        print(f"📝 PR generated at: {pr_path}")
+        # generate_pr_from_dag(dag_path=dag_path, pr_path=pr_path, generate_pr_prompt=generate_pr_prompt, model=model_name, config=config)
+        # print(f"📝 PR generated at: {pr_path}")
         
-        add_title_and_hashtag(pr_path=pr_path, add_title_and_hashtag_prompt=add_title_and_hashtag_prompt, model=model_name, config=config)
-        add_institution_tag(pr_path=pr_path)
+        # add_title_and_hashtag(pr_path=pr_path, add_title_and_hashtag_prompt=add_title_and_hashtag_prompt, model=model_name, config=config)
+        # add_institution_tag(pr_path=pr_path)
         
 
 
-        dedup_consecutive_markdown_images(pr_path, inplace=True)
+        # dedup_consecutive_markdown_images(pr_path, inplace=True)
 
-        print(f"✅ PR markdown post-processed.")
+        # print(f"✅ PR markdown post-processed.")
 
-        print(f"🖊️ Refining PR markdown with LLM...")
-        pr_refine_path=os.path.join(auto_path, "markdown_refined.md")
+        # print(f"🖊️ Refining PR markdown with LLM...")
+        # pr_refine_path=os.path.join(auto_path, "markdown_refined.md")
 
-        refinement_pr(pr_path=pr_path, pr_refine_path=pr_refine_path, prompts=pr_refinement_prompt, model=model_name, config=config)
-        print (f"✅ PR markdown refined.")
+        # refinement_pr(pr_path=pr_path, pr_refine_path=pr_refine_path, prompts=pr_refinement_prompt, model=model_name, config=config)
+        # print (f"✅ PR markdown refined.")
 
 
 
         # =============================================================
         # 在auto目录下创建success.txt作为标志
-        success_file_path = os.path.join(auto_path, "success_pr.txt")
+        success_file_path = os.path.join(auto_path, "success_poster.txt")
         open(success_file_path, "w").close()