更新了应用程序代码

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,26 +17,28 @@ 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=[
                 {
                     "role": "system",
-                    "content": "你是一个专业的专利代理人，擅长撰写专利文档。",
+                    "content": "你是一个专业的专利代理人，正在完成一份专利文档的撰写。",
                 },
                 {
                     "role": "user",
                     "content": prompt,
                 },
             ],
-            model="qwen2.5-1.5b-instruct",
+            model="qwen-max-latest",
             temperature=temperature,
             max_tokens=3000,
             n=1
         )
+        print("API调用成功")  # 添加日志
         return chat_completion.choices[0].message.content.strip()
     except Exception as e:
-        print(f"API调用出错：{str(e)}")
-        return f"生成失败：{str(e)}"
+        print(f"API调用详细错误：{type(e).__name__}: {str(e)}")  # 增强错误信息
+        return f"生成失败：{type(e).__name__}: {str(e)}"
 
 # 定义每个部分的Agent Prompt模板
 BACKGROUND_PROMPT = """
@@ -73,9 +75,9 @@ SHORTCOMING_PROMPT = """
 
 PROBLEM_PROMPT = """
 请根据以下信息编写第三部分内容：
-三、本发明决的技术问题是：
+三、本发明解决的技术问题是：
 要求：
-- 对第二分所列出的缺点和不足，描述本发明试图解决的具体技术问题。
+- 对第二部分所列出的缺点和不足，描述本发明试图解决的具体技术问题。
 - 内容应与本发明拟解决的问题相呼应。
 
 已有输出：
@@ -90,7 +92,7 @@ PROBLEM_PROMPT = """
 """
 
 SOLUTION_PROMPT = """
-请根以下信息撰写第四部分内容：
+请根据以下信息撰写第四部分内容：
 四、本发明技术方案的详细阐述：
 要求：
 - 详细描述本发明的技术方案，包括各功能模块/步骤的技术实现方式、结构图（可文字描述）和原理说明。
@@ -126,9 +128,9 @@ KEYPOINT_PROMPT = """
 ADVANTAGE_PROMPT = """
 请根据以下信息撰写第六部分内容：
 六、与第二条所述最好的现有技术相比，本发明的优：
-求：
+要求：
 - 简要介绍本发明相对现有技术的有益效果和优势。
-- 结技术方案来描述，做到有理有据。
+- 结合技术方案来描述，做到有理有据。
 
 已有输出：
 第二部分（现有技术的缺点）：
@@ -156,129 +158,157 @@ ALTERNATIVE_PROMPT = """
 用户提供的需求：
 {user_input}
 """
-def generate_patent_document(
-    bg_input, 
-    shortcoming_input, 
-    problem_input, 
-    solution_input, 
-    keypoint_input,
-    advantage_input, 
-    alternative_input,
-    progress=gr.Progress()
-):
-    progress(0, desc="正在生成背景技术部分...")
-    bg_prompt = BACKGROUND_PROMPT.format(user_input=bg_input)
-    background_content = call_openai_api(bg_prompt)
-
-    progress(0.15, desc="正在生成现有技术缺点部分...")
-    short_prompt = SHORTCOMING_PROMPT.format(
-        previous_content=background_content,
-        user_input=shortcoming_input
-    )
-    shortcoming_content = call_openai_api(short_prompt)
 
-    progress(0.3, desc="正在生成技术问题部分...")
-    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="正在生成技术方案部分...")
-    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="正在生成关键点部分...")
-    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="正在生成优点部分...")
-    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="正在生成替代方案部分...")
-    alternative_prompt_full = ALTERNATIVE_PROMPT.format(
-        solution_content=solution_content,
-        user_input=alternative_input
-    )
-    alternative_content = call_openai_api(alternative_prompt_full)
-
-    progress(1.0, desc="生成完成！")
+# 验证函数示例
+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)
     
