import sqlite3

# مسیر پایگاه داده
DATABASE_PATH = 'database.db'

# ایجاد یا بازنشانی جداول پایگاه داده
def create_db():
    conn = sqlite3.connect(DATABASE_PATH)
    c = conn.cursor()
    c.execute("""
        CREATE TABLE IF NOT EXISTS inputs (
            id INTEGER PRIMARY KEY,
            input_text TEXT,
            selected_model TEXT
        )
    """)
    c.execute("""
        CREATE TABLE IF NOT EXISTS chats (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            chat_id TEXT NOT NULL,
            username TEXT NOT NULL,
            user_message TEXT NOT NULL,
            ai_response TEXT NOT NULL
        )
    """)
    conn.commit()
    conn.close()

# درج چت در جدول chats
def insert_chat(chat_id, username, user_message, ai_response):
    try:
        conn = sqlite3.connect(DATABASE_PATH)
        cursor = conn.cursor()
        cursor.execute("""
            INSERT INTO chats (chat_id, username, user_message, ai_response)
            VALUES (?, ?, ?, ?)
        """, (str(chat_id), str(username), str(user_message), str(ai_response)))
        conn.commit()
    except sqlite3.Error as e:
        print(f"Error inserting chat: {e}")
    finally:
        conn.close()

# درج داده در جدول inputs
def insert_into_db(input_text, selected_model):
    try:
        conn = sqlite3.connect(DATABASE_PATH)
        c = conn.cursor()
        c.execute("""
            INSERT INTO inputs (input_text, selected_model)
            VALUES (?, ?)
        """, (str(input_text), str(selected_model)))
        conn.commit()
    except sqlite3.Error as e:
        print(f"Error inserting into inputs: {e}")
    finally:
        conn.close()

# پاک کردن داده‌های جدول inputs
def clear_database():
    try:
        conn = sqlite3.connect(DATABASE_PATH)
        c = conn.cursor()
        c.execute("DELETE FROM inputs")
        conn.commit()
    except sqlite3.Error as e:
        print(f"Error clearing database: {e}")
    finally:
        conn.close()

# بازیابی تمام ورودی‌ها از جدول inputs
def fetch_all_inputs():
    try:
        conn = sqlite3.connect(DATABASE_PATH)
        c = conn.cursor()
        c.execute("SELECT input_text, selected_model FROM inputs")
        results = c.fetchall()
        return results
    except sqlite3.Error as e:
        print(f"Error fetching inputs from database: {e}")
        return []
    finally:
        conn.close()

# بازیابی پیام‌ها و پاسخ‌های مرتبط با یک chat_id
def fetch_chats_by_id(chat_id):
    try:
        conn = sqlite3.connect(DATABASE_PATH)
        cursor = conn.cursor()
        cursor.execute("""
            SELECT user_message, ai_response FROM chats
            WHERE chat_id = ?
        """, (str(chat_id),))
        rows = cursor.fetchall()
        return rows
    except sqlite3.Error as e:
        print(f"Error fetching chats by ID: {e}")
        return []
    finally:
        conn.close()

# بازیابی chat_id ها برای یک کاربر خاص
def fetch_ids_by_user(username):
    try:
        conn = sqlite3.connect(DATABASE_PATH)
        cursor = conn.cursor()
        cursor.execute("""
            SELECT chat_id FROM chats
            WHERE username = ?
        """, (str(username),))
        rows = cursor.fetchall()
        return rows
    except sqlite3.Error as e:
        print(f"Error fetching chat IDs by username: {e}")
        return []
    finally:
        conn.close()

# حذف چت‌های مرتبط با یک کاربر خاص
def clear_chats_by_username(username):
    try:
        conn = sqlite3.connect(DATABASE_PATH)
        cursor = conn.cursor()
        cursor.execute("""
            DELETE FROM chats
            WHERE username = ?
        """, (str(username),))
        conn.commit()
    except sqlite3.Error as e:
        print(f"Error clearing chats by username: {e}")
    finally:
        conn.close()