Persist generated_content in CosmosDB and restore on conversation load

- Added generated_content parameter to save_conversation in cosmos_service.py
- Added save_generated_content method to update existing conversations
- Updated /api/generate endpoint to save full generated_content object
- Updated handleSelectConversation in App.tsx to restore generatedContent from DB

Author: hunterjam

content-gen/src/app.py

@@ -328,6 +328,8 @@ async def generate():
                 cosmos_service = await get_cosmos_service()
                 text_content = response.get("text_content", {})
                 headline = text_content.get("headline", "") if isinstance(text_content, dict) else ""
+                
+                # Save the message
                 await cosmos_service.add_message_to_conversation(
                     conversation_id=conversation_id,
                     user_id=user_id,
@@ -338,6 +340,21 @@ async def generate():
                         "timestamp": datetime.now(timezone.utc).isoformat()
                     }
                 )
+                
+                # Save the full generated content for restoration
+                generated_content_to_save = {
+                    "text_content": response.get("text_content"),
+                    "image_url": response.get("image_url"),
+                    "image_prompt": response.get("image_prompt"),
+                    "image_revised_prompt": response.get("image_revised_prompt"),
+                    "violations": response.get("violations", []),
+                    "requires_modification": response.get("requires_modification", False)
+                }
+                await cosmos_service.save_generated_content(
+                    conversation_id=conversation_id,
+                    user_id=user_id,
+                    generated_content=generated_content_to_save
+                )
             except Exception as e:
                 logger.warning(f"Failed to save generated content to CosmosDB: {e}")
 

content-gen/src/backend/services/cosmos_service.py

@@ -255,7 +255,8 @@ async def save_conversation(
         user_id: str,
         messages: List[dict],
         brief: Optional[CreativeBrief] = None,
-        metadata: Optional[dict] = None
+        metadata: Optional[dict] = None,
+        generated_content: Optional[dict] = None
     ) -> dict:
         """
         Save or update a conversation.
@@ -266,6 +267,7 @@ async def save_conversation(
             messages: List of conversation messages
             brief: Associated creative brief
             metadata: Additional metadata
+            generated_content: Generated marketing content
         
         Returns:
             The saved conversation document
@@ -278,12 +280,49 @@ async def save_conversation(
             "messages": messages,
             "brief": brief.model_dump() if brief else None,
             "metadata": metadata or {},
+            "generated_content": generated_content,
             "updated_at": datetime.now(timezone.utc).isoformat()
         }
 
         result = await self._conversations_container.upsert_item(item)
         return result
 
+    async def save_generated_content(
+        self,
+        conversation_id: str,
+        user_id: str,
+        generated_content: dict
+    ) -> dict:
+        """
+        Save generated content to an existing conversation.
+        
+        Args:
+            conversation_id: Unique conversation identifier
+            user_id: User ID for partition key
+            generated_content: The generated content to save
+        
+        Returns:
+            Updated conversation document
+        """
+        await self.initialize()
+        
+        conversation = await self.get_conversation(conversation_id, user_id)
+        
+        if conversation:
+            conversation["generated_content"] = generated_content
+            conversation["updated_at"] = datetime.now(timezone.utc).isoformat()
+        else:
+            conversation = {
+                "id": conversation_id,
+                "user_id": user_id,
+                "messages": [],
+                "generated_content": generated_content,
+                "updated_at": datetime.now(timezone.utc).isoformat()
+            }
+        
+        result = await self._conversations_container.upsert_item(conversation)
+        return result
+    
     async def add_message_to_conversation(
         self,
         conversation_id: str,

content-gen/src/frontend/src/App.tsx

@@ -52,7 +52,40 @@ function App() {
         setMessages(loadedMessages);
         setPendingBrief(null);
         setConfirmedBrief(data.brief || null);
-        setGeneratedContent(null);
+        
+        // Restore generated content if it exists
+        if (data.generated_content) {
+          const gc = data.generated_content;
+          // Parse text_content if it's a string
+          let textContent = gc.text_content;
+          if (typeof textContent === 'string') {
+            try {
+              textContent = JSON.parse(textContent);
+            } catch {
+              // Keep as string if not valid JSON
+            }
+          }
+          
+          const restoredContent: GeneratedContent = {
+            text_content: typeof textContent === 'object' && textContent ? {
+              headline: textContent?.headline,
+              body: textContent?.body,
+              cta_text: textContent?.cta,
+              tagline: textContent?.tagline,
+            } : undefined,
+            image_content: (gc.image_url || gc.image_prompt) ? {
+              image_url: gc.image_url,
+              prompt_used: gc.image_prompt,
+              alt_text: gc.image_revised_prompt || 'Generated marketing image',
+            } : undefined,
+            violations: gc.violations || [],
+            requires_modification: gc.requires_modification || false,
+          };
+          setGeneratedContent(restoredContent);
+        } else {
+          setGeneratedContent(null);
+        }
+        
         setSelectedProducts([]);
       }
     } catch (error) {