Update app.py

app.py

@@ -1,49 +1,69 @@
-# -*- coding: utf-8 -*-
-"""Untitled1.ipynb
-Automatically generated by Colab.
-Original file is located at
-    https://colab.research.google.com/drive/1huUmrgIbG1wwEKxs3zsCdRNir_IY5ICf
-"""
-
 import os
-
-from openai import OpenAI
+import time
 import openai
+import streamlit as st
+from streamlit_chat import message
 
+# OpenAI API 설정
 openai_api_key = os.getenv("OPENAI_API_KEY")
-
-client = openai.OpenAI(api_key = openai_api_key)
+openai.api_key = openai_api_key
 
 def openai_chat(text):
-    from openai import OpenAI
-
-    completion = client.chat.completions.create(
-      model="gpt-4o",
-      messages=[
-        {"role": "system", "content": "초등학교 6학년 수준 범위 내에서 학생의 질문에 친절하게 답하는 교사 역할, 설명은 구체적인 예시를 들어서, 어려운 단어는 별도로 설명해줘. 설명글의 분량은 1000~2000자 정도로 해줘."},
-        {"role": "user", "content": text}
-      ],
-  temperature=1,
-  max_tokens=2000,
-  top_p=0.9,
-  frequency_penalty=0,
-  presence_penalty=0
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "초등학교 6학년 수준 범위 내에서 학생의 질문에 친절하게 답하는 교사 역할, 설명은 구체적인 예시를 들어서, 어려운 단어는 별도로 설명해줘. 설명글의 분량은 1000~2000자 정도로 해줘."},
+                {"role": "user", "content": text}
+            ],
+            temperature=1,
+            max_tokens=2000,
+            top_p=0.9,
+            frequency_penalty=0,
+            presence_penalty=0
+        )
+        return response.choices[0].message["content"]
+    except Exception as e:
+        return f"에러 발생: {str(e)}"
+
+# Streamlit 앱 설정
+st.set_page_config(page_title="친절한 AI 선생님(GPT-4)", layout="centered")
+
+st.markdown(
+    """
+    <div style='font-size: 30px; font-weight: bold; text-align: center;'>친절한 AI 선생님(GPT-4)</div>
+    <div style='text-align: center; margin-bottom: 20px;'>궁금한 것은 무엇이든 물어보세요. AI 선생님이 친절하게 알려줄 거예요!</div>
+    """,
+    unsafe_allow_html=True
 )
 
-    return completion.choices[0].message.content
+# 채팅 히스토리 관리
+if "history" not in st.session_state:
+    st.session_state.history = []
 
-openai_chat("한글은 누가 만들었어?")
+# 사용자 입력 받기
+user_input = st.text_input(
+    "질문", key="user_input", placeholder="질문을 넣어주세요.",
+    help="AI 선생님에게 궁금한 것을 물어보세요."
+)
 
-import gradio as gr
+def respond(user_input):
+    if user_input:
+        st.session_state.history.append({"user": user_input, "ai": ""})
+        response = openai_chat(user_input)
+        st.session_state.history[-1]["ai"] = response
 
-with gr.Blocks() as demo:
-  gr.Markdown("<div style='font-size: 30px; font-weight: bold;'>친절한 AI 선생님(GPT-4o)</div>")
-  gr.Markdown("궁금한 것은 무엇이든 물어보세요. AI 선생님이 친절하게 알려줄 거예요!")
-    
-  Q = gr.Textbox(label="질문", placeholder="질문을 넣어주세요.")
-  btn = gr.Button("질문하기")
-  A = gr.TextArea(label="AI 선생님의 설명", placeholder="AI 선생님이 답변 중 입니다.")
+# 응답 생성 및 출력
+if user_input:
+    respond(user_input)
 
-  btn.click(openai_chat, inputs=Q, outputs=A)
+# 채팅 메시지 출력
+for chat in st.session_state.history:
+    if "user" in chat:
+        message(chat["user"], is_user=True, key=f"user_{chat['user']}")
+    if "ai" in chat:
+        message(chat["ai"], is_user=False, key=f"ai_{chat['ai']}")
 
-demo.launch()
+# 초기화 버튼 클릭 시
+if st.button("초기화"):
+    st.session_state.history = []