feat: Add message editing functionality with AI re-response

[bzqzheng]

Add Message Editing Functionality

Overview

This PR introduces message editing, allowing users to revise their prior inputs and view regenerated AI responses. It enables iterative prompt refinement and A/B testing of message variations.

Key Features

• Edit Button: Hover-to-reveal button on user messages
• Inline Editing: Seamless textarea interface with smooth transitions
• Save/Cancel: Save changes or discard edits with intuitive controls
• Cmd+Enter Shortcut: Instantly saves and re-prompts the AI
• AI Re-response: Triggers new AI response after edits
• Message Pruning: Editing any message (e.g. message 2 of 5) prunes all subsequent messages
• Visual Feedback: Loading indicators, edit mode, and “edited” labels
• Context Preservation: Maintains coherent conversation flow after edit-triggered regeneration

Technical Details

• Added edit state to UserMessage component
• Hooked into useMessageStream for AI regeneration
• Managed message updates and pruning logic in ChatView
• Applied consistent UI/UX patterns
• Included error handling and logging

Files Changed

• ui/desktop/src/components/UserMessage.tsx: Edit functionality and UI
• ui/desktop/src/components/ChatView.tsx: Message update and pruning logic
• ui/desktop/package-lock.json: Dependency updates (if applicable)

This feature enhances the conversation experience by allowing prompt revisions and exploring different AI outputs without breaking the existing flow.

Screenshots

Updated screenshots in the PR thread below.

[zanesq]

@nahiyankhan @spencrmartin for design review

[bzqzheng]

@nahiyankhan @spencrmartin for design review

The current textarea/bubble looks tiny in edit mode. I made some UI changes, and here are the screenshots.
It's not pushed to this PR yet. Let me know what you all think, and I can push these changes if they look better.

[spencrmartin]

Can we cut the emoji use and when edit mode is active rely on the form field to communicate that mode vs showing the status?

[bzqzheng]

Can we cut the emoji use and when edit mode is active rely on the form field to communicate that mode vs showing the status?

done.

[michaelneale]

I like this - would be great if @zanesq gives the blessing as ChatView and UserMessage.tsx as super high trafficked bits of code and very sensitive, but yeah I think this is worthwhile, good idea!

[zanesq]

Will take a look and try it out asap!

[bzqzheng]

Will take a look and try it out asap!

Hi Zane, following up on this feature PR. Is this still being reviewed? It would be a great UX improvement for goose!

[zanesq]

@bzqzheng thanks for the reminder. Agreed, its a nice feature and would definitely want to help get this in. We have a revamped UI coming out shortly so unfortunately you'll need to rebase off that soon. I'll wait to review until we get that out, most likely next week.

[DOsinga]

ok, let's revisit after the big merge (hopefully soon!)

[DOsinga]

ok, merged, so let's continue

[zanesq]

@bzqzheng looking forward to this feature, can you pull in main and get this working again?

[bzqzheng]

@bzqzheng looking forward to this feature, can you pull in main and get this working again?

Sure, will get to it some time next week.

[bzqzheng]

@zanesq the UI major update makes this PR "unmergable" since some files are revamped and deleted.
Created this new PR with the same feature change.