Update app.py

app.py

@@ -1,7 +1,5 @@
 import os
 import time
-import threading
-
 import gradio as gr
 import google.generativeai as genai
 
@@ -22,48 +20,32 @@ model = genai.GenerativeModel(
   system_instruction="초등학생이 교과 내용에 대해 이해할 수 있게 친절하게 설명하는 선생님. 다양한 예시와 구체적 사례를 들어서 설명. 사진이나 동영상 자료가 있으면 첨부. 출처 표시.",
 )
 
-# 챗 세션 초기화
-chat_session = None
-
-# 출력 텍스트를 저장할 변수
-output_text = ""
-
-def chatbot_worker(input_text):
-  global chat_session, output_text
-  if chat_session is None:
-    chat_session = model.start_chat()
-
-  response = chat_session.send_message(input_text)
-
-  # 실시간 타이핑 효과를 위한 코드
-  for char in response.text:
-    output_text += char
-    time.sleep(0.02)  # 타이핑 속도 조절
-    yield output_text
-
-  yield output_text
-
-def chatbot(input_text):
-  global output_text
-  output_text = ""
-  thread = threading.Thread(target=chatbot_worker, args=(input_text,))
-  thread.start()
-  return output_text
-
-def clear_session():
-  global chat_session, output_text
-  chat_session = None
-  output_text = ""
-  return ""
-
-# Gradio 인터페이스 생성
-iface = gr.Interface(
-  fn=chatbot,
-  inputs=gr.Textbox(lines=5, placeholder="여기에 질문을 입력하세요..."),
-  outputs=gr.Textbox(lines=10, placeholder="AI 선생님의 답변이 여기에 나타납니다...", label="AI 설명"),
-  title="AI 선생님",
-  description="초등 교과 과정에 대한 질문을 입력해보세요.",
-)
-
-# 초기화 버튼 추가
-iface.launch(share=True).queue().button(fn=clear_session, inputs=[], outputs=[iface.components[1]], label="초기화")
+def stream_response(message, history):
+  response = ""
+  for chunk in model.generate_content(message):
+    chunk_message = chunk.text
+    response += chunk_message
+    yield response
+    time.sleep(0.05)
+
+# 인터페이스 구성
+with gr.Blocks() as interface:
+  chatbot = gr.Chatbot()
+  msg = gr.Textbox()
+  clear = gr.Button("대화 초기화")
+
+  def user_message(user_message, history):
+    return "", history + [[user_message, None]]
+
+  def bot_message(history):
+    bot_message = stream_response(history[-1][0], history)
+    history[-1][1] = ""
+    return history
+
+  msg.submit(user_message, [msg, chatbot], [msg, chatbot], queue=False).then(
+      bot_message, chatbot, chatbot
+  )
+  clear.click(lambda: None, None, chatbot, queue=False)
+
+# 인터페이스 실행
+interface.launch()