import os
import openai
import gradio as gr

# 환경 변수에서 API 키 가져오기
openai.api_key = os.getenv("OPENAI_API_KEY")

def call_api(content, system_message="You are a helpful assistant.", max_tokens=2000, temperature=0.7, top_p=1.0):
    response = openai.ChatCompletion.create(
        model="gpt-4o-mini",  # 또는 다른 모델 ID 사용
        messages=[
            {"role": "system", "content": system_message},
            {"role": "user", "content": content},
        ],
        max_tokens=max_tokens,
        temperature=temperature,
        top_p=top_p,
    )
    return response.choices[0].message['content']

def translate_code_english_to_korean(code):
    system_msg = "당신은 영어로 작성된 코드를 한국어로 정확하게 번역하는 도우미입니다."
    user_msg = f"다음 코드를 한국어로 번역해 주세요:\n\n{code}"
    translated_code = call_api(user_msg, system_message=system_msg, max_tokens=500)
    return translated_code

iface = gr.Interface(
    fn=translate_code_english_to_korean,
    inputs=gr.Textbox(lines=10, label="영어 코드 입력"),
    outputs=gr.Textbox(label="한국어 번역된 코드"),
    title="영어 코드 → 한국어 번역기",
    description="GPT-4o-mini 모델을 사용하여 영어로 작성된 코드를 한국어로 번역해줍니다."
)

if __name__ == "__main__":
    iface.launch()