Update app.py

app.py

@@ -135,7 +135,8 @@ def respond_claude_qna(
     system_message: str,
     max_tokens: int,
     temperature: float,
-    top_p: float
+    top_p: float,
+    model_name: str  # 모델 이름 파라미터 추가
 ) -> str:
     """
     Claude API를 사용한 개선된 응답 생성 함수
@@ -147,9 +148,9 @@ def respond_claude_qna(
     try:
         client = anthropic.Anthropic(api_key=claude_api_key)
         
-        # 메시지 생성
+        # 선택된 모델 이름(model_name)을 사용
         message = client.messages.create(
-            model="claude-3-5-sonnet-20241022",
+            model=model_name,
             max_tokens=max_tokens,
             temperature=temperature,
             system=system_message,
@@ -278,7 +279,18 @@ with gr.Blocks() as demo:
     #################
     # Claude 탭
     #################
-    with gr.Tab("claude-3-haiku"):
+    with gr.Tab("Claude"):
+        # 라디오 버튼 추가: 모델 선택 (세 가지 옵션)
+        claude_model_radio = gr.Radio(
+            choices=[
+                "claude-3-haiku-20240307",
+                "claude-3-5-haiku-20241022",
+                "claude-3-5-sonnet-20241022"
+            ],
+            label="모델 선택",
+            value="claude-3-5-sonnet-20241022"  # 기본값 설정
+        )
+    
         claude_input1 = gr.Textbox(label="입력1", lines=1)
         claude_input2 = gr.Textbox(label="입력2", lines=1)
         claude_input3 = gr.Textbox(label="입력3", lines=1)
@@ -319,14 +331,15 @@ with gr.Blocks() as demo:
 
         claude_submit_button = gr.Button("전송")
 
-        def merge_and_call_claude(i1, i2, i3, i4, i5, sys_msg, mt, temp, top_p_):
+        def merge_and_call_claude(i1, i2, i3, i4, i5, sys_msg, mt, temp, top_p_, model_radio):
             question = " ".join([i1, i2, i3, i4, i5])
             return respond_claude_qna(
                 question=question,
                 system_message=sys_msg,
                 max_tokens=mt,
                 temperature=temp,
-                top_p=top_p_
+                top_p=top_p_,
+                model_name=model_radio  # 라디오 버튼으로 선택한 모델 이름 전달
             )
 
         claude_submit_button.click(
@@ -336,7 +349,8 @@ with gr.Blocks() as demo:
                 claude_system_message,
                 claude_max_tokens,
                 claude_temperature,
-                claude_top_p
+                claude_top_p,
+                claude_model_radio  # 추가된 라디오 버튼 입력
             ],
             outputs=claude_answer_output
         )
@@ -409,4 +423,4 @@ with gr.Blocks() as demo:
 # 메인 실행부
 #############################
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()