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

import sqlite3
import os

def add_thai_qa_data():
    """添加泰语问答数据"""
    
    # 数据库路径
    db_path = '/root/老业务网站8.30/backend/ai_agent_qa.db'
    
    # 泰语问答数据 (基于中文数据精确翻译)
    thai_qa_data = [
        {
            'question': 'บริษัทของคุณทำอะไร?',
            'answer': 'สวัสดีครับ! เราเชี่ยวชาญในการสร้างเว็บไซต์ความรู้ที่ขับเคลื่อนด้วย AI อัตโนมัติ เปลี่ยนเว็บไซต์ให้เป็นเครื่องมือการตลาดและบริการลูกค้าที่มีประสิทธิภาพ ผ่านเทคโนโลยี AI เราช่วยธุรกิจสร้างระบบนิเวศเว็บไซต์อัจฉริยะ คุณอยากทราบเพิ่มเติมเกี่ยวกับโซลูชันใดเป็นพิเศษไหมครับ?',
            'category': 'company_info',
            'keywords': 'บริษัท, ธุรกิจ, AI เว็บไซต์, ความรู้'
        },
        {
            'question': 'ผลิตภัณฑ์ของคุณคืออะไร?',
            'answer': 'สวัสดีครับ! เราให้บริการโซลูชันเว็บไซต์หลัก 3 แบบ: เว็บไซต์การตลาด (AI สร้างเนื้อหาการตลาดอย่างต่อเนื่อง เพิ่มประสิทธิภาพการแนะนำของ Google), เว็บไซต์ความรู้ (เสริมสร้างความผูกพันของลูกค้า ส่งเสริมการเข้าชมอย่างต่อเนื่อง), เว็บไซต์ถาม-ตอบ (จับคู่ความต้องการผู้ใช้อย่างแม่นยำ ให้ความรู้สึกบริการมืออาชีพ) คุณสนใจโซลูชันไหนที่อยากรู้รายละเอียดเพิ่มเติมครับ?',
            'category': 'service_inquiry',
            'keywords': 'ผลิตภัณฑ์, บริการ, เว็บไซต์, AI, การตลาด'
        },
        {
            'question': 'จุดเด่นของคุณคืออะไร?',
            'answer': 'จุดเด่นหลักของเราคือกลไกการทำงานอัตโนมัติด้วย AI - เว็บไซต์สามารถให้บริการลูกค้า 24/7 ด้วยเนื้อหาและบริการที่มีคุณค่า ทำการตลาดอัตโนมัติ เพิ่มประสิทธิภาพประสบการณ์ลูกค้า ทำให้เว็บไซต์กลายเป็นเครื่องมือการตลาดและบริการที่มีประสิทธิภาพที่สุดของธุรกิจครับ',
            'category': 'company_info',
            'keywords': 'จุดเด่น, AI อัตโนมัติ, 24/7, การตลาด'
        },
        {
            'question': 'คุณให้บริการอะไรบ้าง?',
            'answer': 'เราให้บริการโซลูชันเว็บไซต์หลัก 3 แบบ: เว็บไซต์การตลาดเพื่อให้ AI สร้างเนื้อหาการตลาดอย่างต่อเนื่อง เพิ่มประสิทธิภาพการแนะนำของ Google; เว็บไซต์ความรู้เสริมสร้างความผูกพันของลูกค้า ส่งเสริมการเข้าชมอย่างต่อเนื่อง; เว็บไซต์ถาม-ตอบจับคู่ความต้องการผู้ใช้อย่างแม่นยำ เพื่อให้ผู้ใช้รู้สึกถึงบริการมืออาชีพครับ',
            'category': 'service_inquiry',
            'keywords': 'บริการ, เว็บไซต์การตลาด, ความรู้, ถาม-ตอบ'
        },
        {
            'question': 'ลูกค้าเป้าหมายของคุณคือใคร?',
            'answer': 'ลูกค้าเป้าหมายของเราคือธุรกิจที่ต้องการทำการตลาดผ่านเว็บไซต์ บริการลูกค้า การจัดการความรู้ โดยเฉพาะธุรกิจที่ต้องการทำการตลาดอัตโนมัติ เพิ่มประสิทธิภาพประสบการณ์ลูกค้า เสริมสร้างอิทธิพลแบรนด์ครับ',
            'category': 'company_info',
            'keywords': 'ลูกค้าเป้าหมาย, ธุรกิจ, การตลาด, แบรนด์'
        },
        {
            'question': 'ฟังก์ชันการตลาดมีอะไรบ้าง?',
            'answer': 'ฟังก์ชันการตลาดประกอบด้วย: การประมวลผลความรู้อัตโนมัติด้วย AI สร้างเนื้อหาความรู้และเนื้อหาถาม-ตอบอัตโนมัติ; สร้างคีย์เวิร์ดหางยาวมากขึ้น ขยายขอบเขตการค้นหาเว็บไซต์; ปรับปรุงงาน SEO ระบบให้ Google แนะนำและดึงข้อมูลอย่างต่อเนื่องได้ง่ายขึ้น ใช้หลักในการตลาดเว็บไซต์องค์กร การส่งเสริมแบรนด์ การปรับปรุง SEO การตลาดเนื้อหา การดึงดูดผู้ใช้ครับ',
            'category': 'service_inquiry',
            'keywords': 'การตลาด, AI, SEO, คีย์เวิร์ด, Google'
        },
        {
            'question': 'ฟังก์ชันความรู้มีลักษณะอย่างไร?',
            'answer': 'ลักษณะฟังก์ชันความรู้ประกอบด้วย: AI สร้างเนื้อหาอย่างต่อเนื่อง รักษาการอัปเดตเนื้อหา; เพิ่มการตอบสนองอัจฉริยะ AI ทำให้ผู้ใช้รู้สึกว่าเว็บไซต์มีคุณค่า; ส่งเสริมให้ลูกค้าเข้าชมเว็บไซต์อย่างต่อเนื่อง เพิ่มความรู้เพิ่มความผูกพัน ใช้หลักในการจัดการความรู้ภายในองค์กร การศึกษาลูกค้า การให้คำปรึกษามืออาชีพครับ',
            'category': 'service_inquiry',
            'keywords': 'ความรู้, AI, เนื้อหา, คุณค่า, ความผูกพัน'
        },
        {
            'question': 'ราคาบริการเป็นอย่างไร?',
            'answer': 'สวัสดีครับ! ราคาบริการของเราปรับแต่งตามขนาดและความต้องการเฉพาะของโครงการ เราจึงมีแผนการกำหนดราคาที่ยืดหยุ่น สำหรับราคาที่ละเอียด คุณสามารถติดต่อผ่านแบบฟอร์มติดต่อบนเว็บไซต์ ทีมงานจะสนับสนุนคุณทันที! คุณต้องการข้อมูลเพิ่มเติมอะไรไหมครับ?',
            'category': 'price_inquiry',
            'keywords': 'ราคา, บริการ, ใบเสนอราคา, ปรับแต่ง'
        },
        {
            'question': 'จะติดต่อคุณได้อย่างไร?',
            'answer': 'คุณสามารถติดต่อเราผ่านแบบฟอร์มติดต่อบนเว็บไซต์นี้ครับ เราจะตอบกลับอย่างรวดเร็วและให้คำปรึกษาโดยละเอียดเกี่ยวกับโซลูชันที่เหมาะสมที่สุดสำหรับธุรกิจของคุณครับ',
            'category': 'contact',
            'keywords': 'ติดต่อ, แบบฟอร์ม, คำปรึกษา, ตอบกลับ'
        },
        {
            'question': 'เว็บไซต์ AI สามารถทำอะไรได้บ้าง?',
            'answer': 'เว็บไซต์ AI ของเราสามารถสร้างเนื้อหาอัตโนมัติ ตอบคำถามลูกค้า 24/7 ปรับปรุง SEO เพิ่มความผูกพันผู้ใช้ และทำการตลาดอัตโนมัติ นี่คือโซลูชันครบวงจรที่ช่วยธุรกิจประหยัดเวลาและเพิ่มประสิทธิภาพการดำเนินธุรกิจครับ',
            'category': 'service_inquiry',
            'keywords': 'เว็บไซต์ AI, อัตโนมัติ, เนื้อหา, SEO, ประสิทธิภาพ'
        }
    ]
    
    try:
        # 连接数据库
        conn = sqlite3.connect(db_path)
        cursor = conn.cursor()
        
        # 检查表是否存在
        cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='qa_pairs'")
        if not cursor.fetchone():
            print("❌ 表 qa_pairs 不存在")
            return False
        
        # 删除现有的泰语数据（如果有）
        cursor.execute("DELETE FROM qa_pairs WHERE language = 'th'")
        print(f"🗑️ 已清理现有泰语数据")
        
        # 插入泰语数据
        insert_count = 0
        for qa in thai_qa_data:
            cursor.execute("""
                INSERT INTO qa_pairs (question, answer, language, category, keywords)
                VALUES (?, ?, 'th', ?, ?)
            """, (qa['question'], qa['answer'], qa['category'], qa['keywords']))
            insert_count += 1
        
        # 提交更改
        conn.commit()
        print(f"✅ 成功添加 {insert_count} 条泰语问答数据")
        
        # 验证数据
        cursor.execute("SELECT COUNT(*) FROM qa_pairs WHERE language = 'th'")
        count = cursor.fetchone()[0]
        print(f"📊 数据库中现有 {count} 条泰语记录")
        
        # 显示几条示例
        cursor.execute("SELECT question, answer FROM qa_pairs WHERE language = 'th' LIMIT 3")
        examples = cursor.fetchall()
        print("\n📝 示例数据:")
        for i, (q, a) in enumerate(examples, 1):
            print(f"{i}. Q: {q}")
            print(f"   A: {a[:100]}...")
            print()
        
        return True
        
    except Exception as e:
        print(f"❌ 添加泰语数据失败: {e}")
        return False
    finally:
        if conn:
            conn.close()

if __name__ == "__main__":
    print("🇹🇭 开始添加泰语问答数据...")
    success = add_thai_qa_data()
    if success:
        print("🎉 泰语数据添加完成！")
    else:
        print("💥 泰语数据添加失败！")