Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
+import time
 import google.generativeai as genai
 import gradio as gr
-import time
 
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
@@ -29,23 +29,22 @@ model = genai.GenerativeModel(
 
 chat_session = model.start_chat(history=[])
 
+
 def respond(user_input, history):
-    response = chat_session.send_message(user_input)
-    text = response.text
-    history.append((user_input, ""))
+    history.append((user_input, ""))  # 빈 답변으로 시작
+    yield "", history
 
-    # 타이핑 효과 구현
-    for i in range(1, len(text) + 1):
-        history[-1] = (user_input, text[:i])
+    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)  # 타이핑 속도 조절
-        yield "", history  # 입력창을 비우고 history 업데이트
-    
-    # 전체 응답을 반환
-    history[-1] = (user_input, text)
-    yield "", history
+
 
 with gr.Blocks() as demo:
-    gr.Markdown("<div style='font-size: 30px; font-weight: bold;'>gemini chatbot</div>")
+    gr.Markdown("<div style='font-size: 30px; font-weight: bold;'>AI 선생님</div>")
     chatbot = gr.Chatbot(label="채팅창")
     msg = gr.Textbox(label="입력")
     clear = gr.Button("초기화")