-    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}"
-    )
-    return final_document
-
-def gradio_interface(
-    bg_input, 
-    shortcoming_input, 
-    problem_input, 
-    solution_input, 
-    keypoint_input,
-    advantage_input, 
-    alternative_input,
-    progress=gr.Progress()
-):
-    progress_output = gr.Textbox()  # 用于显示进度的临时输出
-    doc = generate_patent_document(
-        bg_input, 
-        shortcoming_input, 
-        problem_input, 
-        solution_input, 
-        keypoint_input,
-        advantage_input, 
-        alternative_input,
-        progress=progress
-    )
-    return progress_output, doc
-
-# ... existing code ...
+    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 = ""
+    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
+    except Exception as e:
+        return f"生成过程中发生错误：{str(e)}"
 
 def generate_filename(text):
     """根据专利交底书内容生成合适的文件名"""
     try:
-        prompt = """
+        prompt = f"""
         请根据以下专利交底书内容，生成一个简短的文件名（不超过50个字符）。
         文件名应该能反映出发明的核心内容和技术领域。
         只需要返回文件名，不需要其他解释。
 
         专利交底书内容：
-        {text}
-        """.format(text=text[:1000])  # 只取前1000个字符用于生成文件名
-        
+        {text[:1000]}
+        """
         filename = call_openai_api(prompt).strip()
         # 确保文件名合法
         filename = "".join(c for c in filename if c not in r'\/:*?"<>|')
         return filename + ".txt"
     except Exception as e:
         print(f"生成文件名失败：{str(e)}")
-        return "专利交底书.txt"  # 默认文件名
+        return "专利交底书.txt"
 
 def download_text(text):
     """将文本保存为临时文件并返回文件路径"""
     filename = generate_filename(text)
-    # 创建临时文件，使用生成的文件名
     temp_dir = tempfile.gettempdir()
     file_path = os.path.join(temp_dir, filename)
     
@@ -286,16 +316,14 @@ def download_text(text):
         f.write(text)
     return file_path
 
-# ... existing code ...
 def clear_all():
     """清空所有输入和输出"""
     try:
-        # 清空下载文件组件前，先将其值设为None
-        return [""] * 9 + [None]  # 7个输入框 + 2个输出框 + 1个下载文件框(None)
+        return [""] * 7 + ["", "", None]  # 7个输入框 + 2个输出框 + 1个下载文件框(None)
     except Exception as e:
         print(f"清空过程中出错：{str(e)}")
-        return [""] * 9 + [None]
-    
+        return [""] * 7 + ["", "", None]
+
 def cleanup_temp_files():
     """清理临时文件"""
     temp_dir = tempfile.gettempdir()
@@ -309,61 +337,110 @@ def cleanup_temp_files():
                 print(f"删除临时文件失败：{str(e)}")
                 continue
 
+atexit.register(cleanup_temp_files)
 
