app update

app.py

@@ -1,39 +1,47 @@
 import os
 from openai import OpenAI
 import gradio as gr
-# 删除原有的 openai.api_key 设置
-# 替换为新的客户端初始化
+from dotenv import load_dotenv
+import tempfile
+import atexit
 
+# 加载环境变量
+load_dotenv()
+
+# 初始化OpenAI客户端
 client = OpenAI(
     api_key=os.getenv("DASHSCOPE_API_KEY"),
     base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
 )
 
-# 修改 call_openai_api 函数
 def call_openai_api(prompt, temperature=0.7):
-    chat_completion = client.chat.completions.create(
-        messages=[
-            {
-                "role": "system",
-                "content": "你是一个专业的专利代理人，擅长撰写专利文档。",
-            },
-            {
-                "role": "user",
-                "content": prompt,
-            },
-        ],
-        model="qwen-plus",
-        temperature=temperature,
-        max_tokens=3000,
-        n=1
-    )
-    return chat_completion.choices[0].message.content.strip()
-
-# 定义每个部分的Agent Prompt模版
-# 在实际使用时，需根据您想要的生成要求对这些模版进行详细调整
+    """调用OpenAI API生成内容"""
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "system",
+                    "content": "你是一个专业的专利代理人，擅长撰写专利文档。",
+                },
+                {
+                    "role": "user",
+                    "content": prompt,
+                },
+            ],
+            model="qwen2.5-1.5b-instruct",
+            temperature=temperature,
+            max_tokens=3000,
+            n=1
+        )
+        return chat_completion.choices[0].message.content.strip()
+    except Exception as e:
+        print(f"API调用出错：{str(e)}")
+        return f"生成失败：{str(e)}"
+
+# 定义每个部分的Agent Prompt模板
 BACKGROUND_PROMPT = """
 你是一名资深的专利代理人，现在需要根据以下输入信息来撰写一份专利申请技术交底书的第一部分内容。
-请根据用户提供的背景材料和参考文献信息，详细说明本发明所处的技术领域背景和已有技术方案。
+请根据用户提供的背景材料和参考文献信息，详说明本发明所处的技术领域背景和已有技术方案。
 
 要求：
 1. 内容必须清晰、详实，包括宏观技术背景和具体现有技术方案。
@@ -67,7 +75,7 @@ PROBLEM_PROMPT = """
 请根据以下信息编写第三部分内容：
 三、本发明决的技术问题是：
 要求：
-- 针对第二部分所列出的缺点和不足，描述本发明试图解决的具体技术问题。
+- 对第二分所列出的缺点和不足，描述本发明试图解决的具体技术问题。
 - 内容应与本发明拟解决的问题相呼应。
 
 已有输出：
