使用logtail记录日志

app.py

@@ -4,6 +4,9 @@ import gradio as gr
 from dotenv import load_dotenv
 import tempfile
 import atexit
+from logtail import LogtailHandler
+import logging
+import datetime
 
 # 加载环境变量
 load_dotenv()
@@ -15,20 +18,20 @@ client = OpenAI(
 )
 
 
-import datetime  # 导入 datetime 模块
-
-USER_COUNT_FILE = "click_log.txt"  # 修改计数器文件名，更名为日志文件名
+logtail_handler = LogtailHandler(source_token=os.getenv("LOGTAIL_TOKEN"))
+logger = logging.getLogger(__name__)
+logger.addHandler(logtail_handler)
 
 def increment_user_counter():
-    """记录用户点击事件的时间戳到日志文件"""
-    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") # 获取当前时间并格式化
-    log_entry = f"{timestamp}\n" # 创建日志条目，包含时间戳和换行符
+    """使用Logtail记录点击事件（替换原文件写入方法）"""
     try:
-        with open(USER_COUNT_FILE, "a") as f: # 以追加模式打开日志文件
-            f.write(log_entry) # 写入日志条目
-        print(f"用户点击记录已添加: {timestamp}") # 可选：在控制台打印日志信息
+        logger.info("用户点击生成事件", extra={
+            'event_type': 'generate_click',
+            'timestamp': datetime.datetime.now().isoformat(),
+            'service_version': '1.0'
+        })
     except Exception as e:
-        print(f"记录用户点击日志出错: {str(e)}")
+        print(f"日志记录失败: {str(e)}")
 
 def call_openai_api(prompt, temperature=0.7, model="deepseek-v3"):
     """调用OpenAI API生成内容"""
@@ -378,6 +381,7 @@ with gr.Blocks(theme=gr.themes.Soft(
     neutral_hue="slate"
 )) as demo:
     gr.Markdown("## 📝 专利交底书生成系统\n### **基于语言模型技术，为您自动生成专利文档。**")
+    gr.Markdown("## ❗❗❗ 大模型引用的任何专利号或者论文资料等都需要二次审查❗❗❗ **")
     gr.Markdown('- 请按顺序填写下列所有部分的信息，然后点击\"🚀 开始生成\"按钮。')
     gr.Markdown("- 请提供分点的自然语言信息，得到更高质量的输出结果")