-with gr.Blocks() as demo:
-    gr.Markdown("## 专利交底书生成系统 (基于大模型的Agent生成)")
-    
-    with gr.Column():
-        bg = gr.Textbox(label="一、背景技术与现有技术方案输入")
-        sh = gr.Textbox(label="二、现有技术缺点输入")
-        pr = gr.Textbox(label="三、本发明解决的技术问题输入")
-        so = gr.Textbox(label="四、本发明技术方案输入")
-        kp = gr.Textbox(label="五、关键点输入")
-        adv = gr.Textbox(label="六、本发明优点输入")
-        alt = gr.Textbox(label="七、替代方案输入")
-        
-        with gr.Row():
-            generate_button = gr.Button("开始生成", variant="primary")
-            clear_button = gr.Button("清空所有", variant="secondary")
-    
-        progress_output = gr.Textbox(label="生成进度", lines=2)
-        final_output = gr.Markdown(label="生成的专利交底书文本")
-        
-        with gr.Row():
-            download_button = gr.Button("创建下载文件", variant="secondary")
-            download_file = gr.File(label="点击下载文件", visible=True)
-            
-    # 设置下载按钮
+custom_css = """
+.gradio-container {
+    background-color: #f0f2f6;
+    padding: 20px;
+}
+.gradio-container h2 {
+    color: #333333;
+}
+.gr-button-primary {
+    background-color: #4B6CB7;
+    color: white;
+}
+.gr-button-secondary {
+    background-color: #A5A5A5;
+    color: white;
+}
+.gr-textbox, .gr-markdown {
+    border-radius: 5px;
+    padding: 10px;
+    border: 1px solid #ccc;
+}
+"""
+
+with gr.Blocks(theme=gr.themes.Soft(
+    primary_hue="blue",
+    secondary_hue="indigo",
+    neutral_hue="slate"
+)) as demo:
+    gr.Markdown("## 📝 专利交底书生成系统\n**基于先进的语言模型技术，为您自动生成高质量的专利文档。**\n\n请按顺序填写下列所有部分的信息，然后点击\"🚀 开始生成\"按钮。")
+
+    with gr.Accordion("📥 输入信息", open=True):
+        bg = gr.Textbox(
+            label="一、背景技术与现有技术方案输入",
+            placeholder="请输入背景技术和现有技术方案...",
+            lines=4
+        )
+        sh = gr.Textbox(
+            label="二、现有技术缺点输入",
+            placeholder="请输入现有技术的缺点...",
+            lines=4
+        )
+        pr = gr.Textbox(
+            label="三、本发明解决的技术问题输入",
+            placeholder="请输入本发明解决的技术问题...",
+            lines=4
+        )
+        so = gr.Textbox(
+            label="四、本发明技术方案输入",
+            placeholder="请输入本发明的技术方案...",
+            lines=4
+        )
+        kp = gr.Textbox(
+            label="五、关键点输入",
+            placeholder="请输入本发明的关键点...",
+            lines=3
+        )
+        adv = gr.Textbox(
+            label="六、本发明优点输入",
+            placeholder="请输入本发明的优点...",
+            lines=3
+        )
+        alt = gr.Textbox(
+            label="七、替代方案输入",
+            placeholder="请输入替代方案...",
+            lines=3
+        )
+
+    with gr.Row():
+        generate_button = gr.Button("🚀 开始生成", variant="primary")
+        clear_button = gr.Button("🧹 清空所有", variant="secondary")
+
+    progress_bar = gr.Progress()
+    status_box = gr.Textbox(label="生成状态", interactive=False)  # 添加标签和设置为只读
+
+    final_output = gr.Markdown(label="生成的专利交底书文本")
+    with gr.Row():
+        download_button = gr.Button("⬇️ 创建下载文件", variant="secondary")
+        download_file = gr.File(label="点击下载文件", visible=False)
+
+    # 设置下载按钮行为
     download_button.click(
         fn=download_text,
         inputs=[final_output],
         outputs=[download_file]
     )
-    
-    # 设置生成按钮
+
+    # 设置生成按钮行为
     generate_button.click(
-        fn=gradio_interface,
+        fn=generate_patent_document,
         inputs=[bg, sh, pr, so, kp, adv, alt],
-        outputs=[progress_output, final_output]
+        outputs=[final_output]
+    ).then(
+        # 修改 lambda 函数，添加输入参数
+        fn=lambda x: gr.update(visible=True),
+        inputs=[final_output],
+        outputs=[download_file]
     )
-    
-    # 设置清空按钮
+
     clear_button.click(
         fn=clear_all,
         inputs=[],
-        outputs=[bg, sh, pr, so, kp, adv, alt, progress_output, final_output, download_file]
+        outputs=[bg, sh, pr, so, kp, adv, alt, final_output, download_file]
     )
 
-# 在程序退出时清理临时文件
-def cleanup_temp_files():
-    temp_dir = tempfile.gettempdir()
-    for file in os.listdir(temp_dir):
-        if file.endswith(".txt"):
-            try:
-                os.remove(os.path.join(temp_dir, file))
-            except:
-                pass
-
-atexit.register(cleanup_temp_files)
-
-demo.launch()
+demo.launch()