Update app.py

app.py

@@ -12,7 +12,6 @@ logger = logging.getLogger(__name__)
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 if GEMINI_API_KEY:
     logger.info("API 키가 환경변수에서 감지되었습니다.")
-    # 키의 일부만 표시 (보안을 위해)
     logger.info(f"API 키 미리보기: {GEMINI_API_KEY[:8]}...")
 else:
     logger.warning("GEMINI_API_KEY가 환경변수에서 감지되지 않았습니다.")
@@ -22,14 +21,12 @@ else:
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
     gr.Markdown(
         """
-        # ♊️ Gemini API 챗봇 (Secrets 사용)
+        # ♊️ Gemini API 챗봇
         Google Gemini API를 사용하는 챗봇입니다. 
         
         **중요**: Hugging Face Spaces의 **Settings → Repository secrets**에 `GEMINI_API_KEY`가 설정되어 있어야 합니다.
         
         [API 키 발급받기](https://aistudio.google.com/app/apikey)
-        
-        **디버깅 정보**:
         """
     )
     
@@ -41,8 +38,12 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
             interactive=False
         )
     
-    # Gradio 챗봇 UI 컴포넌트
-    chatbot = gr.Chatbot(label="Gemini 챗봇", height=600)
+    # Gradio 챗봇 UI 컴포넌트 - type 파라미터 추가
+    chatbot = gr.Chatbot(
+        label="Gemini 챗봇", 
+        height=600,
+        type="messages"  # 이 파라미터를 추가하여 경고 해결
+    )
 
     with gr.Row():
         # 사용자 메시지 입력란
@@ -50,6 +51,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
             label="메시지 입력",
             placeholder="무엇이든 물어보세요...",
             scale=7,
+            lines=1
         )
         # 전송 버튼
         submit_button = gr.Button("전송", variant="primary", scale=1)
@@ -57,15 +59,25 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
     with gr.Accordion("고급 설정", open=False):
         # LLM의 역할을 정의하는 시스템 메시지
         system_message = gr.Textbox(
-            value="You are a helpful and friendly chatbot.", label="시스템 메시지"
+            value="You are a helpful and friendly chatbot.", 
+            label="시스템 메시지",
+            lines=2
         )
         # 모델의 창의성을 조절하는 슬라이더
         temperature = gr.Slider(
-            minimum=0.0, maximum=1.0, value=0.7, step=0.1, label="Temperature"
+            minimum=0.0, 
+            maximum=1.0, 
+            value=0.7, 
+            step=0.1, 
+            label="Temperature"
         )
         # 생성할 최대 토큰 수를 조절하는 슬라이더
         max_tokens = gr.Slider(
-            minimum=1, maximum=4096, value=1024, step=1, label="Max new tokens"
+            minimum=1, 
+            maximum=4096, 
+            value=1024, 
+            step=1, 
+            label="Max new tokens"
         )
 
     # 환경변수 새로고침 버튼
@@ -116,8 +128,9 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
         
         # 사용할 모델과 시스템 프롬프트를 설정합니다.
         try:
+            # 사용 가능한 모델로 변경
             model = genai.GenerativeModel(
-                model_name='gemini-2.0-flash-exp',  # 또는 'gemini-1.5-flash'
+                model_name='gemini-2.0-flash',  # 안정적인 모델 사용
                 system_instruction=system_prompt
             )
         except Exception as e:
@@ -126,11 +139,23 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
 
         # Gradio의 대화 기록을 Gemini API가 이해할 수 있는 형식으로 변환합니다.
         gemini_history = []
-        for user_msg, model_msg in chat_history:
-            if user_msg:
-                gemini_history.append({"role": "user", "parts": [user_msg]})
-            if model_msg:
-                gemini_history.append({"role": "model", "parts": [model_msg]})
+        
+        # type="messages" 형식에 맞게 처리
+        if isinstance(chat_history, list) and len(chat_history) > 0:
+            # 새로운 메시지 형식 처리
+            if isinstance(chat_history[0], dict):
+                for msg in chat_history:
+                    if msg.get("role") == "user":
+                        gemini_history.append({"role": "user", "parts": [msg.get("content", "")]})
+                    elif msg.get("role") == "assistant":
+                        gemini_history.append({"role": "model", "parts": [msg.get("content", "")]})
+            # 이전 튜플 형식도 지원
+            else:
+                for user_msg, model_msg in chat_history:
+                    if user_msg:
+                        gemini_history.append({"role": "user", "parts": [user_msg]})
+                    if model_msg:
+                        gemini_history.append({"role": "model", "parts": [model_msg]})
             
         # 이전 대화 기록을 바탕으로 채팅 세션을 시작합니다.
         chat = model.start_chat(history=gemini_history)
@@ -170,14 +195,26 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
     def on_submit(message, chat_history, system_prompt, temp, max_output_tokens):
         if not message.strip():
             return "", chat_history
-            
-        chat_history.append((message, None))
-        bot_response_stream = respond(message, chat_history, system_prompt, temp, max_output_tokens)
+        
+        # 새로운 메시지 형식 사용
+        chat_history = chat_history or []
+        
+        # 사용자 메시지 추가
+        chat_history.append({"role": "user", "content": message})
+        
+        # 봇 응답 스트리밍
+        bot_response_stream = respond(message, chat_history[:-1], system_prompt, temp, max_output_tokens)
         
         for partial_response in bot_response_stream:
-            chat_history[-1] = (message, partial_response)
+            # 마지막 메시지가 사용자 메시지인 경우에만 봇 응답 추가
+            if chat_history and chat_history[-1]["role"] == "user":
+                chat_history.append({"role": "assistant", "content": partial_response})
+            else:
+                # 봇 응답 업데이트
+                chat_history[-1]["content"] = partial_response
             yield "", chat_history
     
+    # 이벤트 핸들러 연결
     msg.submit(
         on_submit, 
         [msg, chatbot, system_message, temperature, max_tokens], 
@@ -189,6 +226,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
         [msg, chatbot]
     )
 
+# 메인 실행 부분
 if __name__ == "__main__":
     # 디버깅 모드로 실행
     demo.launch(debug=True)