Update app.py

app.py

@@ -171,6 +171,39 @@ def respond_claude_qna(
     except Exception as e:
         return f"예상치 못한 오류가 발생했습니다: {str(e)}"
 
+def respond_o1mini_qna(
+    question: str,
+    system_message: str,
+    max_tokens: int,
+    temperature: float
+):
+    """
+    o1-mini 모델을 이용해 한 번의 질문(question)에 대한 답변을 반환하는 함수.
+    o1-mini에서는 advanced setting 중 top_p 옵션은 지원하지 않으므로, 해당 옵션은 제거함.
+    """
+    openai_token = os.getenv("OPENAI_TOKEN")
+    if not openai_token:
+        return "OpenAI API 토큰이 필요합니다."
+
+    openai.api_key = openai_token
+
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": question}
+    ]
+
+    try:
+        response = openai.ChatCompletion.create(
+            model="o1-mini",
+            messages=messages,
+            max_tokens=max_tokens,
+            temperature=temperature,
+        )
+        assistant_message = response.choices[0].message['content']
+        return assistant_message
+    except Exception as e:
+        return f"오류가 발생했습니다: {str(e)}"
+
 #############################
 # [기본코드] UI 부분 - 수정/삭제 불가
 #############################
@@ -419,6 +452,53 @@ with gr.Blocks() as demo:
             outputs=deepseek_answer_output
         )
 
+    #################
+    # 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에서 개발한 경량 언어 모델이다.
+내가 요구하는 것을 최대한 자세하고 정확하게 답변하라.
+""",
+                label="System Message",
+                lines=3
+            )
+            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
+            )
+
+        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
+        )
+
 #############################
 # 메인 실행부
 #############################