更新了应用程序代码

app.py

@@ -1,9 +1,9 @@
 import os
+from openai import OpenAI
 import gradio as gr
+from dotenv import load_dotenv
 import tempfile
 import atexit
-from dotenv import load_dotenv
-from openai import OpenAI
 
 # 加载环境变量
 load_dotenv()
@@ -17,7 +17,6 @@ client = OpenAI(
 def call_openai_api(prompt, temperature=0.7):
     """调用OpenAI API生成内容"""
     try:
-        print(f"开始调用API，prompt长度: {len(prompt)}")  # 添加日志
         chat_completion = client.chat.completions.create(
             messages=[
                 {
@@ -34,11 +33,10 @@ def call_openai_api(prompt, temperature=0.7):
             max_tokens=3000,
             n=1
         )
-        print("API调用成功")  # 添加日志
         return chat_completion.choices[0].message.content.strip()
     except Exception as e:
-        print(f"API调用详细错误：{type(e).__name__}: {str(e)}")  # 增强错误信息
-        return f"生成失败：{type(e).__name__}: {str(e)}"
+        print(f"API调用出错：{str(e)}")
+        return f"生成失败：{str(e)}"
 
 # 定义每个部分的Agent Prompt模板
 BACKGROUND_PROMPT = """
@@ -159,133 +157,93 @@ ALTERNATIVE_PROMPT = """
 {user_input}
 """
 
-
-# 验证函数示例
-def validate_content(content, expected_keywords=None):
-    """
-    自动化验证内容是否合理。
-    - 检查内容是否包含预期的关键词。
-    - 确保内容长度合理。
-    - 其他自定义验证规则。
-    """
-    if expected_keywords:
-        for keyword in expected_keywords:
-            if keyword.lower() not in content.lower():
-                raise ValidationError(f"内容缺少关键词: {keyword}")
-    if len(content) < 100:
-        raise ValidationError("内容过短，可能不够详实。")
-    # 添加更多验证规则
-    return True
-
-class ValidationError(Exception):
-    """自定义验证错误"""
-    def __init__(self, message):
-        self.message = message
-        super().__init__(self.message)
-    
-    def get_retry_prompt(self):
-        return f"请根据反馈修改内容: {self.message}"
-
-def generate_with_hitl(prompt_template, user_input, previous_contents=None, max_retries=3):
-    """通用的生成函数，包含HITL流程"""
-    print(f"开始生成内容，用户输入长度: {len(user_input)}")  # 添加日志
-    retry_prompt = ""
-    retries = 0
-    success = False
-    final_content = ""
-    
-    while retries < max_retries and not success:
-        try:
-            # 生成内容
-            if previous_contents:
-                prompt = prompt_template.format(
-                    user_input=user_input,
-                    **previous_contents
-                )
-            else:
-                prompt = prompt_template.format(user_input=user_input)
-            
-            print(f"第{retries + 1}次尝试生成")  # 添加日志
-            content = call_openai_api(prompt)
-            
-            # 检查返回内容是否包含错误信息
-            if content.startswith("生成失败"):
-                print(f"生成返回错误: {content}")  # 添加日志
-                raise Exception(content)
-                
-            # 自动化验证
-            validate_content(content, expected_keywords=["技术", "方案"])
-            success = True
-            final_content = content
-            print("内容生成成功")  # 添加日志
-
-        except ValidationError as e:
-            print(f"验证失败: {str(e)}")  # 添加日志
-            retry_prompt = e.get_retry_prompt()
-            retries += 1
-        except Exception as e:
-            print(f"生成过程中发生错误: {type(e).__name__}: {str(e)}")  # 增强错误信息
-            retries += 1
-    
-    if not success:
-        final_content = "生成失败，请检查输入或稍后重试。"
-    
-    return final_content
-def generate_patent_document(bg_input, sh_input, pr_input, so_input, kp_input, adv_input, alt_input, progress=gr.Progress()):
-    """
-    生成完整的专利交底书，包含HITL流程的每个模块。
-    """
-    final_document = ""
+def generate_patent_document(
+    bg_input, 
+    shortcoming_input, 
+    problem_input, 
+    solution_input, 
+    keypoint_input,
+    advantage_input, 
+    alternative_input,
+    progress=gr.Progress()  # 使用progress参数
+):
     try:
-        # 生成第一部分：背景技术
-        progress(0.14, desc="生成背景技术部分...")
-        background_content = generate_with_hitl(BACKGROUND_PROMPT, bg_input)
-        final_document += f"{background_content}\n\n"
-        
-        # 生成第二部分：现有技术缺点
-        progress(0.28, desc="生成现有技术缺点部分...")
-        shortcoming_content = generate_with_hitl(SHORTCOMING_PROMPT, sh_input, {"background_content": background_content})
-        final_document += f"{shortcoming_content}\n\n"
-        
-        # 生成第三部分：技术问题
-        progress(0.42, desc="生成技术问题部分...")
-        problem_content = generate_with_hitl(PROBLEM_PROMPT, pr_input, {
-            "background_content": background_content,
-            "shortcoming_content": shortcoming_content
-        })
-        final_document += f"{problem_content}\n\n"
-        
-        # 生成第四部分：技术方案
-        progress(0.56, desc="生成技术方案部分...")
-        solution_content = generate_with_hitl(SOLUTION_PROMPT, so_input, {
-            "background_content": background_content,
-            "shortcoming_content": shortcoming_content,
-            "problem_content": problem_content
-        })
-        final_document += f"{solution_content}\n\n"
-        
-        # 生成第五部分：关键点
-        progress(0.70, desc="生成关键点部分...")
-        keypoint_content = generate_with_hitl(KEYPOINT_PROMPT, kp_input, {"solution_content": solution_content})
-        final_document += f"{keypoint_content}\n\n"
-        
-        # 生成第六部分：优点
-        progress(0.84, desc="生成优点部分...")
-        advantage_content = generate_with_hitl(ADVANTAGE_PROMPT, adv_input, {
-            "shortcoming_content": shortcoming_content,
-            "problem_content": problem_content,
-            "solution_content": solution_content
-        })
-        final_document += f"{advantage_content}\n\n"
-        
-        # 生成第七部分：替代方案
-        progress(1.0, desc="生成替代方案部分...")
-        alternative_content = generate_with_hitl(ALTERNATIVE_PROMPT, alt_input, {"solution_content": solution_content})
-        final_document += f"{alternative_content}"
-        
-        return final_document
+        # 依次生成各部分，并在每个阶段更新进度
+        progress(0, desc="正在生成背景技术部分...")
+        yield "111正在生成背景技术部分...", None  # 更新状态文本框
+        bg_prompt = BACKGROUND_PROMPT.format(user_input=bg_input)
+        background_content = call_openai_api(bg_prompt)
+
+        progress(0.15, desc="正在生成现有技术缺点部分...")
+        yield "111正在生成现有技术缺点部分...", None
+        short_prompt = SHORTCOMING_PROMPT.format(
+            previous_content=background_content,
+            user_input=shortcoming_input
+        )
+        shortcoming_content = call_openai_api(short_prompt)
+
+        progress(0.3, desc="正在生成技术问题部分...")
+        yield "111正在生成技术问题部分...", None
+        problem_prompt_full = PROBLEM_PROMPT.format(
+            background_content=background_content,
+            shortcoming_content=shortcoming_content,
+            user_input=problem_input
+        )
+        problem_content = call_openai_api(problem_prompt_full)
+
+        progress(0.45, desc="正在生成技术方案部分...")
+        yield "111正在生成技术方案部分...", None
+        solution_prompt_full = SOLUTION_PROMPT.format(
+            background_content=background_content,
+            shortcoming_content=shortcoming_content,
+            problem_content=problem_content,
+            user_input=solution_input
+        )
+        solution_content = call_openai_api(solution_prompt_full)
+
+        progress(0.6, desc="正在生成关键点部分...")
+        yield "111正在生成关键点部分...", None
+        keypoint_prompt_full = KEYPOINT_PROMPT.format(
+            solution_content=solution_content,
+            user_input=keypoint_input
+        )
+        keypoint_content = call_openai_api(keypoint_prompt_full)
+
+        progress(0.75, desc="正在生成优点部分...")
+        yield "正在生成优点部分...", None
+        advantage_prompt_full = ADVANTAGE_PROMPT.format(
+            shortcoming_content=shortcoming_content,
+            problem_content=problem_content,
+            solution_content=solution_content,
+            user_input=advantage_input
+        )
+        advantage_content = call_openai_api(advantage_prompt_full)
+
+        progress(0.9, desc="正在生成替代方案部分...")
+        yield "正在生成替代方案部分...", None
+        alternative_prompt_full = ALTERNATIVE_PROMPT.format(
+            solution_content=solution_content,
+            user_input=alternative_input
+        )
+        alternative_content = call_openai_api(alternative_prompt_full)
+
+
+        final_document = (
+            f"{background_content}\n\n"
+            f"{shortcoming_content}\n\n"
+            f"{problem_content}\n\n"
+            f"{solution_content}\n\n"
+            f"{keypoint_content}\n\n"
+            f"{advantage_content}\n\n"
+            f"{alternative_content}"
+        )
+
+        progress(1.0, desc="生成完成！")
+        yield "生成完成！", final_document
+
     except Exception as e:
-        return f"生成过程中发生错误：{str(e)}"
+        error_message = f"生成过程中发生错误：{str(e)}"
+        yield error_message, error_message
 
 def generate_filename(text):
     """根据专利交底书内容生成合适的文件名"""
@@ -425,18 +383,19 @@ with gr.Blocks(theme=gr.themes.Soft(
         outputs=[download_file]
     )
 
-    # 设置生成按钮行为
+    # 不使用status和show_progress，仅依赖progress参数来显示进度条
     generate_button.click(
         fn=generate_patent_document,
         inputs=[bg, sh, pr, so, kp, adv, alt],
-        outputs=[final_output]
+        outputs=[status_box, final_output]
     ).then(
         # 修改 lambda 函数，添加输入参数
         fn=lambda x: gr.update(visible=True),
-        inputs=[final_output],
+        inputs=[final_output],  # 添加输入
         outputs=[download_file]
     )
 
+
     clear_button.click(
         fn=clear_all,
         inputs=[],