Update app.py

app.py

@@ -48,15 +48,14 @@ async def echo(message, history, state, persona):
             })
             attached_file = path_gcp
 
-    user_message = [message['text']]
-    if attached_file: user_message.append(attached_file)
-
-    chat_history = state['messages']
-    chat_history = chat_history + user_message
-    state['messages'] = chat_history
+    user_message_parts = [types.Part.from_text(text=message['text'])]
+    if attached_file: user_message_parts.append(attached_file)
+    user_message = [types.Content(role='user', parts=user_message_parts)]
+    
+    state['messages'] = state['messages'] + user_message
+    print(state['messages'])
 
     response_chunks = ""
-    print(state['messages'])
     model_content_stream = await client.models.generate_content_stream(
         model=args.model, 
         contents=state['messages'], 
@@ -78,6 +77,10 @@ async def echo(message, history, state, persona):
             ),
             gr.DownloadButton(visible=False)
         )        
+
+    state['messages'] = state['messages'] + [
+        types.Content(role='model', [types.Part.from_text(text=response_chunks)])
+    ]
     
     # make summary
     response = await client.models.generate_content(