콘텐츠로 이동

구글 Gemini 챗봇 실행하기

필요한 모듈 설치하기

pip installgoogle-generativeai

가장 간단한 실행 예제

import google.generativeai as genai

GOOGLE_API_KEY="your key"

genai.configure(api_key=GOOGLE_API_KEY)

model = genai.GenerativeModel('gemini-pro')

response = model.generate_content("중세시대 배경에서 의사의 삶에 대해 작성해줘.")

print(response.text)

Gemini 챗봇에 필요한 모듈 설치하기

!pip install -q google-generativeai gradio

구글 Gemini 챗봇 실행하기

import gradio as gr
import google.generativeai as genai

# 1. API 키 설정
GOOGLE_API_KEY = "여기에_당신의_API_KEY_입력"
genai.configure(api_key=GOOGLE_API_KEY)

# 2. Gemini 모델 초기화
model = genai.GenerativeModel('gemini-pro')

# 3. 사용자 입력에 대한 응답 함수
def chat_with_gemini(user_input):
    try:
        response = model.generate_content(user_input)
        return response.text
    except Exception as e:
        return f"에러 발생: {e}"

# 4. Gradio 인터페이스 생성
demo = gr.Interface(
    fn=chat_with_gemini,
    inputs=gr.Textbox(lines=2, placeholder="메시지를 입력하세요..."),
    outputs="text",
    title="Gemini 챗봇",
    description="Google Gemini API를 활용한 간단한 챗봇 데모입니다."
)

# 5. 실행
demo.launch()