Update app.py

app.py

@@ -4,7 +4,7 @@ import openai
 import anthropic
 import os
 
-# Cohere Command R+ 모델 ID 정의
+# Cohere Command R+ 모델 ID 정의 (참고용 상수)
 COHERE_MODEL = "CohereForAI/c4ai-command-r-plus-08-2024"
 
 def get_client(model_name: str):
@@ -16,8 +16,10 @@ def get_client(model_name: str):
     if not hf_token:
         raise ValueError("HuggingFace API 토큰이 필요합니다. (환경변수 HF_TOKEN 미설정)")
 
-    if model_name == "Cohere Command R+":
-        model_id = COHERE_MODEL
+    if model_name == "c4ai-command-r-plus-08-2024":
+        model_id = "CohereForAI/c4ai-command-r-plus-08-2024"
+    elif model_name == "c4ai-command-r7b-12-2024":
+        model_id = "CohereForAI/c4ai-command-r7b-12-2024"
     else:
         raise ValueError("유효하지 않은 모델 이름입니다.")
     return InferenceClient(model_id, token=hf_token)
@@ -26,6 +28,7 @@ def cohere_respond(
     message,
     chat_history,
     system_message,
+    cohere_model_choice,
     max_tokens,
     temperature,
     top_p,
@@ -34,7 +37,7 @@ def cohere_respond(
     Cohere Command R+ 모델 응답 함수.
     HF 토큰은 함수 내부에서 os.environ을 통해 불러온다.
     """
-    model_name = "Cohere Command R+"
+    model_name = cohere_model_choice
     try:
         client = get_client(model_name)
     except ValueError as e:
@@ -223,7 +226,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,6 +237,11 @@ with gr.Blocks() as demo:
                 label="System Message",
                 lines=3
             )
+            cohere_model_choice = gr.Radio(
+                choices=["c4ai-command-r-plus-08-2024", "c4ai-command-r7b-12-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(
@@ -253,7 +261,8 @@ 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, 
             cohere_top_p