from ai.unlock_cookies import fetch_cookies
from bardapi.constants import SESSION_HEADERS
from bardapi import Bard
import requests

def get_cookies(domain_name):
    """특정 도메인의 쿠키를 가져오는 함수"""
    # unlock_cookies 함수를 호출하여 쿠키 파일의 잠금을 해제하고,
    # 특정 도메인에 대한 쿠키를 가져옵니다.
    cookies = fetch_cookies(domain_name)
    #cookies = new_fetch_cookies(domain_name)
    print(f"cookies = {cookies}")

    # 쿠키를 딕셔너리 형태로 변환합니다.
    cookies_dict = {}
    for cookie in cookies:
        cookies_dict[cookie.name] = cookie.value

    return cookies_dict

# bard.google.com 도메인의 쿠키를 가져옵니다.
cookies = get_cookies(".google.com")

# 필요한 쿠키를 추출합니다.
token1 = cookies.get("__Secure-1PSID")
print(f"token1 = {token1}")
token2 = cookies.get("__Secure-1PSIDTS")
print(f"token2 = {token2}")
token3 = cookies.get("__Secure-1PSIDCC")
print(f"token3 = {token3}")

session = requests.Session()
session.headers = SESSION_HEADERS
session.cookies.set("__Secure-1PSID", token1)
session.cookies.set("__Secure-1PSIDTS", token2)
session.cookies.set("__Secure-1PSIDCC", token3)


bard = Bard(token=token1, session=session)

#print(bard.get_answer("나와 내 동년배들이 좋아하는 뉴진스에 대해서 알려줘")['content'])
#print(bard.get_answer("나와 내 동년배들이 좋아하는 뉴진스에 대해서 알려줘"))


# 이미지 URL
image_url = 'https://file.percenty.co.kr/public/652bed8e865b1f32ea62bf1f/products/6575b191d39b6b71ccac77ce/a5ffe56b-3349-4dca-bc36-fa569a68c337.jpg'

# requests를 사용하여 이미지 다운로드
response = requests.get(image_url)
response.raise_for_status()  # HTTP 요청 에러 체크

# 이미지 데이터를 Bard에 전달
image_data = response.content
bard_answer = bard.ask_about_image('이미지의 상품을 한국의 쇼핑몰에서 판매할때 적절한 홍보문구를 3줄로 만들어줘.', image_data)
print(bard_answer['content'])