Update app.py

app.py

@@ -175,13 +175,14 @@ def respond_o1mini_qna(
     question: str,
     system_message: str,
     max_tokens: int,
-    temperature: float  # 입력값은 있으나 사용하지 않습니다.
+    temperature: float
 ):
     """
     o1-mini 모델을 이용해 한 번의 질문(question)에 대한 답변을 반환하는 함수.
     o1-mini에서는 'system' 역할 메시지를 지원하지 않으므로,
     system_message와 question을 하나의 'user' 메시지로 합쳐 전달합니다.
-    또한, 온도(temperature)는 고정값 1만 지원하므로 해당 값으로 호출합니다.
+    또한, o1-mini에서는 'max_tokens' 대신 'max_completion_tokens' 파라미터를 사용하며,
+    온도(temperature)는 고정값 1만 지원합니다.
     """
     openai_token = os.getenv("OPENAI_TOKEN")
     if not openai_token:
@@ -189,7 +190,6 @@ def respond_o1mini_qna(
 
     openai.api_key = openai_token
 
-    # system_message와 question을 하나의 메시지로 합침
     combined_message = f"{system_message}\n\n{question}"
     messages = [
         {"role": "user", "content": combined_message}
@@ -208,7 +208,7 @@ def respond_o1mini_qna(
         return f"오류가 발생했습니다: {str(e)}"
 
 #############################
-# [기본코드] UI 부분 - 수정/삭제 불가
+# [기본코드] UI 부분 - 수정/삭제 불가 (단, OpenAI 탭 추가)
 #############################
 
 with gr.Blocks() as demo:
@@ -456,50 +456,111 @@ with gr.Blocks() as demo:
         )
 
     #################
-    # o1-mini 탭
+    # OpenAI 탭 (gpt-4o-mini / o1-mini 통합)
     #################
-    with gr.Tab("o1-mini"):
-        o1mini_input1 = gr.Textbox(label="입력1", lines=1)
-        o1mini_input2 = gr.Textbox(label="입력2", lines=1)
-        o1mini_input3 = gr.Textbox(label="입력3", lines=1)
-        o1mini_input4 = gr.Textbox(label="입력4", lines=1)
-        o1mini_input5 = gr.Textbox(label="입력5", lines=1)
-
-        o1mini_answer_output = gr.Textbox(label="결과", lines=5, interactive=False)
-
-        with gr.Accordion("고급 설정 (o1-mini)", open=False):
-            o1mini_system_message = gr.Textbox(
-                value="""반드시 한글로 답변할 것.
-너는 o1-mini, OpenAI에서 개발한 경량 언어 모델이다.
+    with gr.Tab("OpenAI"):
+        # 모델 선택 라디오 버튼 (gpt-4o-mini와 o1-mini)
+        openai_model_radio = gr.Radio(
+            choices=["gpt-4o-mini", "o1-mini"],
+            label="모델 선택",
+            value="gpt-4o-mini"
+        )
+        
+        # gpt-4o-mini 전용 UI 그룹 (초기 visible)
+        with gr.Column(visible=True) as chatgpt_ui:
+            chatgpt_input1_o = gr.Textbox(label="입력1", lines=1)
+            chatgpt_input2_o = gr.Textbox(label="입력2", lines=1)
+            chatgpt_input3_o = gr.Textbox(label="입력3", lines=1)
+            chatgpt_input4_o = gr.Textbox(label="입력4", lines=1)
+            chatgpt_input5_o = gr.Textbox(label="입력5", lines=1)
+            chatgpt_answer_output_o = gr.Textbox(label="결과", lines=5, interactive=False)
+            with gr.Accordion("고급 설정 (gpt-4o-mini)", open=False):
+                chatgpt_system_message_o = gr.Textbox(
+                    value="""반드시 한글로 답변할 것.
+너는 ChatGPT, OpenAI에서 개발한 언어 모델이다.
 내가 요구하는 것을 최대한 자세하고 정확하게 답변하라.
 """,
-                label="System Message",
-                lines=3
+                    label="System Message",
+                    lines=3
+                )
+                chatgpt_max_tokens_o = gr.Slider(minimum=100, maximum=4000, value=2000, step=100, label="Max Tokens")
+                chatgpt_temperature_o = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
+                chatgpt_top_p_o = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
+            chatgpt_submit_button_o = gr.Button("전송")
+            
+            def merge_and_call_chatgpt_o(i1, i2, i3, i4, i5, sys_msg, mt, temp, top_p_):
+                question = " ".join([i1, i2, i3, i4, i5])
+                return respond_chatgpt_qna(
+                    question=question,
+                    system_message=sys_msg,
+                    max_tokens=mt,
+                    temperature=temp,
+                    top_p=top_p_
+                )
+            chatgpt_submit_button_o.click(
+                fn=merge_and_call_chatgpt_o,
+                inputs=[
+                    chatgpt_input1_o, chatgpt_input2_o, chatgpt_input3_o, chatgpt_input4_o, chatgpt_input5_o,
+                    chatgpt_system_message_o,
+                    chatgpt_max_tokens_o,
+                    chatgpt_temperature_o,
+                    chatgpt_top_p_o
+                ],
+                outputs=chatgpt_answer_output_o
             )
-            o1mini_max_tokens = gr.Slider(minimum=100, maximum=4000, value=2000, step=100, label="Max Tokens")
-            o1mini_temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
-            # o1-mini는 top_p 지원이 없으므로 해당 옵션은 UI에서 제외함
-
-        o1mini_submit_button = gr.Button("전송")
-
-        def merge_and_call_o1mini(i1, i2, i3, i4, i5, sys_msg, mt, temp):
-            question = " ".join([i1, i2, i3, i4, i5])
-            return respond_o1mini_qna(
-                question=question,
-                system_message=sys_msg,
-                max_tokens=mt,
-                temperature=temp
+        
+        # o1-mini 전용 UI 그룹 (초기 hidden)
+        with gr.Column(visible=False) as o1mini_ui:
+            o1mini_input1_o = gr.Textbox(label="입력1", lines=1)
+            o1mini_input2_o = gr.Textbox(label="입력2", lines=1)
+            o1mini_input3_o = gr.Textbox(label="입력3", lines=1)
+            o1mini_input4_o = gr.Textbox(label="입력4", lines=1)
+            o1mini_input5_o = gr.Textbox(label="입력5", lines=1)
+            o1mini_answer_output_o = gr.Textbox(label="결과", lines=5, interactive=False)
+            with gr.Accordion("고급 설정 (o1-mini)", open=False):
+                o1mini_system_message_o = gr.Textbox(
+                    value="""반드시 한글로 답변할 것.
+너는 o1-mini, OpenAI에서 개발한 경량 언어 모델이다.
+내가 요구하는 것을 최대한 자세하고 정확하게 답변하라.
+""",
+                    label="System Message",
+                    lines=3
+                )
+                o1mini_max_tokens_o = gr.Slider(minimum=100, maximum=4000, value=2000, step=100, label="Max Tokens")
+                o1mini_temperature_o = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
+                # o1-mini는 top_p 지원이 없으므로 해당 옵션은 UI에서 제외함
+            o1mini_submit_button_o = gr.Button("전송")
+            
+            def merge_and_call_o1mini_o(i1, i2, i3, i4, i5, sys_msg, mt, temp):
+                question = " ".join([i1, i2, i3, i4, i5])
+                return respond_o1mini_qna(
+                    question=question,
+                    system_message=sys_msg,
+                    max_tokens=mt,
+                    temperature=temp
+                )
+            o1mini_submit_button_o.click(
+                fn=merge_and_call_o1mini_o,
+                inputs=[
+                    o1mini_input1_o, o1mini_input2_o, o1mini_input3_o, o1mini_input4_o, o1mini_input5_o,
+                    o1mini_system_message_o,
+                    o1mini_max_tokens_o,
+                    o1mini_temperature_o
+                ],
+                outputs=o1mini_answer_output_o
             )
-
-        o1mini_submit_button.click(
-            fn=merge_and_call_o1mini,
-            inputs=[
-                o1mini_input1, o1mini_input2, o1mini_input3, o1mini_input4, o1mini_input5,
-                o1mini_system_message,
-                o1mini_max_tokens,
-                o1mini_temperature
-            ],
-            outputs=o1mini_answer_output
+        
+        # UI 업데이트: 라디오 버튼 선택에 따라 gpt-4o-mini / o1-mini UI 전환
+        def update_openai_ui(model_choice):
+            if model_choice == "gpt-4o-mini":
+                return gr.update(visible=True), gr.update(visible=False)
+            else:
+                return gr.update(visible=False), gr.update(visible=True)
+        
+        openai_model_radio.change(
+            fn=update_openai_ui,
+            inputs=openai_model_radio,
+            outputs=[chatgpt_ui, o1mini_ui]
         )
 
 #############################