#!/usr/bin/env python3
# test_app.py - 測試 DocumentAI 與 Supabase RPC 整合
import os
import logging
from pprint import pprint

logging.basicConfig(level=logging.INFO)

try:
    from app import DocumentAI
except Exception as e:
    print("❌ 無法匯入 DocumentAI:", e)
    exit(1)

def main():
    print("🚀 測試 DocumentAI 初始化與 RPC 呼叫")
    system = DocumentAI()

    # 測試 Supabase 連線
    print("\n[1] Supabase 連線測試")
    conn = system.test_supabase_connection()
    pprint(conn)

    # 測試嵌入模型
    print("\n[2] 嵌入模型資訊")
    try:
        info = system.embedding_service.get_model_info()
        pprint(info)
    except Exception as e:
        print("❌ 無法取得嵌入模型資訊:", e)

    # 測試 RPC 查詢
    print("\n[3] RPC 查詢測試")
    try:
        query = "這是一段測試內容"
        results = system._search_similar_in_supabase(query, limit=3)
        if results:
            print("✅ RPC 查詢成功，回傳結果:")
            pprint(results)
        else:
            print("⚠️ RPC 查詢成功但無匹配結果")
    except Exception as e:
        print("❌ RPC 查詢失敗:", e)

if __name__ == "__main__":
    main()