Update app.py

app.py

@@ -3,11 +3,6 @@ import gradio as gr
 import google.generativeai as genai
 import os
 
-# --- Hugging Face Spaces의 Secrets 또는 로컬 환경변수에서 API 키를 가져옵니다. ---
-# 이 앱을 Spaces에 배포할 때는 'Settings' -> 'Repository secrets'에
-# 'GEMINI_API_KEY'라는 이름으로 키를 저장해야 합니다.
-GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
-
 # --- UI 및 챗봇 설명 ---
 # Gradio Blocks를 사용하여 좀 더 유연한 UI를 구성합니다.
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
@@ -20,16 +15,11 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
         """
     )
     
-    # API 키가 설정되었는지 확인하고 안내 메시지를 표시합니다.
-    if not GEMINI_API_KEY:
-        gr.Warning("⚠️ Gemini API 키가 설정되지 않았습니다. Hugging Face Spaces의 'Repository secrets'에 GEMINI_API_KEY를 추가해주세요.")
-
     # Gradio 챗봇 UI 컴포넌트
     chatbot = gr.Chatbot(label="Gemini 챗봇", height=600)
 
     with gr.Row():
         # 사용자 메시지 입력란
-        # container=False 속성을 제거하여 Gradio 내부 오류를 수정합니다.
         msg = gr.Textbox(
             label="메시지 입력",
             placeholder="무엇이든 물어보세요...",
@@ -54,9 +44,15 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
 
     # --- Gemini API 호출 함수 ---
     def respond(message, chat_history, system_prompt, temp, max_output_tokens):
+        # 함수가 호출될 때마다 환경변수에서 API 키를 직접 가져옵니다.
+        # 이렇게 하면 앱 시작 시점에 키를 못 불러오는 문제를 해결할 수 있습니다.
+        GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+
         # 환경변수에서 가져온 API 키가 없으면 안내 메시지를 띄웁니다.
         if not GEMINI_API_KEY:
-            yield "Google API 키가 설정되지 않았습니다. 관리자에게 문의하세요."
+            # UI에 직접 경고를 표시하기 위해 gr.Warning을 사용할 수 있지만,
+            # 여기서는 채팅 응답으로 처리합니다.
+            yield "⚠️ **오류**: `GEMINI_API_KEY`가 설정되지 않았습니다.\n\nHugging Face Spaces의 **Settings > Repository secrets**에 API 키를 추가했는지 확인해주세요."
             return
 
         try:
@@ -109,20 +105,14 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
             yield f"응답 생성 중 오류가 발생했습니다: {e}"
 
     # --- Gradio 이벤트 리스너 ---
-    # 사용자가 메시지를 입력하고 '전송' 버튼을 누르거나 엔터를 쳤을 때 실행될 로직
     def on_submit(message, chat_history, system_prompt, temp, max_output_tokens):
-        # 대화 기록에 사용자 메시지를 추가합니다.
         chat_history.append((message, None))
-        # 스트리밍 응답을 위해 빈 문자열로 응답을 시작합니다.
         bot_response_stream = respond(message, chat_history, system_prompt, temp, max_output_tokens)
         
-        # 스트리밍 응답을 UI에 업데이트합니다.
         for partial_response in bot_response_stream:
             chat_history[-1] = (message, partial_response)
             yield "", chat_history
     
-    # 메시지 전송(버튼 클릭 또는 엔터) 시 on_submit 함수를 호출합니다.
-    # UI에서 API 키 입력란이 제거되었으므로, 입력(inputs) 리스트에서도 제외합니다.
     msg.submit(
         on_submit, 
         [msg, chatbot, system_message, temperature, max_tokens], 
@@ -134,6 +124,5 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
         [msg, chatbot]
     )
 
-# 스크립트가 직접 실행될 때 Gradio 앱을 실행합니다.
 if __name__ == "__main__":
     demo.launch(debug=True)