Update app.py

app.py

@@ -99,7 +99,8 @@ def respond_deepseek_qna(
     system_message: str,
     max_tokens: int,
     temperature: float,
-    top_p: float
+    top_p: float,
+    model_name: str  # 모델 이름 추가
 ):
     """
     DeepSeek 모델을 이용해 한 번의 질문(question)에 대한 답변을 반환하는 함수.
@@ -118,7 +119,7 @@ def respond_deepseek_qna(
 
     try:
         response = openai.ChatCompletion.create(
-            model="deepseek-chat",
+            model=model_name,  # 선택된 모델 사용
             messages=messages,
             max_tokens=max_tokens,
             temperature=temperature,
@@ -344,6 +345,13 @@ with gr.Blocks() as demo:
     # DeepSeek 탭
     #################
     with gr.Tab("DeepSeek-V3"):
+        # 라디오 버튼 추가
+        deepseek_model_radio = gr.Radio(
+            choices=["V3 (deepseek-chat)", "R1 (deepseek-reasoner)"],  # 선택지
+            label="모델 선택",  # 라벨
+            value="V3 (deepseek-chat)"  # 기본값
+        )
+
         deepseek_input1 = gr.Textbox(label="입력1", lines=1)
         deepseek_input2 = gr.Textbox(label="입력2", lines=1)
         deepseek_input3 = gr.Textbox(label="입력3", lines=1)
@@ -367,14 +375,21 @@ with gr.Blocks() as demo:
 
         deepseek_submit_button = gr.Button("전송")
 
-        def merge_and_call_deepseek(i1, i2, i3, i4, i5, sys_msg, mt, temp, top_p_):
+        def merge_and_call_deepseek(i1, i2, i3, i4, i5, sys_msg, mt, temp, top_p_, model_radio):
+            # 라디오 버튼에서 선택된 모델 이름 추출
+            if model_radio == "V3 (deepseek-chat)":
+                model_name = "deepseek-chat"
+            else:
+                model_name = "deepseek-reasoner"
+
             question = " ".join([i1, i2, i3, i4, i5])
             return respond_deepseek_qna(
                 question=question,
                 system_message=sys_msg,
                 max_tokens=mt,
                 temperature=temp,
-                top_p=top_p_
+                top_p=top_p_,
+                model_name=model_name  # 선택된 모델 이름 전달
             )
 
         deepseek_submit_button.click(
@@ -384,7 +399,8 @@ with gr.Blocks() as demo:
                 deepseek_system_message,
                 deepseek_max_tokens,
                 deepseek_temperature,
-                deepseek_top_p
+                deepseek_top_p,
+                deepseek_model_radio  # 라디오 버튼 입력 추가
             ],
             outputs=deepseek_answer_output
         )
@@ -393,4 +409,4 @@ with gr.Blocks() as demo:
 # 메인 실행부
 #############################
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()