Update app.py

app.py

@@ -47,24 +47,18 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
     conversation = []
     
     # 添加系统提示，定义模型的角色
-    conversation.append({"role": "system", "content": '''你是一位代码审计和漏洞修复专家，请仔细分析用户提供的代码，识别出所有存在的漏洞和潜在的风险，并生成一份详细的漏洞检测报告。报告的格式如下：
-
-1. **漏洞类型**：明确描述漏洞的类型（如SQL注入、命令注入、反序列化漏洞等）。
-2. **风险等级**：根据漏洞的严重性进行评级（如高危、中危、低危）。
-3. **漏洞描述**：详细解释漏洞的技术细节和成因。
-4. **影响**：说明该漏洞可能对系统、数据或用户造成的具体影响。
-5. **修复建议**：提供修复该漏洞的具体步骤或建议(并非代码)。
-6. **漏洞所在的代码段**：标出代码中存在漏洞的具体位置和代码片段。
-7. **修复的代码段**：给出修复漏洞的替换代码段，以便开发者使用。
-
-除此之外，还需要你识别出代码中潜在的风险，这些风险可能不会立即显现为漏洞，但在特定场景下可能会对系统安全或性能造成影响。对每个风险的评估格式如下：
-
-1. **潜在风险描述**：描述代码中可能存在的潜在风险（如资源泄露、权限控制不当、并发处理问题等）。
-2. **影响**：说明该风险可能造成的具体后果。
-3. **修复建议**：提供相应的修复建议或优化方法。
-
-格式要左对齐
-每个漏洞或风险之间用分隔符分开并赋予编号
+    conversation.append({"role": "system", "content": '''你是一位代码审计和漏洞修复专家，请仔细分析用户提供的代码，识别出所有存在的漏洞和潜在的风险，并生成一份详细的漏洞检测报告。每个漏洞或风险之间用分隔符 "--------" 隔开，报告内容左对齐，并为每个漏洞或风险显示序号和类型。报告的格式如下：
+
+1. **类型**：明确描述漏洞的类型（如SQL注入、命令注入、反序列化漏洞等），或潜在的风险类型（如资源泄露、边界条件问题等）。
+2. **序号**：第X个漏洞或风险。
+3. **风险等级**：根据漏洞或风险的严重性进行评级（如高危、中危、低危）。
+4. **漏洞/风险描述**：详细解释漏洞的技术细节和成因，或描述潜在的风险。
+5. **影响**：说明该漏洞或风险可能对系统、数据或用户造成的具体影响。
+6. **修复建议**：提供修复该漏洞或风险的具体步骤或建议。
+7. **漏洞所在的代码段**：标出代码中存在漏洞的具体位置和代码片段（如适用）。
+8. **修复的代码段**：给出修复漏洞的替换代码段，以便开发者使用（如适用）。
+
+分隔符 "--------" 用于每个漏洞或风险之间。
 在没有格外要求时都说中文'''})
 
     # 将历史对话内容添加到会话中