import re

from Brain.src.firebase.cloudmessage import CloudMessage
from Brain.src.firebase.firebase import initialize_app
from Brain.src.model.req_model import ReqModel
from Brain.src.rising_plugin.risingplugin import getCompletion, getTextFromImage
from Brain.src.rising_plugin.image_embedding import query_image_text

TEST_IAMGE_NAME = "0ddffe51-3763-48d9-ab74-2086de529217"
TEST_UUID = "TEST_UUID"
test_setting = ReqModel(
    data={
        "token": "test-token",
        "uuid": TEST_UUID,
        "openai_key": "",
        "pinecone_key": "",
        "pinecone_env": "",
        "firebase_key": "",
        "settings": {"temperature": 0.6},
    }
)

firebase_app = initialize_app(test_setting)


def test_langchain():
    error = "Error happened while analyzing your prompt. Please ask me again"
    result = getCompletion(
        query="open three.js website", setting=test_setting, firebase_app=firebase_app
    )
    print(result)
    assert result != error


def test_image2text():
    error = "Error happened while analyzing your prompt. Please ask me again"
    result = getTextFromImage(filename=TEST_IAMGE_NAME, firebase_app=firebase_app)
    print(result)
    assert result != error


def test_firebase_cloud_message():
    cloud_message = CloudMessage(firebase_app=firebase_app)
    token_list = cloud_message.get_tokens()
    notification = {"title": "alert", "content": "test"}

    pattern = r"send to \d+ devices, with \d+ successed, with \d+ failed."

    state, value = cloud_message.send_message(notification, token_list)
    assert re.match(pattern, value)


def test_query_image_text():
    error = "Error happened in querying image & text"
    result = query_image_text(
        image_content=getTextFromImage(
            filename=TEST_IAMGE_NAME, firebase_app=firebase_app
        ),
        message="",
        setting=test_setting,
    )
    print(result)
    assert result != error