conversion to message/send and new protocol version

app.py

@@ -94,7 +94,8 @@ def load_session_history(session_id: str) -> list:
 
 def respond(message: str, history: list) -> tuple[str, list]:
     server_url = get_selected_server()
-    return "", send_message(server_url, message)
+    history = send_message(server_url, message)  
+    return "", history
 
 def create_chat_interface():
     with gr.Blocks() as demo:
@@ -153,11 +154,10 @@ def create_chat_interface():
             with gr.Column(scale=2):
                 chatbot = gr.Chatbot(label="Chat", type="messages")
                 chat_input = gr.Textbox(label="Message", placeholder="Type a message...")
-                chat_state = gr.State([])
 
                 chat_input.submit(
                     respond,
-                    inputs=[chat_input, chat_state],
+                    inputs=[chat_input],
                     outputs=[chat_input, chatbot],
                     queue=False,
                 )

task_management.py

@@ -2,6 +2,7 @@ import uuid
 import json
 import requests
 import time
+import gradio as gr
 
 # Data class of a session with ID and task history
 class Session:
@@ -101,14 +102,14 @@ def send_message(server_url, content):
         if "error" in result:
             error_message = result["error"].get("message", "Unknown error")
             sessions[current_session_id].add_task({"role": "assistant", "content": f"Error: {error_message}"})
-            yield sessions[current_session_id].get_history()
+            return sessions[current_session_id].get_history()
             return
 
         # If it's a task, we need to poll for updates.
         task_id = result.get("id")
         if not task_id:
             sessions[current_session_id].add_task({"role": "assistant", "content": "No task ID returned."})
-            yield sessions[current_session_id].get_history()
+            return sessions[current_session_id].get_history()
             return
         # TODO allow for multiple concurrent tasks in the client
         # should store task IDs in the session and add a view into interface
@@ -116,13 +117,18 @@ def send_message(server_url, content):
         # Poll for task updates
         return poll_for_task_completion(server_url, task_id, current_session_id)
 
-    # If the response contains a result, it should be a direct message.
-    agent_reply = result.get("result", {}).get("content", "")
-    
+    # If the response contains a result, it should be a direct message object
+    # Get the parts where the kind is "text", then get the text value
+    message_parts = result.get("result", {}).get("parts", [])
+    agent_reply = next((part["text"] for part in message_parts if part.get("kind") == "text"), "")
+    # convert agent_reply_object from a string of json into a Python object to get the text field
+
     if agent_reply:
         sessions[current_session_id].add_task({"role": "assistant", "content": agent_reply})
     
-    yield sessions[current_session_id].get_history()
+    history = sessions[current_session_id].get_history()
+    return history
+
 
 # Poll for task updates
 def poll_for_task_completion(server_url, task_id, current_session_id):