aliceblue11 commited on
Commit
0b103eb
·
verified ·
1 Parent(s): e8d0bc9

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import openai
3
+ import gradio as gr
4
+
5
+ # 환경 변수에서 API 키 가져오기
6
+ openai.api_key = os.getenv("OPENAI_API_KEY")
7
+
8
+ def call_api(content, system_message="You are a helpful assistant.", max_tokens=150, temperature=0.7, top_p=1.0):
9
+ response = openai.ChatCompletion.create(
10
+ model="gpt-4o-mini", # 또는 다른 모델 ID 사용
11
+ messages=[
12
+ {"role": "system", "content": system_message},
13
+ {"role": "user", "content": content},
14
+ ],
15
+ max_tokens=max_tokens,
16
+ temperature=temperature,
17
+ top_p=top_p,
18
+ )
19
+ return response.choices[0].message['content']
20
+
21
+ def translate_code_english_to_korean(code):
22
+ system_msg = "당신은 영어로 작성된 코드를 한국어로 정확하게 번역하는 도우미입니다."
23
+ user_msg = f"다음 코드를 한국어로 번역해 주세요:\n\n{code}"
24
+ translated_code = call_api(user_msg, system_message=system_msg, max_tokens=500)
25
+ return translated_code
26
+
27
+ iface = gr.Interface(
28
+ fn=translate_code_english_to_korean,
29
+ inputs=gr.inputs.Textbox(lines=10, label="영어 코드 입력"),
30
+ outputs=gr.outputs.Textbox(label="한국어 번역된 코드"),
31
+ title="영어 코드 → 한국어 번역기",
32
+ description="GPT-4o-mini 모델을 사용하여 영어로 작성된 코드를 한국어로 번역해줍니다."
33
+ )
34
+
35
+ if __name__ == "__main__":
36
+ iface.launch()