main.py

import chainlit as cl
import os
from dotenv import load_dotenv
import requests

# .env 파일을 로드합니다.
load_dotenv()

# 환경 변수에서 Gemini API 키를 가져옵니다.
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")

# 최신 Gemini API 엔드포인트 및 모델명
GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent"

@cl.on_message
async def main(message: cl.Message):
    user_input = message.content

    if not GEMINI_API_KEY:
        response = "GEMINI_API_KEY가 설정되지 않았습니다."
        await cl.Message(content=response).send()
        return

    headers = {
        "Content-Type": "application/json"
    }
    params = {
        "key": GEMINI_API_KEY
    }
    data = {
        "contents": [
            {
                "parts": [
                    {"text": user_input}
                ]
            }
        ]
    }

    try:
        resp = requests.post(GEMINI_API_URL, headers=headers, params=params, json=data, timeout=30)
        resp.raise_for_status()
        result = resp.json()
        llm_response = result["candidates"][0]["content"]["parts"][0]["text"]
        response = f"Gemini 응답: {llm_response}"
    except Exception as e:
        response = f"Gemini API 호출 중 오류 발생: {e}"

    await cl.Message(content=response).send()