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

# 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("나와 내 동년배들이 좋아하는 뉴진스에 대해서 알려줘"))

def extract_numbered_lines(text):
    # 정규 표현식 패턴: 숫자 + 점 + 공백으로 시작하고, 줄바꿈 문자가 나올 때까지의 내용을 찾음
    pattern = r'\d\.\s.+?(?=\n|$)'
    matches = re.findall(pattern, text)

    return "\n".join(matches)

def bard_img(image_url):
    
    # 이미지 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)
    
    contents = extract_numbered_lines(bard_answer['content'])
    
    print("이미지 분석 결과입니다")
    print(contents)
    
    result = extract_numbered_lines(contents)
    print("분석 결과 파싱 데이터입니다")
    print(result)

    

    return result