Update app.py

app.py

@@ -1,64 +1,80 @@
 import os
 import time
-
 import google.generativeai as genai
 import gradio as gr
 
-# Google Gemini API 키 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
 generation_config = {
-    "temperature": 0.8,
+    "temperature": 1,
     "top_p": 0.95,
     "top_k": 64,
-    "max_output_tokens": 1024,
+    "max_output_tokens": 15000,
     "response_mime_type": "text/plain",
 }
-model = genai.GenerativeModel(
-    model_name="gemini-1.5-pro",
-    generation_config=generation_config,
-    system_instruction="""뛰어난 전문성을 갖춘 언어 치료 전문가, 한글 교육 전문가, 한국어 발음 교정 전문가로서 친절하고 공감하는 태도로 사용자와 상담을 진행하세요.""",
-)
-
-# 상담 내용 저장
-chat_history = []
+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"},
+]
+
+# 전역 변수로 chat_session 초기화
 chat_session = None
 
 
-def predict(message, history):
+def start_new_chat_session():
     global chat_session
-    if chat_session is None:
-        chat_session = model.start_chat()
+    chat_session = (
+        genai.GenerativeModel(
+            model_name="gemini-1.5-pro",
+            safety_settings=safety_settings,
+            generation_config=generation_config,
+            system_instruction="""당신은 뛰어난 전문성을 갖춘 언어 치료 전문가, 한글 교육 전문가, 한국어 발음 교정 전문가입니다. 
+            선생님이 학생의 언어 발달, 발음, 문장 표현 등에 대해 질문하면, 친절하고 공감하는 태도로 상담을 진행하세요. 
+            선생님에게 학생을 지도하기 위한 조언과 핵심적인 팁을 제공하여, 선생님이 학생의 언어 능력을 향상시키도록 도와주세요.""",
+        )
+        .start_chat(history=[])
+    )
+
 
-    if message.strip() == "":
-        return history, ""
+# 초기 세션 시작
+start_new_chat_session()
 
-    history = history + [[message, None]]
-    response = chat_session.send_message(message).text
 
-    # 타이핑 효과를 위해 한 글자씩 출력
-    for i in range(len(response) + 1):
-        time.sleep(0.03)
-        yield history + [[message, response[:i]]]
+def respond(user_input, history):
+    global chat_session
+    history.append((user_input, ""))
+    yield "", history
+
+    response = chat_session.send_message(user_input)
+    full_response = ""
+    for chunk in response.text.split():
+        full_response += chunk + " "
+        history[-1] = (user_input, full_response)
+        yield "", history
+        time.sleep(0.05)
 
 
-def new_session(chatbot, msg):
-    global chat_session, chat_history
-    chat_session = None
-    chat_history = []
-    chatbot = []
-    msg = ""
-    return chatbot, msg, "새로운 상담을 시작합니다."
+def clear_chat():
+    start_new_chat_session()
+    return []
 
 
-# Gradio 인터페이스 설정
 with gr.Blocks() as demo:
+    gr.Markdown(
+        "<div style='font-size: 30px; font-weight: bold;'>AI 언어 교육 상담</div>"
+    )
+    gr.Markdown(
+        "학생의 언어 발달, 발음, 문장 표현 등에 대해 궁금한 점을 AI 언어 교육 전문가에게 질문하세요."
+    )
     chatbot = gr.Chatbot(label="채팅창")
     msg = gr.Textbox(label="입력")
     clear = gr.Button("초기화")
 
-    msg.submit(predict, [msg, chatbot], [chatbot, msg])
-    clear.click(new_session, [chatbot, msg], [chatbot, msg, msg])
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    # chatbot 상태 초기화 및 새로운 세션 시작
+    clear.click(clear_chat, outputs=chatbot, queue=False)
 
 demo.launch(share=True)