[Python] 구글 바드 API 사용법

안녕하세요.

오늘은 간단하게 구글의 AI 챗봇 바드 API를 발급 받아 파이썬으로 활용하는 방법에 대해 알아보겠습니다.

 

※ 구글 바드에 대해 궁금하신 분들은 아래 포스팅 글을 참조하시기 바랍니다.

2023.04.20 - [Programming/Websites & API] - 구글 바드 사용법 및 예시

 

 파이썬으로 구글 바드 API 활용하기

▣ 바드 API 발급 받기

 - 먼저, 구글 크롬 환경에서 바드 사이트를 들어감

 - F12를 눌러 개발자 메뉴로 들어간 후, Application > Cookies > https://bard.google.com > __Secure-1PSID 에서 쿠키 값 복사 (아래 참조)

바드 F12 화면

 

▣ 파이썬 bardapi

pip install bardapi

 

 - 챗GPT API처럼 구글 바드에서 정식으로 나온 API 발급 방법이 아니며, bardapi 패키지에 대한 상세한 내용은 아래 사이트를 참고하면 됨

: https://github.com/dsdanielpark/Bard-API

GitHub - dsdanielpark/Bard-API: The unofficial python package that returns response of Google Bard through cookie value.

 

 - 바드 API를 따로 찾아서 발급 받기 귀찮다면 아래 코드를 활용하면 됨

from bardapi import Bard

bard = Bard(token_from_browser=True)
res = bard.get_answer("Do you like cookies?")
print(res['content'])

 

 - 단순히 채팅기능을 파이썬에서 활용하는 방법은 다음과 같음

from bardapi import Bard

token = 'Cookie Value'

bard = Bard(token=token)
print(bard.get_answer("코딩하는 금융인에 대해 아니?")['content'])

# chat_bard 함수 만들기
def chat_bard():
    while True:
        msg = input("User : ")
        
        if(msg == "exit"):
            break
        else:
            # 바드 대답
            response = bardapi.core.Bard().get_answer(msg)
            print(response['content'])
            
            # 선택형 답변을 모두 보여주기
            #for i, choice in enumerate(response['choices']):
            #    print(f"Choice {i+1}:\n", choice['content'][0], "\n")

if __name__ == '__main__':
    print("채팅 종료는 'exit' 입력하세요")
    chat_bard()