#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import sqlite3

def fix_vietnamese_translations():
    """修正越南语翻译，确保与中文原文一致"""
    
    db_path = '/root/老业务网站8.30/backend/ai_agent_qa.db'
    
    # 修正后的越南语数据
    corrected_data = [
        {
            'old_question': 'Sản phẩm của bạn là gì?',
            'new_question': 'Sản phẩm của bạn là gì?',
            'new_answer': 'Chào bạn! Chúng tôi cung cấp ba giải pháp website cốt lõi: Website marketing giúp AI liên tục tạo ra nội dung marketing, nâng cao hiệu quả đề xuất của Google; Website tri thức tăng cường độ gắn kết khách hàng, khuyến khích khách hàng truy cập liên tục; Website hỏi đáp khớp chính xác nhu cầu người dùng, mang lại cảm giác dịch vụ chuyên nghiệp. Bạn quan tâm đến giải pháp nào nhất? Tôi sẵn sàng hỗ trợ thêm!'
        },
        {
            'old_question': 'Bạn cung cấp dịch vụ gì?',
            'new_question': 'Bạn cung cấp dịch vụ gì?',
            'new_answer': 'Chúng tôi cung cấp ba giải pháp website cốt lõi: Website marketing để AI liên tục tạo ra nội dung marketing, nâng cao hiệu quả đề xuất của Google; Website tri thức tăng cường độ gắn kết khách hàng, khuyến khích khách hàng truy cập liên tục; Website hỏi đáp khớp chính xác nhu cầu người dùng, để người dùng cảm nhận được dịch vụ chuyên nghiệp.'
        },
        {
            'old_question': 'Công ty bạn làm gì?',
            'new_question': 'Công ty bạn làm gì?',
            'new_answer': 'Chào bạn! Chúng tôi chuyên tạo ra các website tri thức tự động vận hành bằng AI, biến website thành công cụ marketing và phục vụ khách hàng hiệu quả. Thông qua công nghệ AI, chúng tôi giúp doanh nghiệp xây dựng hệ sinh thái website thông minh. Bạn có muốn tìm hiểu thêm về giải pháp cụ thể nào không?'
        }
    ]
    
    try:
        conn = sqlite3.connect(db_path)
        cursor = conn.cursor()
        
        print("🔧 开始修正越南语翻译...")
        
        for item in corrected_data:
            # 更新答案
            cursor.execute("""
                UPDATE qa_pairs 
                SET answer = ? 
                WHERE language = 'vi' AND question = ?
            """, (item['new_answer'], item['old_question']))
            
            print(f"✅ 已更新: {item['old_question']}")
        
        conn.commit()
        print(f"\n🎉 成功修正 {len(corrected_data)} 条越南语翻译")
        
        # 验证修正结果
        print("\n📝 修正后的内容预览:")
        cursor.execute("SELECT question, answer FROM qa_pairs WHERE language = 'vi' AND question LIKE '%Sản phẩm%'")
        result = cursor.fetchone()
        if result:
            print(f"Q: {result[0]}")
            print(f"A: {result[1][:150]}...")
        
        return True
        
    except Exception as e:
        print(f"❌ 修正失败: {e}")
        return False
    finally:
        if conn:
            conn.close()

if __name__ == "__main__":
    fix_vietnamese_translations()