Update app.py

app.py

@@ -1,46 +1,51 @@
-# -*- 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
 
-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
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+
+genai.configure(api_key=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-pro",
+    safety_settings=safety_settings,
+    generation_config=generation_config,
+    system_instruction="교사가 학습 목표를 입력하면, 학생들이 학습 목표에 도달할 수 있게 질문을 계속 해줘, 학생들은 네 질문에 답을 할 것이야. 그러면 바로 답을 제시하지 말고 질문형태로 피드백을 해줘. 질문을 통해서 학생들이 스스로 학습 목표를 도달할 수 있도록 도와줘.",
 )
 
-    return completion.choices[0].message.content
-
-openai_chat("한글은 누가 만들었어?")
+chat_session = model.start_chat(history=[])
 
-import gradio as gr
+def chatbot(message, history):
+    """
+    챗봇과의 대화를 처리하는 함수
+    """
+    global chat_session
 
-with gr.Blocks() as demo:
-  Q = gr.Textbox(label="질문", placeholder="질문을 넣어주세요.")
-  btn = gr.Button("질문하기")
-  A = gr.TextArea(label="AI 선생님의 설명", placeholder="AI 선생님이 답변 중 입니다.")
+    # 새로운 메시지가 입력되면 챗봇에 전달
+    response = chat_session.send(message)
+    history.append((message, response))
+    return history, response
 
-  btn.click(openai_chat, inputs=Q, outputs=A)
+iface = gr.ChatInterface(
+    fn=chatbot,
+    title="학습 챗봇",
+    description="학습 목표를 입력하고 챗봇과 대화해보세요!",
+    theme="compact",
+)
 
-demo.launch()
+# 실행
+iface.launch(share=True)