Improve message handling and logging in chat flow

Removed unnecessary console logs from PlanChatBody. Enhanced logging in WebSocketService for better message type tracking. Fixed PlanPage to clear streaming buffer and correctly handle agent and final result messages, improving UI state management.

Author: Fr4nc3

src/frontend/src/components/content/PlanChatBody.tsx

@@ -14,8 +14,6 @@ const PlanChatBody: React.FC<SimplifiedPlanChatProps> = ({
     OnChatSubmit,
     waitingForPlan
 }) => {
-    console.log("Rendering PlanChatBody", { planData, input, submittingChatDisableInput, waitingForPlan });
-
     return (
 
         <div style={{

src/frontend/src/pages/PlanPage.tsx

@@ -149,6 +149,7 @@ const PlanPage: React.FC = () => {
             console.log('✅ Parsed clarification message:', agentMessageData);
             setClarificationMessage(clarificationMessage.data as ParsedUserClarification | null);
             setAgentMessages(prev => [...prev, agentMessageData]);
+            setStreamingMessageBuffer("");
             setSubmittingChatDisableInput(false);
             scrollToBottom();
 
@@ -178,12 +179,12 @@ const PlanPage: React.FC = () => {
                 timestamp: Date.now(),
                 steps: [],   // intentionally always empty
                 next_steps: [],  // intentionally always empty
-                raw_content: finalMessage.content || '',
-                raw_data: finalMessage || '',
+                raw_content: finalMessage.data.content || '',
+                raw_data: finalMessage.data || '',
             } as AgentMessageData;
             console.log('✅ Parsed final result message:', agentMessageData);
+            setStreamingMessageBuffer("");
             setAgentMessages(prev => [...prev, agentMessageData]);
-            setSubmittingChatDisableInput(true);
             scrollToBottom();
 
         });
@@ -438,7 +439,7 @@ const PlanPage: React.FC = () => {
                     );
                 }
             } finally {
-                setSubmittingChatDisableInput(false);
+
             }
         },
         [planData?.plan, showToast, dismissToast, loadPlanData]

src/frontend/src/services/WebSocketService.tsx

@@ -183,13 +183,13 @@ class WebSocketService {
 
         //console.log('WebSocket message received:', message);
         const hasClarification = /\bclarifications?\b/i.test(message.data || '');
-        console.log("Message ':", message);
+
         if (hasClarification) {
             console.log("Message contains 'clarification':", message.data);
         }
         switch (message.type) {
             case WebsocketMessageType.PLAN_APPROVAL_REQUEST: {
-                console.log("enter plan approval request");
+                console.log("Message Plan Approval Request':", message);
                 const parsedData = PlanDataService.parsePlanApprovalRequest(message.data);
                 if (parsedData) {
                     const structuredMessage: ParsedPlanApprovalRequest = {
@@ -206,6 +206,7 @@ class WebSocketService {
             }
 
             case WebsocketMessageType.AGENT_MESSAGE: {
+                console.log("Message Agent':", message);
                 if (message.data) {
                     console.log('WebSocket message received:', message);
                     const transformed = PlanDataService.parseAgentMessage(message);
@@ -224,6 +225,7 @@ class WebSocketService {
             }
 
             case WebsocketMessageType.USER_CLARIFICATION_REQUEST: {
+                console.log("Message clarification':", message);
                 if (message.data) {
                     const transformed = PlanDataService.parseUserClarificationRequest(message);
                     console.log('WebSocket USER_CLARIFICATION_REQUEST message received:', transformed);
@@ -234,13 +236,15 @@ class WebSocketService {
 
 
             case WebsocketMessageType.AGENT_TOOL_MESSAGE: {
+                console.log("Message agent tool':", message);
                 if (message.data) {
                     //const transformed = PlanDataService.parseUserClarificationRequest(message);
                     this.emit(WebsocketMessageType.AGENT_TOOL_MESSAGE, message);
                 }
                 break;
             }
             case WebsocketMessageType.FINAL_RESULT_MESSAGE: {
+                console.log("Message final result':", message);
                 if (message.data) {
                     //const transformed = PlanDataService.parseFinalResultMessage(message);
                     this.emit(WebsocketMessageType.FINAL_RESULT_MESSAGE, message);
@@ -254,11 +258,13 @@ class WebSocketService {
             case WebsocketMessageType.AGENT_STREAM_START:
             case WebsocketMessageType.AGENT_STREAM_END:
             case WebsocketMessageType.SYSTEM_MESSAGE: {
+                console.log("Message other types':", message);
                 this.emit(message.type, message);
                 break;
             }
 
             default: {
+                console.log("Message default':", message);
                 this.emit(message.type, message);
                 break;
             }