'Update'

main.py

@@ -30,7 +30,15 @@ handler = WebhookHandler(os.environ.get("CHANNEL_SECRET"))
 
 genai.configure(api_key=os.environ.get("GOOGLE_API_KEY"))
 model = genai.GenerativeModel('gemini-pro')
+chats: dict[str, genai.ChatSession] = {}
 
+def get_chat(user_id: str) -> genai.ChatSession:
+    if user_id in chats:
+        return chats.get(user_id)
+    else:
+        chat = model.start_chat()
+        chats[user_id] = chat
+        return chat
 
 @app.route("/")
 def home():
@@ -58,9 +66,10 @@ def callback():
 
 @handler.add(MessageEvent, message=TextMessageContent)
 def handle_message(event):
-    response = model.generate_content(event.message.text)
     app.logger.info(f"UserID: {event.source.user_id}")
     app.logger.info(f"Q: {event.message.text}")
+    chat = get_chat(event.source.user_id)
+    response = chat.send_message(event.message.text)
     app.logger.info(f"A: {response.text}")
     with ApiClient(configuration) as api_client:
         line_bot_api = MessagingApi(api_client)