Update app.py

app.py

@@ -1,59 +1,46 @@
+# -*- 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 google.generativeai as genai
-import gradio as gr
 
-# Google Gemini API 키 설정
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
-
-# 모델 설정
-generation_config = {
-    "temperature": 1,
-    "top_p": 0.95,
-    "top_k": 64,
-    "max_output_tokens": 8192,
-    "response_mime_type": "text/plain",
-}
-safety_settings = [
-    {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE",
-    },
-    {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE",
-    },
-    {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE",
-    },
-    {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE",
-    },
-]
-
-model = genai.GenerativeModel(
-    model_name="gemini-1.5-flash",
-    safety_settings=safety_settings,
-    generation_config=generation_config,
-    system_instruction="교과 관련 개념이나 내용을 입력하면, 내용에 대해 이해할 수 있게 친절하게 설명하는 선생님. 다양한 예시와 구체적 사례를 들어서 상세하게 설명해줘. 2000자 이상 분량으로 설명글 작성. 출처 표시가 가능하면 출처표시",
+from openai import OpenAI
+import openai
+
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+client = openai.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학년 수준 범위 내에서 학생의 질문에 친절하게 답하는 교사 역할, 설명은 구체적인 예시를 들어서, 어려운 단어는 별도로 설명해줘. 잘하면 10달러의 팁을 줄게."},
+        {"role": "user", "content": text}
+      ],
+  temperature=1,
+  max_tokens=1000,
+  top_p=0.9,
+  frequency_penalty=0,
+  presence_penalty=0
 )
 
-chat_session = model.start_chat(history=[])
+    return completion.choices[0].message.content
 
-# Gradio 챗봇 함수 정의
-def gemini_chat(text):
-    response = chat_session.send_message(text)
-    return response.text
+openai_chat("한글은 누가 만들었어?")
+
+import gradio as gr
 
-# Gradio 인터페이스 설정
 with gr.Blocks() as demo:
-    gr.Markdown("## AI 선생님께 질문하기")
-    Q = gr.Textbox(label="질문", placeholder="질문을 넣어주세요.")
-    btn = gr.Button("질문하기")
-    A = gr.TextArea(label="AI 선생님의 설명", placeholder="AI 선생님이 답변 중 입니다.")
+  Q = gr.Textbox(label="질문", placeholder="질문을 넣어주세요.")
+  btn = gr.Button("질문하기")
+  A = gr.TextArea(label="AI 선생님의 설명", placeholder="AI 선생님이 답변 중 입니다.")
 
-    btn.click(gemini_chat, inputs=Q, outputs=A)
+  btn.click(openai_chat, inputs=Q, outputs=A)
 
-# 데모 실행
 demo.launch()