Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import openai
 import os
 
 # 제거할 모델들을 MODELS 사전에서 제외
@@ -126,6 +127,41 @@ def cohere_respond(
         chat_history.append((message, error_message))
         return chat_history
 
+def chatgpt_respond(
+    message,
+    chat_history,
+    system_message,
+    max_tokens,
+    temperature,
+    top_p,
+):
+    openai.api_key = os.getenv("OPENAI_API_KEY")
+    if not openai.api_key:
+        chat_history.append((message, "OPENAI_API_KEY 환경 변수가 필요합니다."))
+        return chat_history
+
+    messages = [{"role": "system", "content": system_message}]
+    for human, assistant in chat_history:
+        messages.append({"role": "user", "content": human})
+        messages.append({"role": "assistant", "content": assistant})
+    messages.append({"role": "user", "content": message})
+
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4o-mini",  # 또는 다른 모델 ID 사용
+            messages=messages,
+            max_tokens=max_tokens,
+            temperature=temperature,
+            top_p=top_p,
+        )
+        assistant_message = response.choices[0].message['content']
+        chat_history.append((message, assistant_message))
+        return chat_history
+    except Exception as e:
+        error_message = f"오류가 발생했습니다: {str(e)}"
+        chat_history.append((message, error_message))
+        return chat_history
+
 def clear_conversation():
     return []
 
@@ -201,6 +237,44 @@ with gr.Blocks() as demo:
             cohere_chatbot
         )
         cohere_clear_button.click(clear_conversation, outputs=cohere_chatbot, queue=False)
+    
+    with gr.Tab("ChatGPT"):
+        with gr.Row():
+            chatgpt_system_message = gr.Textbox(
+                value="""반드시 한글로 답변할 것.
+너는 ChatGPT, OpenAI에서 개발한 언어 모델이다.
+내가 요구하는 것을 최대한 자세하고 정확하게 답변하라.
+""",
+                label="System Message",
+                lines=3
+            )
+            chatgpt_max_tokens = gr.Slider(minimum=1, maximum=4096, value=1024, step=1, label="Max Tokens")
+            chatgpt_temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
+            chatgpt_top_p = gr.Slider(
+                minimum=0.1,
+                maximum=1.0,
+                value=0.95,
+                step=0.05,
+                label="Top-P",
+            )
+        
+        chatgpt_chatbot = gr.Chatbot(height=600)
+        chatgpt_msg = gr.Textbox(label="메세지를 입력하세요")
+        with gr.Row():
+            chatgpt_submit_button = gr.Button("전송")
+            chatgpt_clear_button = gr.Button("대화 내역 지우기")
+        
+        chatgpt_msg.submit(
+            chatgpt_respond,
+            [chatgpt_msg, chatgpt_chatbot, chatgpt_system_message, chatgpt_max_tokens, chatgpt_temperature, chatgpt_top_p],
+            chatgpt_chatbot
+        )
+        chatgpt_submit_button.click(
+            chatgpt_respond,
+            [chatgpt_msg, chatgpt_chatbot, chatgpt_system_message, chatgpt_max_tokens, chatgpt_temperature, chatgpt_top_p],
+            chatgpt_chatbot
+        )
+        chatgpt_clear_button.click(clear_conversation, outputs=chatgpt_chatbot, queue=False)
 
 if __name__ == "__main__":
     demo.launch()