Update app.py

app.py

@@ -1,32 +1,23 @@
 import gradio as gr
 import os
+import openai
 
-HF_TOKEN = os.getenv('HF_TOKEN')
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "dae-gu-ai-5th-preview")
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-dict = {
-    "ko2en" : {
-        "사과" : "apple",
-        "사자" : "tiger",
-        "사랑" : "love"
-    },
-    "en2ko" : {
-        "apple" : "사과",
-        "lion" : "사자",
-        "love" : "사랑"
-    }
-}
 def trans(type, word):
-    return dict[type][word]
+    msg = '아래 내용을 한국어에서 영어로 번역해줘\n'
+    if(type == 'en2ko') :
+        msg = '아래 내용을 영어에서 한국어로 번역해줘\n'
+    msg = msg + word
+    chat_completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": msg}])
+    return chat_completion.choices[0].message.content
 app = gr.Interface(
     fn=trans,
     inputs=[
         gr.Radio(['ko2en', 'en2ko']),
         gr.Textbox(placeholder="한국어")
     ],
-    outputs="text",
-    allow_flagging="auto", 
-    flagging_callback=hf_writer
+    outputs="text"
 )
 app.launch(debug=True)