Update app.py

app.py

@@ -1,49 +1,60 @@
-# -*- coding: utf-8 -*-
-"""Untitled1.ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1huUmrgIbG1wwEKxs3zsCdRNir_IY5ICf
-"""
-
 import os
+import gradio as gr
+from openai.openai import OpenAI
 
-from openai import OpenAI
-import openai
-
+# OpenAI API 키 설정 (환경 변수에서 불러오기)
 openai_api_key = os.getenv("OPENAI_API_KEY")
 
-client = openai.OpenAI(api_key = openai_api_key)
-
-
-response = client.chat.completions.create(
-  model="gpt-4-turbo",
-  messages=[
-    {
-      "role": "system",
-      "content": "초등학교 6학년 수준 범위 내에서 학생의 질문에 친절하게 답하는 교사 역할, 설명은 구체적인 예시를 들어서, 어려운 단어는 별도로 설명, 사진이 있으면 사진도 함께 제시, 자료들의 출처를 제시"
-    },
-    {
-      "role": "user",
-      "content": ""
-    }
-  ],
-  temperature=1,
-  max_tokens=1000,
-  top_p=0.9,
-  frequency_penalty=0,
-  presence_penalty=0
-)
+# OpenAI API 클라이언트 생성
+client = OpenAI(api_key=openai_api_key)
+
+
+def openai_chat(question):
+    """
+    OpenAI ChatGPT를 사용하여 질문에 답변합니다.
+
+    Args:
+        question (str): 사용자가 입력한 질문 문장
+
+    Returns:
+        str: ChatGPT의 답변 문장
+    """
+    response = client.chat.completions.create(
+        model="gpt-4-turbo",
+        messages=[
+            {
+                "role": "system",
+                "content": "초등학교 6학년 수준 범위 내에서 학생의 질문에 친절하게 답하는 교사 역할, 설명은 구체적인 예시를 들어서, 어려운 단어는 별도로 설명, 사진이 있으면 사진도 함께 제시, 자료들의 출처를 제시"
+            },
+            {
+                "role": "user",
+                "content": question
+            }
+        ],
+        temperature=1,
+        max_tokens=1000,
+        top_p=0.9,
+        frequency_penalty=0,
+        presence_penalty=0
+    )
+
+    # ChatGPT 응답 추출
+    answer = response["choices"][0]["text"]
+
+    return answer
 
+with gr.Blocks() as demo:
+    # 질문 입력 텍스트 박스
+    Q = gr.Textbox(label="질문", placeholder="질문을 넣어주세요.")
 
-import gradio as gr
+    # 질문하기 버튼
+    btn = gr.Button("질문하기")
 
-with gr.Blocks() as demo:
-  Q = gr.Textbox(label="질문", placeholder="질문을 넣어주세요.")
-  btn = gr.Button("질문하기")
-  A = gr.TextArea(label="AI 선생님의 설명", placeholder="AI 선생님이 답변 중 입니다.")
+    # 답변 출력 텍스트 박스
+    A = gr.TextArea(label="AI 선생님의 설명", placeholder="AI 선생님이 답변 중 입니다.")
 
-  btn.click(openai_chat, inputs=Q, outputs=A)
+    # 버튼 클릭 시 질문 전송 및 답변 출력
+    btn.click(openai_chat, inputs=Q, outputs=A)
 
+# 웹 인터페이스 실행
 demo.launch()