fix

main.py

@@ -1,5 +1,6 @@
 import base64
 import copy
+from datetime import datetime
 import json
 import fire
 import os
@@ -91,7 +92,7 @@ def generate_paper_poster(
             #     json.dump(poster.model_dump(), f, ensure_ascii=False)
 
             poster_size = replace_figures_size_in_poster(backup_poster, figures)
-            print("Now generating HTML...")
+            print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] Now generating HTML...")
             result = generate_html_v2(vendor, model, poster_size, figures)
 
             html = result["html_with_figures"]

poster/poster.py

@@ -335,7 +335,7 @@ def generate_html_v2(vendor: str, model: str, poster: BaseModel, figures: list[s
         ]
     )
     layout_chain = layout_prompt | llm
-    output = layout_chain.invoke({"style": style, "poster": poster}).content
+    # output = layout_chain.invoke({"style": style, "poster": poster}).content
     layout_prompt.append(
         MessagesPlaceholder(variable_name="react"),
     )
@@ -409,6 +409,7 @@ def generate_html_v2(vendor: str, model: str, poster: BaseModel, figures: list[s
     def generate_single_html(prompt_input):
         """单个HTML生成函数，用于多线程执行"""
         result_output = layout_chain.invoke(prompt_input).content
+        print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] Generated HTML")
         body = re.search(r"```html\n(.*?)\n```", result_output, re.DOTALL).group(1)
         
         html = HTML_TEMPLATE.format(style=style, body=body)
@@ -451,10 +452,10 @@ def generate_html_v2(vendor: str, model: str, poster: BaseModel, figures: list[s
             min_poster_sizes = result["poster_sizes"]
             min_section_sizes = result["section_sizes"]
         
-        # 如果找到满足条件的结果，直接返回
-        if proportion <= 0.1:
-            print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] Initial generation successful, remaining {proportion:.0%} blank spaces.")
-            return {"html": result["html"], "html_with_figures": result["html_with_figures"]}
+    # 如果找到满足条件的结果，直接返回
+    if min_proportion <= 0.1:
+        print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] Initial generation successful, remaining {min_proportion:.0%} blank spaces.")
+        return {"html": min_html, "html_with_figures": min_html_with_figures}
 
     while True:
         attempt += 1