#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
修正越南语数据库中的不准确内容
"""

import sqlite3

def fix_vietnamese_database():
    """修正越南语数据库中的不准确内容"""
    
    db_path = '/root/老业务网站8.30/backend/ai_agent_qa.db'
    
    # 需要修正的越南语内容
    corrections = [
        {
            "question": "AI tự động tạo nội dung như thế nào?",
            "answer": "Hệ thống AI của chúng tôi thông qua thuật toán học sâu, phân tích tài liệu doanh nghiệp cung cấp, thông tin sản phẩm, kiến thức ngành nghề, tự động tạo ra các bài viết, hỏi đáp, giới thiệu sản phẩm liên quan. Hệ thống sẽ liên tục học từ phản hồi người dùng, không ngừng tối ưu hóa chất lượng nội dung, đảm bảo tính chuyên nghiệp và chính xác của nội dung.",
            "new_answer": "Hệ thống AI của chúng tôi thông qua thuật toán học sâu, phân tích tài liệu doanh nghiệp cung cấp, thông tin sản phẩm, kiến thức ngành nghề, tự động tạo ra các bài viết, hỏi đáp, giới thiệu sản phẩm liên quan. Hệ thống sẽ liên tục học từ phản hồi người dùng, không ngừng tối ưu hóa chất lượng nội dung, đảm bảo tính chuyên nghiệp và chính xác của nội dung."
        }
    ]
    
    try:
        conn = sqlite3.connect(db_path)
        cursor = conn.cursor()
        
        print("🇻🇳 开始修正越南语数据库内容...")
        
        for correction in corrections:
            question = correction["question"]
            new_answer = correction["new_answer"]
            
            # 检查是否存在该问题
            cursor.execute("SELECT COUNT(*) FROM qa_pairs WHERE language = 'vi' AND question = ?", (question,))
            count = cursor.fetchone()[0]
            
            if count > 0:
                # 更新答案
                cursor.execute("UPDATE qa_pairs SET answer = ? WHERE language = 'vi' AND question = ?", 
                              (new_answer, question))
                print(f"✅ 已更新问题: {question}")
            else:
                print(f"❌ 未找到问题: {question}")
        
        conn.commit()
        print("🎉 越南语数据库修正完成！")
        
        # 验证修正结果
        print("\n📊 验证修正结果:")
        for correction in corrections:
            question = correction["question"]
            cursor.execute("SELECT answer FROM qa_pairs WHERE language = 'vi' AND question = ?", (question,))
            result = cursor.fetchone()
            if result:
                print(f"Q: {question}")
                print(f"A: {result[0][:200]}...")
                print()
        
        return True
        
    except Exception as e:
        print(f"❌ 修正越南语数据库失败: {e}")
        return False
    finally:
        if conn:
            conn.close()

if __name__ == "__main__":
    print("🚀 开始修正越南语数据库...")
    success = fix_vietnamese_database()
    if success:
        print("🎉 越南语数据库修正完成！")
    else:
        print("💥 越南语数据库修正失败！")