Update app.py

app.py

@@ -1,39 +1,48 @@
 import gradio as gr
-from PIL import Image
+import os
+import openai
 
-def convert_to_grayscale(image):
-    # 이미지를 흑백으로 변환
-    grayscale_image = image.convert("L")
-    return grayscale_image
+# OpenAI API 키 설정
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def blend_images(original, grayscale, blend):
-    # 원본과 흑백 이미지 사이의 블렌드된 이미지를 반환
-    return Image.blend(original, grayscale, blend)
+def translate_code_to_korean(code):
+    """
+    영어 코드를 한국어로 번역하는 함수
+    """
+    system_message = (
+        "You are an AI language model specializing in translating programming-related texts from English to Korean. "
+        "Ensure that the translations are accurate and preserve technical meaning."
+    )
+    try:
+        # OpenAI API 호출
+        response = openai.ChatCompletion.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": system_message},
+                {"role": "user", "content": code},
+            ],
+            max_tokens=1500,
+            temperature=0.7,
+            top_p=0.9,
+        )
+        return response.choices[0].message['content']
+    except Exception as e:
+        return f"에러가 발생했습니다: {str(e)}"
 
-# 그라디오 인터페이스 구성
-def create_interface():
-    with gr.Blocks() as demo:
-        gr.Markdown("# 흑백 사진 변환기\n일반 사진을 흑백 사진으로 변환하고 슬라이드바를 통해 원본과 비교해보세요.")
+# Gradio 인터페이스 설정
+def interface():
+    with gr.Blocks() as app:
+        gr.Markdown("# 코드 영어-한국어 번역기\n영어로 작성된 코드를 입력하면 한국어로 번역됩니다.")
         
         with gr.Row():
-            with gr.Column():
-                image_input = gr.Image(type="pil", label="원본 이미지 입력")
-                blend_slider = gr.Slider(0, 1, value=0.5, label="슬라이드바 (0: 원본, 1: 흑백)")
-            
-            with gr.Column():
-                image_output = gr.Image(type="pil", label="결과 이미지 출력")
+            input_code = gr.Textbox(label="영어 코드 입력", lines=10, placeholder="여기에 코드를 입력하세요...")
+            output_translation = gr.Textbox(label="한국어 번역 결과", lines=10, interactive=False)
         
-        def update_output(image, blend):
-            grayscale = convert_to_grayscale(image)
-            blended_image = blend_images(image, grayscale, blend)
-            return blended_image
-
-        image_input.change(fn=update_output, inputs=[image_input, blend_slider], outputs=image_output)
-        blend_slider.change(fn=update_output, inputs=[image_input, blend_slider], outputs=image_output)
-
-    return demo
+        translate_button = gr.Button("번역 실행")
+        translate_button.click(translate_code_to_korean, inputs=[input_code], outputs=[output_translation])
+        
+    return app
 
-# 인터페이스 실행
 if __name__ == "__main__":
-    demo = create_interface()
-    demo.launch()
+    app = interface()
+    app.launch()