update package

main.py

@@ -71,6 +71,7 @@ app.mount("/screenshots", StaticFiles(directory=CACHE_DIR), name="screenshots")
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
 # API Keys
+SECRET_KEY = os.getenv("SECRET_KEY", "wangyue")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 OPENAI_API_IMAGE_EDIT_KEY = os.getenv("OPENAI_API_IMAGE_EDIT_KEY")
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
@@ -194,17 +195,16 @@ def index() -> FileResponse:
 @app.post("/api/analyze")
 async def analyze_feedback(request: AnalysisRequest):
     try:
-        context = "以下是老板对设计的反馈内容：\n" + request.text
-        
-        if request.image_data:
-            context += "\n\n用户还上传了设计图片作为参考："
-        
+        context = "这是用户上传的设计图。" if request.image_data else ''
+        context += "以下是老板对设计的反馈内容：\n" + request.text
+        context += '\n请根据老板的反馈分析情绪值（用emoji表示），并结合给出的设计图稿，给出三个具体的修改建议，最后给出合适的搜索关键词，以获取合适的参考设计案例。每个建议应该包含一个标题和详细描述。首先你需要对老板的情绪进行解读，使用"情绪值："开头并分为五类：1. 非常满意-😊😊😊 2. 比较满意-🙂🙂🙂 3. 一般般-😐😐😐 4. 不太满意-🙁🙁🙁 5. 非常不满意-😠😠😠，然后在下一行用一句话分析老板的情绪，以"情绪分析："开头。随后，请以"修改建议：\n"开头，并以有序列表分三行说明三个具体建议，并以"\n\n"分隔，比如："1. 提高对比度：xxx\n\n 2. ...\n\n 3. ...\n\n"。记得结合图片进行分析和提出修改建议。最后，你需要使用网页搜索来获取合适的参考UI设计案例，来为修改当前案例提供参考。请在新的一行以"搜索内容："开头，给出合适的搜索内容，以获取合适的参考设计案例，注意，你只能搜索UI界面或虚拟形象设计相关的案例。'
+
         # 调用OpenAI API进行分析
         response = await call_openai_api(
-            system_prompt='你是一位专业的设计顾问，擅长分析客户反馈，提取关键信息，并提供专业建议。请根据老板的反馈分析情绪值（用emoji表示），并结合给出的设计稿，给出三个具体的修改建议。每个建议应该包含一个标题和详细描述。首先你需要对老板的情绪进行解读，使用"情绪值："开头并分为五类：1. 非常满意-😊😊😊 2. 比较满意-🙂🙂🙂 3. 一般般-😐😐😐 4. 不太满意-🙁🙁🙁 5. 非常不满意-😠😠😠，然后在下一行用一句话分析老板的情绪，以"情绪分析："开头。随后，请以"修改建议：\n"开头，并以有序列表分三行说明三个具体建议，并以\n\n分隔，比如："1. 提高对比度：xxx\n\n 2. ...\n\n 3. ...\n\n"。记得结合图片进行分析和提出修改建议。最后，你需要使用网页搜索来获取合适的参考UI设计案例，来为修改当前案例提供参考。请在新的一行以"搜索内容："开头，给出合适的搜索内容，以获取合适的参考设计案例，注意，你只能搜索UI界面或虚拟形象设计相关的案例。',
+            system_prompt='你是一位专业的设计顾问，擅长分析客户反馈，提取关键信息，并提供专业建议，你需要严格遵循用户给出的输出格式要求。',
             user_content=[
-                {"type": "input_text", "text": context},
-                *([{"type": "input_image", "image_url": request.image_data}] if request.image_data else [])
+                *([{"type": "input_image", "image_url": request.image_data}] if request.image_data else []),
+                {"type": "input_text", "text": context}
             ],
             request_model_id=request.request_model_id
         )
@@ -216,7 +216,7 @@ async def analyze_feedback(request: AnalysisRequest):
 def decrypt_user_key(encrypted_key: str) -> str:
     try:
         # 使用与前端相同的私钥
-        private_key = 'wangyuejunhui'
+        private_key = SECRET_KEY
         
         # 创建RSA解密对象
         cipher = PKCS1_OAEP.new(RSA.import_key(private_key))
@@ -291,7 +291,13 @@ async def optimize_design(request: OptimizationRequest):
                     status_code=400,
                     detail="无效的用户密钥"
                 )
-            user_info = eval(user_key)
+            try:
+                user_info = eval(user_key)
+            except:
+                raise HTTPException(
+                    status_code=400,
+                    detail="无效的用户密钥"
+                )
             
             credit_data = {
                 "user_key": request.user_key,