Maybe we fixed bugs?

frontend/src/components/TreeOfThoughtVisual.js

@@ -49,7 +49,7 @@ const TreeOfThoughtVisual = ({ data }) => {
 
   }, [data]);
 
-  return <svg ref={svgRef} width="800" height="600" />;
+  return <svg ref={svgRef} width="1920" height="1080" />;
 };
 
 export default TreeOfThoughtVisual;

src/main/java/com/ioa/agent/AgentInfo.java

@@ -20,4 +20,9 @@ public class AgentInfo {
     public List<String> getCapabilities() {
         return this.capabilities;
     }
+
+    @Override
+    public String toString() {
+        return "AgentInfo{id='" + id + "', name='" + name + "'}";
+    }
 }

src/main/java/com/ioa/team/TeamFormation.java

@@ -63,25 +63,34 @@ public class TeamFormation {
         String[] lines = reasoning.split("\n");
         String selectedTeamLine = "";
         for (String line : lines) {
-            if (line.startsWith("Final Team Selection:")) {
+            if (line.toLowerCase().contains("final team selection")) {
-                selectedTeamLine = line.substring("Final Team Selection:".length()).trim();
+                selectedTeamLine = line.substring(line.indexOf(":") + 1).trim();
                 break;
             }
         }
-
+    
+        System.out.println("DEBUG: Selected team line: " + selectedTeamLine);
+    
         if (selectedTeamLine.isEmpty()) {
             System.out.println("DEBUG: No team selection found in the response.");
             return List.of();
         }
-
+    
-        List<String> selectedIds = Arrays.asList(selectedTeamLine.split(",\\s*"));
+        List<String> selectedIds = Arrays.asList(selectedTeamLine.split("[,\\s]+"));
         System.out.println("DEBUG: Parsed agent IDs: " + selectedIds);
-
+    
         List<AgentInfo> team = potentialAgents.stream()
                 .filter(agent -> selectedIds.contains(agent.getId().trim()))
                 .collect(Collectors.toList());
-
+    
         System.out.println("DEBUG: Final team: " + team);
+    
+        if (team.isEmpty() && !selectedIds.isEmpty()) {
+            // If no agents were matched but we did find IDs, add the first potential agent as a fallback
+            team = potentialAgents.stream().limit(1).collect(Collectors.toList());
+            System.out.println("DEBUG: No agents matched IDs. Using fallback: " + team);
+        }
+    
         return team;
     }
 }