Update app.py

app.py

@@ -49,6 +49,21 @@ async def call_model(prompt: str, model: str = "gpt-4o", api_key: str = None, ma
             raise
     raise Exception(f"Failed to get response after {max_retries} attempts.")
 
+# -------------------- Conversation History Conversion --------------------
+def convert_history(history: List[Dict[str, str]]) -> List[Dict[str, str]]:
+    """
+    Convert our internal conversation history format (with 'agent' and 'message' keys)
+    into the Gradio messages format (with 'role' and 'content').
+    """
+    converted = []
+    for entry in history:
+        if entry["agent"].lower() == "user":
+            converted.append({"role": "user", "content": entry["message"]})
+        else:
+            # You can customize the formatting below as desired.
+            converted.append({"role": "assistant", "content": f'{entry["agent"]}: {entry["message"]}'})
+    return converted
+
 # -------------------- Shared Context --------------------
 class Context:
     def __init__(self, original_task: str, optimized_task: Optional[str] = None,
@@ -64,7 +79,8 @@ class Context:
         self.test_cases = test_cases
         self.test_results = test_results
         self.documentation = documentation
-        self.conversation_history = conversation_history or []
+        # Initialize conversation history with the user's original task.
+        self.conversation_history = conversation_history or [{"agent": "User", "message": original_task}]
 
     def add_conversation_entry(self, agent_name: str, message: str):
         self.conversation_history.append({"agent": agent_name, "message": message})
@@ -292,7 +308,8 @@ def process_conversation_generator(task_message: str, api_key: str,
         try:
             msg = log_queue.get(timeout=0.1)
             if isinstance(msg, tuple) and msg[0] == "result":
-                yield gr.Chatbot.update(value=msg[1], visible=True)
+                # Convert conversation history to the required messages format.
+                yield gr.Chatbot.update(value=convert_history(msg[1]))
             else:
                 yield msg
         except queue.Empty:
@@ -320,7 +337,7 @@ with gr.Blocks() as demo:
     gr.Markdown("## Multi-Agent Task Solver with Human-in-the-Loop")
     
     with gr.Row():
-        # Set type="messages" to avoid deprecation warnings.
+        # Set type="messages" to match the expected format.
         chat_output = gr.Chatbot(label="Conversation", type="messages")
     
     with gr.Row():
@@ -331,7 +348,7 @@ with gr.Blocks() as demo:
     
     send_button = gr.Button("Send")
     
-    # Remove stream=True (not supported in your Gradio version).
+    # Note: 'stream=True' has been removed.
     send_button.click(fn=multi_agent_chat, inputs=[message_input, api_key_input], outputs=chat_output)
 
 if __name__ == "__main__":