app.py

import gradio as gr
import google.generativeai as genai
import os

# API 키 설정
api_key = os.environ.get("GEMINI_API_KEY")
if api_key:
    genai.configure(api_key=api_key)

def chat(message, history):
    if not api_key:
        return "❌ API 키가 설정되지 않았습니다. HF Spaces Settings에서 GEMINI_API_KEY를 추가하세요."
    
    try:
        # Gemini 모델 초기화
        model = genai.GenerativeModel('gemini-2.0-flash')
        
        # 대화 기록 변환
        chat_history = []
        for human, assistant in history:
            if human:
                chat_history.append({"role": "user", "parts": [human]})
            if assistant:
                chat_history.append({"role": "model", "parts": [assistant]})
        
        # 채팅 세션 시작
        chat_session = model.start_chat(history=chat_history)
        
        # 응답 생성
        response = chat_session.send_message(message)
        return response.text
        
    except Exception as e:
        return f"❌ 오류 발생: {str(e)}"

# Gradio 인터페이스
demo = gr.ChatInterface(
    fn=chat,
    title="🤖 Gemini 챗봇",
    description="Google Gemini API를 사용한 간단한 챗봇입니다.",
    examples=["안녕하세요!", "오늘 날씨는 어때?", "파이썬에 대해 설명해줘"],
    retry_btn=None,
    undo_btn="이전 대화 삭제",
    clear_btn="전체 대화 삭제",
)

if __name__ == "__main__":
    demo.launch()