fixed rewrite

Author: gurveervirk

src/App.js

@@ -250,18 +250,18 @@ const AppContent = ({ isGenerating, setIsGenerating, isBookmarking, setIsBookmar
     }
   };
 
-  const handleTriggerRewrite = async (message) => {
+  const handleTriggerRewrite = async (message, messageID = null, messageType = null) => {
     console.log("handleTriggerRewrite message received");
     setIsGenerating(true);
     navigate('/output');
     try {
       const result = await handleRewrite(message.text);
       const outputData = {
-        id: uuidv4(),
+        id: messageID || uuidv4(),
         input: message.text,
         text: result,
         timestamp: new Date().toISOString(),
-        type: 'Composition'
+        type: messageType || 'Composition'
       };
       await saveOutput(outputData);
     } catch (error) {

src/components/ui/OutputBox.js

@@ -54,8 +54,8 @@ const OutputBox = ({ handleTriggerRewrite, isGenerating }) => {
     setTimeout(() => setCopied(false), 2000);
   };
 
-  const handleRewrite = (text) => {
-    handleTriggerRewrite(text);
+  const handleRewrite = (text, id, type) => {
+    handleTriggerRewrite(text, id, type);
   };
 
   const handleDelete = async (id) => {
@@ -195,7 +195,7 @@ const OutputBox = ({ handleTriggerRewrite, isGenerating }) => {
                 {copied ? <CheckIcon fontSize='small'/> : <ContentCopyIcon fontSize='small'/>}
               </IconButton>
               <IconButton
-                onClick={() => handleRewrite(output.text)}
+                onClick={() => handleRewrite(output.text, output.id, output.type)}
                 sx={{
                   color: '#1A73E8',
                   padding: 0.5,