Spaces:
Running
Running
aliceblue11
commited on
Create app.py
Browse files
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()
|