Update app.py

app.py

@@ -8,6 +8,7 @@ import psutil
 import random
 import threading
 import time
+import traceback
 import weakref
 from dataclasses import dataclass, field
 from typing import Any, Optional, Tuple, List, Dict
@@ -32,6 +33,9 @@ API_TOKEN = os.environ.get("API_TOKEN", "")
 if not API_TOKEN:
     raise ValueError("环境变量中未设置API_TOKEN")
 
+# 飞书通知配置
+CHAT_WEBHOOK_URL = os.environ.get("CHAT_WEBHOOK_URL", "")
+
 # 外部链接
 DISCORD_LINK = os.environ.get("DISCORD_LINK", "https://discord.com/invite/AtRtbe9W8w")
 APP_INDEX_LINK = os.environ.get("APP_INDEX_LINK", "https://app.nieta.art/")
@@ -129,6 +133,49 @@ def validate_dimensions(width: int, height: int) -> tuple[int, int]:
     return width, height
 
 
+def feishu_notify(message: str):
+    """发送飞书通知（异步非阻塞）"""
+    threading.Thread(target=_send_feishu_notify, args=(message,)).start()
+
+
+def _send_feishu_notify(message: str):
+    """发送飞书通知的内部实现"""
+    try:
+        headers = {
+            "Content-Type": "application/json",
+        }
+        content = {
+            "msg_type": "text",
+            "content": {
+                "text": message,
+            },
+        }
+
+        with httpx.Client(timeout=10.0) as client:
+            response = client.post(CHAT_WEBHOOK_URL, headers=headers, json=content)
+            if response.status_code == 200:
+                logger.info("飞书通知发送成功")
+            else:
+                logger.warning(f"飞书通知发送失败: {response.status_code}")
+    except Exception as e:
+        logger.error(f"发送飞书通知时出错: {str(e)}")
+
+
+def format_error_for_notification(error_type: str, error_message: str, traceback_str: str) -> str:
+    """格式化错误信息用于飞书通知"""
+    return f"""
+🚨 应用错误警报
+
+错误类型: {error_type}
+错误信息: {error_message}
+
+详细错误栈:
+{traceback_str}
+
+时间: {time.strftime('%Y-%m-%d %H:%M:%S')}
+"""
+
+
 class PolishClient:
     """提示词润色客户端"""
 
@@ -171,10 +218,21 @@ async def polish_prompt(prompt: str) -> str:
     if polish_client is None:
         polish_client = PolishClient()
 
-    logger.info(f"润色提示词: {prompt[:50]}...")
-    polished_prompt = await polish_client.polish_text(prompt)
-    logger.info("提示词润色完成")
-    return polished_prompt
+    try:
+        logger.info(f"润色提示词: {prompt[:50]}...")
+        polished_prompt = await polish_client.polish_text(prompt)
+        logger.info("提示词润色完成")
+        return polished_prompt
+    except Exception as e:
+        error_message = f"提示词润色异常: {str(e)}"
+        traceback_str = traceback.format_exc()
+        logger.error(error_message)
+
+        # 发送飞书通知
+        notification_message = format_error_for_notification("PolishPromptError", error_message, traceback_str)
+        feishu_notify(notification_message)
+
+        return prompt  # 返回原始提示词作为fallback
 
 
 # 核心类定义
@@ -410,7 +468,14 @@ class ImageClient:
                     return None, result["error"]
 
         except Exception as e:
-            logger.error(f"生成图片异常: {str(e)}")
+            error_message = f"生成图片异常: {str(e)}"
+            traceback_str = traceback.format_exc()
+            logger.error(error_message)
+
+            # 发送飞书通知
+            notification_message = format_error_for_notification("ImageGenerationError", error_message, traceback_str)
+            feishu_notify(notification_message)
+
             return None, f"生成图片时发生错误: {str(e)}"
         finally:
             gc.collect()
@@ -575,7 +640,14 @@ async def infer(
     except gr.Error:
         raise
     except Exception as e:
-        logger.error(f"推理过程异常: {str(e)}")
+        error_message = f"推理过程异常: {str(e)}"
+        traceback_str = traceback.format_exc()
+        logger.error(error_message)
+
+        # 发送飞书通知
+        notification_message = format_error_for_notification("InferenceError", error_message, traceback_str)
+        feishu_notify(notification_message)
+
         raise gr.Error(f"生成图像时发生意外错误: {str(e)}")
     finally:
         gc.collect()
@@ -820,14 +892,27 @@ def main():
     except KeyboardInterrupt:
         logger.info("接收到键盘中断，正在关闭...")
     except Exception as e:
-        logger.error(f"应用启动失败: {str(e)}")
+        error_message = f"应用启动失败: {str(e)}"
+        traceback_str = traceback.format_exc()
+        logger.error(error_message)
+
+        # 发送飞书通知
+        notification_message = format_error_for_notification("ApplicationStartupError", error_message, traceback_str)
+        feishu_notify(notification_message)
+
         raise
     finally:
         # 确保资源清理
         try:
             asyncio.run(cleanup_resources())
         except Exception as e:
-            logger.error(f"最终清理时出错: {str(e)}")
+            error_message = f"最终清理时出��: {str(e)}"
+            traceback_str = traceback.format_exc()
+            logger.error(error_message)
+
+            # 发送飞书通知
+            notification_message = format_error_for_notification("CleanupError", error_message, traceback_str)
+            feishu_notify(notification_message)
         logger.info("应用已退出")