npc_social_network/models/gemini_setup.py

# portfolio/npc_social_network/models/gemini_setup.py
import google.generativeai as genai
from dotenv import load_dotenv
import os


# def load_gemini():
#     load_dotenv() # .env 파일에서 환경 변수 로드

#     genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
#     model = genai.GenerativeModel("gemini-2.0-flash")  # Gemini Flash 사용

#     return model

def initialize_model(model_name: str, api_key: str):
    """ 주어진 모델과 API 키로 Gemini 모델을 동적으로 초기화"""
    try:
        # API 키를 설정
        genai.configure(api_key=api_key)
        # 지정된 모델을 생성
        model = genai.GenerativeModel(model_name)
        # 간단한 테스트 호출로 API 키 유효성 검증
        model.generate_content("hello", generation_config={"max_output_tokens": 1})
        print(f"✅ Gemini 모델 '{model_name}'이(가) 성공적으로 초기화되었습니다.")
        return model
    except Exception as e:
        print(f"❌ Gemini 모델 초기화 실패: {e}")
        return None