@@ -82,7 +90,7 @@ PROBLEM_PROMPT = """
 """
 
 SOLUTION_PROMPT = """
-请根据以下信息撰写第四部分内容：
+请根以下信息撰写第四部分内容：
 四、本发明技术方案的详细阐述：
 要求：
 - 详细描述本发明的技术方案，包括各功能模块/步骤的技术实现方式、结构图（可文字描述）和原理说明。
@@ -117,10 +125,10 @@ KEYPOINT_PROMPT = """
 
 ADVANTAGE_PROMPT = """
 请根据以下信息撰写第六部分内容：
-六、与第二条所述最好的现有技术相比，本发明的优点：
-要求：
+六、与第二条所述最好的现有技术相比，本发明的优：
+求：
 - 简要介绍本发明相对现有技术的有益效果和优势。
-- 结合技术方案来描述，做到有理有据。
+- 结技术方案来描述，做到有理有据。
 
 已有输出：
 第二部分（现有技术的缺点）：
@@ -148,7 +156,6 @@ ALTERNATIVE_PROMPT = """
 用户提供的需求：
 {user_input}
 """
-
 def generate_patent_document(
     bg_input, 
     shortcoming_input, 
@@ -159,12 +166,10 @@ def generate_patent_document(
     alternative_input,
     progress=gr.Progress()
 ):
-    # 1. 背景技术
     progress(0, desc="正在生成背景技术部分...")
     bg_prompt = BACKGROUND_PROMPT.format(user_input=bg_input)
     background_content = call_openai_api(bg_prompt)
 
-    # 2. 现有技术缺点
     progress(0.15, desc="正在生成现有技术缺点部分...")
     short_prompt = SHORTCOMING_PROMPT.format(
         previous_content=background_content,
@@ -172,7 +177,6 @@ def generate_patent_document(
     )
     shortcoming_content = call_openai_api(short_prompt)
 
-    # 3. 本发明解决的技术问题
     progress(0.3, desc="正在生成技术问题部分...")
     problem_prompt_full = PROBLEM_PROMPT.format(
         background_content=background_content,
@@ -181,7 +185,6 @@ def generate_patent_document(
     )
     problem_content = call_openai_api(problem_prompt_full)
 
-    # 4. 本发明技术方案
     progress(0.45, desc="正在生成技术方案部分...")
     solution_prompt_full = SOLUTION_PROMPT.format(
         background_content=background_content,
@@ -191,7 +194,6 @@ def generate_patent_document(
     )
     solution_content = call_openai_api(solution_prompt_full)
 
-    # 5. 关键点
     progress(0.6, desc="正在生成关键点部分...")
     keypoint_prompt_full = KEYPOINT_PROMPT.format(
         solution_content=solution_content,
@@ -199,7 +201,6 @@ def generate_patent_document(
     )
     keypoint_content = call_openai_api(keypoint_prompt_full)
 
-    # 6. 优点
     progress(0.75, desc="正在生成优点部分...")
     advantage_prompt_full = ADVANTAGE_PROMPT.format(
         shortcoming_content=shortcoming_content,
@@ -209,7 +210,6 @@ def generate_patent_document(
     )
     advantage_content = call_openai_api(advantage_prompt_full)
 
-    # 7. 替代方案
     progress(0.9, desc="正在生成替代方案部分...")
     alternative_prompt_full = ALTERNATIVE_PROMPT.format(
         solution_content=solution_content,
@@ -219,11 +219,17 @@ def generate_patent_document(
 
     progress(1.0, desc="生成完成！")
     
-    # 整合所有内容
-    final_document = f"{background_content}\n\n{shortcoming_content}\n\n{problem_content}\n\n{solution_content}\n\n{keypoint_content}\n\n{advantage_content}\n\n{alternative_content}"
+    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
 
-# 使用Gradio构建前端
 def gradio_interface(
     bg_input, 
     shortcoming_input, 
@@ -234,6 +240,7 @@ def gradio_interface(
     alternative_input,
     progress=gr.Progress()
 ):
+    progress_output = gr.Textbox()  # 用于显示进度的临时输出
     doc = generate_patent_document(
         bg_input, 
         shortcoming_input, 
@@ -244,25 +251,119 @@ def gradio_interface(
         alternative_input,
         progress=progress
     )
-    return doc
+    return progress_output, doc
+
+# ... existing code ...
+
+def generate_filename(text):
+    """根据专利交底书内容生成合适的文件名"""
+    try:
+        prompt = """
+        请根据以下专利交底书内容，生成一个简短的文件名（不超过50个字符）。
+        文件名应该能反映出发明的核心内容和技术领域。
+        只需要返回文件名，不需要其他解释。
+
+        专利交底书内容：
+        {text}
+        """.format(text=text[:1000])  # 只取前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"  # 默认文件名
+
+def download_text(text):
+    """将文本保存为临时文件并返回文件路径"""
+    filename = generate_filename(text)
+    # 创建临时文件，使用生成的文件名
+    temp_dir = tempfile.gettempdir()
+    file_path = os.path.join(temp_dir, filename)
+    
+    with open(file_path, mode="w", encoding="utf-8") as f:
+        f.write(text)
+    return file_path
+
+# ... existing code ...
+def clear_all():
+    """清空所有输入和输出"""
+    try:
+        # 清空下载文件组件前，先将其值设为None
+        return [""] * 9 + [None]  # 7个输入框 + 2个输出框 + 1个下载文件框(None)
+    except Exception as e:
+        print(f"清空过程中出错：{str(e)}")
+        return [""] * 9 + [None]
+    
+def cleanup_temp_files():
+    """清理临时文件"""
+    temp_dir = tempfile.gettempdir()
+    for file in os.listdir(temp_dir):
+        if file.endswith(".txt"):
+            try:
+                file_path = os.path.join(temp_dir, file)
+                if os.path.exists(file_path):
+                    os.remove(file_path)
+            except Exception as e:
+                print(f"删除临时文件失败：{str(e)}")
+                continue
+
 
 with gr.Blocks() as demo:
     gr.Markdown("## 专利交底书生成系统 (基于大模型的Agent生成)")
-    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="七、替代方案输入")
-
-    generate_button = gr.Button("开始生成")
-    output = gr.Textbox(label="生成的专利交底书文本", lines=30)
-
+    
+    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)
+            
+    # 设置下载按钮
+    download_button.click(
+        fn=download_text,
+        inputs=[final_output],
+        outputs=[download_file]
+    )
+    
+    # 设置生成按钮
     generate_button.click(
-        fn=gradio_interface, 
+        fn=gradio_interface,
         inputs=[bg, sh, pr, so, kp, adv, alt],
-        outputs=[output]
+        outputs=[progress_output, final_output]
     )
+    
+    # 设置清空按钮
+    clear_button.click(
+        fn=clear_all,
+        inputs=[],
+        outputs=[bg, sh, pr, so, kp, adv, alt, progress_output, 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()