Update app.py

app.py

@@ -4,6 +4,9 @@ import openai
 import anthropic
 import os
 
+# Cohere Command R+ 모델 ID 정의
+COHERE_MODEL = "CohereForAI/c4ai-command-r-plus-08-2024"
+
 def get_client(model_name: str):
     """
     모델 이름에 맞춰 InferenceClient 생성.
@@ -13,10 +16,8 @@ def get_client(model_name: str):
     if not hf_token:
         raise ValueError("HuggingFace API 토큰이 필요합니다. (환경변수 HF_TOKEN 미설정)")
 
-    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"
+    if model_name == "Cohere Command R+":
+        model_id = COHERE_MODEL
     else:
         raise ValueError("유효하지 않은 모델 이름입니다.")
     return InferenceClient(model_id, token=hf_token)
@@ -25,7 +26,6 @@ def cohere_respond(
     message,
     chat_history,
     system_message,
-    cohere_model_choice,
     max_tokens,
     temperature,
     top_p,
@@ -34,7 +34,7 @@ def cohere_respond(
     Cohere Command R+ 모델 응답 함수.
     HF 토큰은 함수 내부에서 os.environ을 통해 불러온다.
     """
-    model_name = cohere_model_choice
+    model_name = "Cohere Command R+"
     try:
         client = get_client(model_name)
     except ValueError as e:
@@ -223,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(
@@ -234,11 +234,6 @@ with gr.Blocks() as demo:
                 label="System Message",
                 lines=3
             )
-            cohere_model_choice = gr.Radio(
-                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")
             cohere_temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature")
             cohere_top_p = gr.Slider(
@@ -259,7 +254,6 @@ with gr.Blocks() as demo:
             cohere_msg, 
             cohere_chatbot, 
             cohere_system_message, 
-            cohere_model_choice,
             cohere_max_tokens, 
             cohere_temperature, 
             cohere_top_p
@@ -268,7 +262,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(
@@ -307,7 +301,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(
@@ -346,7 +340,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(
@@ -387,9 +381,10 @@ 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)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()