Update app.py

app.py

@@ -39,22 +39,23 @@ async def echo(message, history, state, persona, use_generated_summaries):
         attached_file = find_attached_file(filename, state["attached_files"])
         if attached_file is None: 
             path_gcp = await client.files.upload(path=path_local)
+            path_wrap = types.Part.from_uri(
+                file_uri=path_gcp.uri, mime_type=path_gcp.mime_type
+            )
             state["attached_files"].append({
                 "name": filename,
                 "path_local": path_local,
                 "gcp_entity": path_gcp,
-                "path_gcp": path_gcp.name,
-                "mime_type=": path_gcp.mime_type,
+                "path_gcp": path_wrap,
+                "mime_type": path_gcp.mime_type,
                 "expiration_time": path_gcp.expiration_time,
             })
-            attached_file = path_gcp
-
-    user_message = [message['text']]
-    if attached_file: user_message.append(attached_file)
+            attached_file = path_wrap
 
-    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
 
     response_chunks = ""
     model_contents = ""
@@ -88,6 +89,10 @@ async def echo(message, history, state, persona, use_generated_summaries):
             ),
             gr.DownloadButton(visible=False)
         )        
+
+    state['messages'] = state['messages'] + [
+        types.Content(role='model', parts=[types.Part.from_text(text=response_chunks)])
+    ]
     
     # make summary
     response = await client.models.generate_content(