Update gen_pr.py

gen_pr.py

@@ -4,13 +4,44 @@ from pathlib import Path
 
 from src import *
 
-import yaml
-import os
-
-# ============= 读取配置 ===============
-def load_config(config_path="config.yaml"):
-    with open(config_path, "r", encoding="utf-8") as f:
-        config = yaml.safe_load(f)
+# ============= 从环境变量动态生成配置 ===============
+def load_config():
+    """
+    不再读取 config.yaml，而是从 Gradio 主程序注入的环境变量中获取专属参数，
+    并组装成原有代码期望的字典结构，实现多用户隔离。
+    """
+    base_dir = os.path.dirname(os.path.abspath(__file__))
+    
+    # 获取由 app.py 注入的专属输出文件夹，如果没有则回退到本地默认文件夹（方便本地单独运行调试）
+    user_root_folder = os.environ.get("USER_OUTPUT_DIR", os.path.join(base_dir, "mineru_outputs"))
+    
+    # 获取由 app.py 注入的 API 凭证
+    api_key = os.environ.get("GEMINI_API_KEY", "")
+    api_base_url = os.environ.get("GEMINI_API_BASE_URL", "")
+    
+    # 可以在这里设置默认使用的模型，或者也改用环境变量传入
+    generation_model = os.environ.get("GENERATION_MODEL", "gemini-3-pro-preview")
+    
+    # 拼装成原版 config.yaml 的字典结构，保证后续代码兼容
+    config = {
+        "model_settings": {
+            "generation_model": generation_model
+        },
+        "path": {
+            "root_folder": user_root_folder
+        },
+        "api_keys": {
+            "gemini_api_key": api_key,
+            "api_base_url": api_base_url
+        }
+    }
+    
+    # 如果你的底层 src 代码库也需要直接用到 os.environ，我们在这里显式注入一下
+    if api_key:
+        os.environ["GEMINI_API_KEY"] = api_key
+    if api_base_url:
+        os.environ["GEMINI_API_BASE_URL"] = api_base_url
+        
     return config
 
 # ========== 读取 prompt 模板 ==========
@@ -26,8 +57,8 @@ model_name = config['model_settings']['generation_model']
 
 # ========== 主流程 ==========
 def main():
-    # 输入总文件夹路径（包含多个论文子文件夹）
-    root_folder = config['path']['root_folder']  # ⚠️ 修改为你的实际路径
+    # 输入总文件夹路径（包含多个论文子文件夹）此时指向用户的专属文件夹
+    root_folder = config['path']['root_folder']  
     prompt_path = "prompt.json"
 
 
@@ -98,7 +129,7 @@ def main():
         output_html = os.path.join(auto_path, "poster.html")
         graph_json_path = os.path.join(auto_path, "graph.json")
 
-        # # === 清理 markdown ===                                              去除无意义的段落，如relative work，reference，appendix等等
+        # # === 清理 markdown ===                                                   去除无意义的段落，如relative work，reference，appendix等等
         # print("🧹 Cleaning markdown before splitting...")
         # cleaned_md_path = clean_paper(md_path, clean_prompt, model="gemini-3-pro-preview", config=config)
 
@@ -133,29 +164,29 @@ def main():
 
 
 
-        # # # =============================  PPT部分  ================================
+        # # =============================  PPT部分  ================================
 
-        # # # ===  按照算法选出结点，以便后续生成outline ===
-        # # selected_node_path=os.path.join(auto_path, "selected_node.json")
-        # # generate_selected_nodes(dag_json_path=dag_path, max_len=15,output_path=selected_node_path)
+        # # ===  按照算法选出结点，以便后续生成outline ===
+        # selected_node_path=os.path.join(auto_path, "selected_node.json")
+        # generate_selected_nodes(dag_json_path=dag_path, max_len=15,output_path=selected_node_path)
 
-        # # # ===  初始化ouline ===
-        # # outline_path= os.path.join(auto_path, "outline.json")
-        # # outline = outline_initialize(dag_json_path=dag_path,outline_initialize_prompt=outline_initialize_prompt,model=model_name, config=config)
+        # # ===  初始化ouline ===
+        # outline_path= os.path.join(auto_path, "outline.json")
+        # outline = outline_initialize(dag_json_path=dag_path,outline_initialize_prompt=outline_initialize_prompt,model=model_name, config=config)
         
