Update app.py

app.py

@@ -64,8 +64,8 @@ else:
     ROOT_BASE = os.getcwd()
 
 TASKS_ROOT_DIR = os.path.join(ROOT_BASE, "nihaisha_tasks_v14_final") 
-CORPUS_ROOT_DIR = os.path.join(ROOT_BASE, "nihaisha_corpus_lib")       
-PERSONAS_FILE = os.path.join(ROOT_BASE, "personas_config.json")       
+CORPUS_ROOT_DIR = os.path.join(ROOT_BASE, "nihaisha_corpus_lib")        
+PERSONAS_FILE = os.path.join(ROOT_BASE, "personas_config.json")        
 
 os.makedirs(TASKS_ROOT_DIR, exist_ok=True)
 os.makedirs(CORPUS_ROOT_DIR, exist_ok=True)
@@ -219,10 +219,11 @@ class TaskManager:
     def create_task(self, task_name_prefix="task", custom_label=None):
         timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
         
-        # 任务文件夹命名逻辑
+        # 任务文件夹命名逻辑 - 【修复】如果输入了主题，直接作为ID前缀
         if custom_label:
-            safe_label = re.sub(r'[\\/*?:"<>|]', "", custom_label)[:15] 
-            task_id = f"{timestamp}_{safe_label}"
+            safe_label = re.sub(r'[\\/*?:"<>|]', "", custom_label)[:30] 
+            # 将自定义标签放在前面，方便查找
+            task_id = f"{safe_label}_{timestamp}"
         else:
             task_id = f"{task_name_prefix}_{timestamp}"
             
@@ -422,22 +423,26 @@ def process_single_topic(idx, topic, persona_name, custom_sys, custom_user, ref_
     
     generated_files = []
     
+    # 【修复】文件1：标题+视频详情 (Meta)
     meta_content = f"【标题】\n{title_part}\n\n【YouTube视频描述】\n{desc_part}"
-    meta_path = os.path.join(res_dir, f"{idx+1}_{safe_filename}_Meta.txt")
+    meta_path = os.path.join(res_dir, f"{idx+1}_{safe_filename}_标题与详情.txt")
     with open(meta_path, "w", encoding="utf-8") as f:
         f.write(meta_content)
     generated_files.append(meta_path)
     
-    tts_path = os.path.join(res_dir, f"{idx+1}_{safe_filename}_TTS.txt")
+    # 【修复】文件2：正文 (TTS)
+    tts_path = os.path.join(res_dir, f"{idx+1}_{safe_filename}_正文(TTS).txt")
     with open(tts_path, "w", encoding="utf-8") as f:
         f.write(cleaned_tts_body)
     generated_files.append(tts_path)
     
+    # Debug文件（可选）
     debug_path = os.path.join(res_dir, f"{idx+1}_debug.log")
     with open(debug_path, "w", encoding="utf-8") as f:
         f.write(f"PROMPT:\n{final_user_prompt}")
         
     logger_func(f"✅ 完成！正文约 {len(cleaned_tts_body)} 字")
+    # 返回生成的文件列表和自动提取的标题
     return generated_files, title_part.split('\n')[0]
 
 def worker(task_id, topics, persona_name, custom_sys, custom_user, selected_refs, manual_topic_name):
@@ -481,8 +486,9 @@ def worker(task_id, topics, persona_name, custom_sys, custom_user, selected_refs
                     traceback.print_exc()
         
         if all_files_to_zip:
+            # 【修复】Zip文件命名逻辑：优先使用手动输入的主题
             if manual_topic_name:
-                safe_zip_name = re.sub(r'[\\/*?:"<>|]', "", manual_topic_name)[:30]
+                safe_zip_name = re.sub(r'[\\/*?:"<>|]', "", manual_topic_name)[:50]
                 zip_filename = f"{safe_zip_name}.zip"
             elif first_auto_title:
                 safe_zip_name = re.sub(r'[\\/*?:"<>|]', "", first_auto_title)[:30]
@@ -491,10 +497,14 @@ def worker(task_id, topics, persona_name, custom_sys, custom_user, selected_refs
                 zip_filename = f"Batch_{len(all_files_to_zip)}_Files.zip"
                 
             zip_path = os.path.join(task_dir, zip_filename)
+            
             with zipfile.ZipFile(zip_path, 'w') as z:
                 for f in all_files_to_zip: z.write(f, os.path.basename(f))
             
-            task_manager.update_status(task_id, "completed", zip_path)
+            # 【修复】确保传给前端的是绝对路径，解决下载失败问题
+            abs_zip_path = os.path.abspath(zip_path)
+            
+            task_manager.update_status(task_id, "completed", abs_zip_path)
             log(f"🎉 任务结束，ZIP已打包: {zip_filename}")
         else:
             task_manager.update_status(task_id, "failed")
@@ -530,10 +540,12 @@ def start_execution(topic_text, persona_name, p_sys, p_user, selected_refs):
         return "❌ 既无主题也无素材，无法生成", None
         
     topics = topic_text.split('\n')
+    # 【修复】确保明确提取第一个主题作为任务名
     manual_topic_name = topics[0].strip() if topic_text.strip() else None
     
     if not topic_text.strip(): topics = [""]
     
+    # 传递 manual_topic_name 给 create_task
     tid, _ = task_manager.create_task(custom_label=manual_topic_name)
     
     threading.Thread(target=worker, args=(tid, topics, persona_name, p_sys, p_user, selected_refs, manual_topic_name)).start()