Update app.py

app.py

@@ -4,9 +4,6 @@ import openai
 import anthropic
 import os
 
-# Cohere 모델 ID 참고용 상수 (이전 모델은 사용하지 않습니다)
-# COHERE_MODEL = "CohereForAI/c4ai-command-r-plus-08-2024"
-
 def get_client(model_name: str):
     """
     모델 이름에 맞춰 InferenceClient 생성.
@@ -16,7 +13,9 @@ def get_client(model_name: str):
     if not hf_token:
         raise ValueError("HuggingFace API 토큰이 필요합니다. (환경변수 HF_TOKEN 미설정)")
 
-    if model_name == "c4ai-command-r-08-2024":
+    if model_name == "c4ai-command-r-plus-08-2024":
+        model_id = "CohereForAI/c4ai-command-r-plus-08-2024"
+    elif model_name == "c4ai-command-r-08-2024":
         model_id = "CohereForAI/c4ai-command-r-08-2024"
     else:
         raise ValueError("유효하지 않은 모델 이름입니다.")
@@ -32,7 +31,7 @@ def cohere_respond(
     top_p,
 ):
     """
-    Cohere 모델 응답 함수.
+    Cohere Command R+ 모델 응답 함수.
     HF 토큰은 함수 내부에서 os.environ을 통해 불러온다.
     """
     model_name = cohere_model_choice
@@ -224,7 +223,7 @@ with gr.Blocks() as demo:
     # 일반 모델 관련 UI/기능 제거 (요청 사항에 따라 삭제)
     # --------------------------------------------------
 
-    # Cohere Command R+ 탭 (모델 선택 수정됨)
+    # Cohere Command R+ 탭 (모델 선택 추가됨)
     with gr.Tab("Cohere Command R+"):
         with gr.Row():
             cohere_system_message = gr.Textbox(
@@ -236,8 +235,8 @@ with gr.Blocks() as demo:
                 lines=3
             )
             cohere_model_choice = gr.Radio(
-                choices=["c4ai-command-r-08-2024"],
-                value="c4ai-command-r-08-2024",
+                choices=["c4ai-command-r-plus-08-2024", "c4ai-command-r-08-2024"],
+                value="c4ai-command-r-plus-08-2024",
                 label="모델 선택"
             )
             cohere_max_tokens = gr.Slider(minimum=100, maximum=8000, value=2000, step=100, label="Max new tokens")
@@ -259,7 +258,7 @@ with gr.Blocks() as demo:
         inputs_for_cohere = [
             cohere_msg, 
             cohere_chatbot, 
-            cohere_system_message,
+            cohere_system_message, 
             cohere_model_choice,
             cohere_max_tokens, 
             cohere_temperature, 
@@ -269,7 +268,7 @@ with gr.Blocks() as demo:
         cohere_submit_button.click(cohere_respond, inputs_for_cohere, cohere_chatbot)
         cohere_clear_button.click(clear_conversation, outputs=cohere_chatbot, queue=False)
 
-    # ChatGPT
+    # ChatGPT 탭
     with gr.Tab("ChatGPT"):
         with gr.Row():
             chatgpt_system_message = gr.Textbox(
@@ -308,7 +307,7 @@ with gr.Blocks() as demo:
         chatgpt_submit_button.click(chatgpt_respond, inputs_for_chatgpt, chatgpt_chatbot)
         chatgpt_clear_button.click(clear_conversation, outputs=chatgpt_chatbot, queue=False)
 
-    # Claude
+    # Claude 탭
     with gr.Tab("Claude"):
         with gr.Row():
             claude_system_message = gr.Textbox(
@@ -347,7 +346,7 @@ with gr.Blocks() as demo:
         claude_submit_button.click(claude_respond, inputs_for_claude, claude_chatbot)
         claude_clear_button.click(clear_conversation, outputs=claude_chatbot, queue=False)
 
-    # DeepSeek
+    # DeepSeek 탭
     with gr.Tab("DeepSeek"):
         with gr.Row():
             deepseek_system_message = gr.Textbox(
@@ -388,7 +387,6 @@ with gr.Blocks() as demo:
             deepseek_temperature,
             deepseek_top_p
         ]
-        # Textbox.submit에서는 stream 인자를 제거합니다.
         deepseek_msg.submit(deepseek_respond, inputs_for_deepseek, deepseek_chatbot)
         deepseek_submit_button.click(deepseek_respond, inputs_for_deepseek, deepseek_chatbot)
         deepseek_clear_button.click(clear_conversation, outputs=deepseek_chatbot, queue=False)