-        # # # ===  生成完整ouline ===
-        # # outline_data=generate_complete_outline(selected_node_path,outline_path,generate_complete_outline_prompt,model=model_name, config=config)
+        # # ===  生成完整ouline ===
+        # outline_data=generate_complete_outline(selected_node_path,outline_path,generate_complete_outline_prompt,model=model_name, config=config)
         
-        # # # ===  配模板 ===
-        # # arrange_template(outline_path,arrange_template_prompt,model=model_name, config=config)
+        # # ===  配模板 ===
+        # arrange_template(outline_path,arrange_template_prompt,model=model_name, config=config)
         
-        # # # ===  生成最终的PPT ===
-        # # ppt_template_path="./ppt_template"
-        # # generate_ppt_prompt = {"role":"system","content":"You are given (1) a slide node (JSON) and (2) an HTML slide template. Your task: revise the HTML template to produce the final slide HTML using the node content. Node fields: text (slide textual content), figure (list of images to display, each has name, caption, resolution), formula (list of formula images to display, each has name, caption, resolution), template (template filename). IMPORTANT RULES: 1) Only modify places in the HTML that are marked by comments like <!-- You need to revise the following parts. X: ... -->. 2) For 'Subjects' sections: replace the placeholder title with ONE concise summary sentence for this slide. 3) For 'Image' sections (<img ...>): replace src with the relative path extracted from node.figure/formula[i].name; the node image name may be markdown like '![](images/abc.jpg)', use only 'images/abc.jpg'. 4) For 'Text' sections: replace the placeholder text with the node.text content, formatted cleanly in HTML; keep it readable and you may use <p>, <ul><li>, <br/> appropriately. 5) If the template expects more images/text blocks than provided by the node, leave the missing positions unchanged and do not invent content. 6) If the node provides more images than the template has slots, fill slots in order and ignore the rest. 7) Preserve all other HTML, CSS, and structure exactly. OUTPUT FORMAT: Return ONLY the revised HTML as plain text. Do NOT wrap it in markdown fences. Do NOT add explanations."}
-        # # generate_ppt(outline_path,ppt_template_path,generate_ppt_prompt,model=model_name, config=config)
+        # # ===  生成最终的PPT ===
+        # ppt_template_path="./ppt_template"
+        # generate_ppt_prompt = {"role":"system","content":"You are given (1) a slide node (JSON) and (2) an HTML slide template. Your task: revise the HTML template to produce the final slide HTML using the node content. Node fields: text (slide textual content), figure (list of images to display, each has name, caption, resolution), formula (list of formula images to display, each has name, caption, resolution), template (template filename). IMPORTANT RULES: 1) Only modify places in the HTML that are marked by comments like . 2) For 'Subjects' sections: replace the placeholder title with ONE concise summary sentence for this slide. 3) For 'Image' sections (<img ...>): replace src with the relative path extracted from node.figure/formula[i].name; the node image name may be markdown like '![](images/abc.jpg)', use only 'images/abc.jpg'. 4) For 'Text' sections: replace the placeholder text with the node.text content, formatted cleanly in HTML; keep it readable and you may use <p>, <ul><li>, <br/> appropriately. 5) If the template expects more images/text blocks than provided by the node, leave the missing positions unchanged and do not invent content. 6) If the node provides more images than the template has slots, fill slots in order and ignore the rest. 7) Preserve all other HTML, CSS, and structure exactly. OUTPUT FORMAT: Return ONLY the revised HTML as plain text. Do NOT wrap it in markdown fences. Do NOT add explanations."}
+        # generate_ppt(outline_path,ppt_template_path,generate_ppt_prompt,model=model_name, config=config)
 
-        # # # ===  Refiner  ===                
-        # # refinement_ppt(input_index=auto_path, prompts=[commenter_prompt, reviser_prompt], model=model_name, max_iterations=3, config=config)
+        # # ===  Refiner  ===                
+        # refinement_ppt(input_index=auto_path, prompts=[commenter_prompt, reviser_prompt], model=model_name, max_iterations=3, config=config)
 
         # # =============================  Poster部分  ================================
         
@@ -236,7 +267,7 @@ def main():
 
         # =============================================================
         # 在auto目录下创建success.txt作为标志
-        success_file_path = os.path.join(auto_path, "success_pr.txt")
+        success_file_path = os.path.join(auto_path, "success_dag.txt")
         open(success_file_path, "w").close()