Update app.py

app.py

@@ -1,68 +1,65 @@
 import os
 import time
-import gradio as gr
 
 import google.generativeai as genai
+from google.generativeai.chat import ChatSession
+import gradio as gr
 
+# Google Gemini API 키 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
 generation_config = {
-    "temperature": 1,
+    "temperature": 0.8,
     "top_p": 0.95,
     "top_k": 64,
-    "max_output_tokens": 8192,
+    "max_output_tokens": 1024,
     "response_mime_type": "text/plain",
 }
+model = genai.GenerativeModel(
+    model_name="gemini-1.5-pro",
+    generation_config=generation_config,
+    system_instruction="""뛰어난 전문성을 갖춘 언어 치료 전문가, 한글 교육 전문가, 한국어 발음 교정 전문가로서 친절하고 공감하는 태도로 사용자와 상담을 진행하세요.""",
+)
 
-# 시스템 명령어 설정
-system_instruction = "언어 치료 전문가, 한글 교육 전문가, 한국어 발음 교정 전문가"
+# 상담 내용 저장
+chat_history = []
+chat_session = None
 
-def type_text(text):
-    for char in text:
-        print(char, end='', flush=True)
-        time.sleep(0.03) # 타이핑 속도 조절
-    print("\n") # 줄바꿈 추가
 
-def chat(message, history):
-    # 새로운 상담 시작 시 기록 초기화
-    if message == "/새로운 상담 시작":
-        history = []
-        return "새로운 상담을 시작합니다.", history
+def predict(message, history):
+    global chat_session
+    if chat_session is None:
+        chat_session = model.start_chat()
 
-    # 모델 생성 및 채팅 세션 시작
-    model = genai.GenerativeModel(
-        model_name="gemini-1.5-pro",
-        generation_config=generation_config,
-        system_instruction=system_instruction,
-    )
-    chat_session = model.start_chat(history=history)
+    if message.strip() == "":
+        return history, ""
 
-    # 사용자 메시지 추가
-    history.append({"role": "user", "message": message})
+    history = history + [[message, None]]
+    response = chat_session.send_message(message).text
 
-    # AI 응답 생성 및 추가
-    response = chat_session.send_message(message)
-    history.append({"role": "assistant", "message": response.text})
+    # 타이핑 효과를 위해 한 글자씩 출력
+    for i in range(len(response) + 1):
+        time.sleep(0.03)
+        yield history + [[message, response[:i]]]
 
-    # 타이핑 효과 출력
-    type_text(response.text)
 
-    return "", history
+def new_session(chatbot, msg):
+    global chat_session, chat_history
+    chat_session = None
+    chat_history = []
+    chatbot = []
+    msg = ""
+    return chatbot, msg, "새로운 상담을 시작합니다."
+
 
 # Gradio 인터페이스 설정
-iface = gr.Interface(
-    fn=chat,
-    inputs=[
-        gr.inputs.Textbox(lines=5, placeholder="문의 내용을 입력하세요."),
-        gr.State([]),
-    ],
-    outputs=[
-        gr.outputs.Textbox(label="AI 답변"),
-        gr.State(),
-    ],
-    title="AI 언어 치료 상담 챗봇",
-    description="언어 치료 전문가 AI와 상담해보세요. 새로운 상담을 시작하려면 '/새로운 상담 시작'을 입력하세요.",
-)
+with gr.Blocks() as demo:
+    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])
 
-iface.launch()
+demo.launch(share=True)