Update app.py

app.py

@@ -1,49 +1,43 @@
 import os
+import openai
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# 환경 변수에서 Hugging Face 토큰을 가져옵니다.
-HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
+# OpenAI API 키를 환경 변수에서 가져오기
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-# 모델과 토크나이저를 로드합니다.
-model_id = "CohereForAI/c4ai-command-r-plus-08-2024"
-tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=HUGGINGFACE_TOKEN)
-model = AutoModelForCausalLM.from_pretrained(model_id, use_auth_token=HUGGINGFACE_TOKEN)
-
-def translate_code(english_code):
+def translate_to_korean(english_text):
     """
-    영어 코드를 한국어로 번역하는 함수입니다.
+    입력된 영어 텍스트를 한국어로 번역하는 함수
     """
-    # 메시지 포맷팅
-    messages = [{"role": "user", "content": english_code}]
-    input_ids = tokenizer.apply_chat_template(
-        messages, 
-        tokenize=True, 
-        add_generation_prompt=True, 
-        return_tensors="pt"
-    )
-    
-    # 텍스트 생성
-    gen_tokens = model.generate(
-        input_ids, 
-        max_new_tokens=200, 
-        do_sample=True, 
-        temperature=0.3,
-    )
-    
-    # 생성된 텍스트 디코딩
-    gen_text = tokenizer.decode(gen_tokens[0], skip_special_tokens=True)
-    
-    # 번역된 한국어 텍스트 반환
-    return gen_text
+    system_message = "You are a helpful assistant that translates English text to Korean."
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4o-mini",  # 모델 ID
+            messages=[
+                {"role": "system", "content": system_message},
+                {"role": "user", "content": english_text},
+            ],
+            max_tokens=1000,      # 필요한 경우 조정
+            temperature=0.3,      # 창의성 조절
+            top_p=1.0,            # 확률 분포 조절
+        )
+        korean_text = response.choices[0].message['content'].strip()
+        return korean_text
+    except Exception as e:
+        return f"오류가 발생했습니다: {str(e)}"
 
-# Gradio 인터페이스 설정
+# Gradio 인터페이스 구성
 iface = gr.Interface(
-    fn=translate_code,
-    inputs=gr.inputs.Textbox(lines=10, placeholder="영어 코드를 입력하세요..."),
+    fn=translate_to_korean,
+    inputs=gr.inputs.Textbox(lines=10, placeholder="번역할 영어 텍스트를 입력하세요..."),
     outputs=gr.outputs.Textbox(),
-    title="코드 번역기",
-    description="영어로 작성된 코드를 한국어로 번역해드립니다."
+    title="영어-한국어 번역기",
+    description="GPT-4o-mini 모델을 사용하여 영어 텍스트를 한국어로 번역합니다.",
+    examples=[
+        ["Hello, how are you?"],
+        ["This is a test sentence for translation."],
+        ["OpenAI provides powerful language models."]
+    ],
 )
 
 if __name__ == "__main__":