Upload app.py

app.py

@@ -28,6 +28,7 @@ def get_db_connection():
 
 
 def save_to_db(table, data):
+    conn = None  # 确保 conn 在函数开始时就被定义
     try:
         conn = get_db_connection()
         with conn.cursor() as cursor:
@@ -39,7 +40,9 @@ def save_to_db(table, data):
     except Exception as e:
         print(f"数据库写入失败: {e}")
     finally:
-        conn.close()
+        if conn:
+            conn.close()
+
 
 # ======================== 模型加载 ========================
 NER_MODEL_NAME = "hfl/chinese-roberta-wwm-ext-large"