fine tuning

backend/igny8_core/ai/functions/generate_content.py

@@ -161,6 +161,7 @@ class GenerateContentFunction(BaseAIFunction):
         """
         STAGE 3: Save content using final Stage 1 Content model schema.
         Creates independent Content record (no OneToOne to Task).
+        Handles tags and categories from AI response.
         """
         if isinstance(original_data, list):
             task = original_data[0] if original_data else None
@@ -179,6 +180,9 @@ class GenerateContentFunction(BaseAIFunction):
             meta_description = parsed.get('meta_description') or parsed.get('seo_description')
             primary_keyword = parsed.get('primary_keyword') or parsed.get('focus_keyword')
             secondary_keywords = parsed.get('secondary_keywords') or parsed.get('keywords', [])
+            # Extract tags and categories from AI response
+            tags_from_response = parsed.get('tags', [])
+            categories_from_response = parsed.get('categories', [])
         else:
             # Plain text response
             content_html = str(parsed)
@@ -187,6 +191,8 @@ class GenerateContentFunction(BaseAIFunction):
             meta_description = None
             primary_keyword = None
             secondary_keywords = []
+            tags_from_response = []
+            categories_from_response = []
         
         # Calculate word count
         word_count = 0
@@ -222,8 +228,51 @@ class GenerateContentFunction(BaseAIFunction):
         if task.taxonomy_term:
             content_record.taxonomy_terms.add(task.taxonomy_term)
         
-        # Link all keywords from task as taxonomy terms (if they have taxonomy mappings)
-        # This is optional - keywords are M2M on Task, not directly on Content
+        # Process tags from AI response
+        if tags_from_response and isinstance(tags_from_response, list):
+            from django.utils.text import slugify
+            for tag_name in tags_from_response:
+                if tag_name and isinstance(tag_name, str):
+                    tag_name = tag_name.strip()
+                    if tag_name:
+                        try:
+                            # Get or create tag taxonomy term
+                            tag_obj, _ = ContentTaxonomy.objects.get_or_create(
+                                site=task.site,
+                                name=tag_name,
+                                taxonomy_type='tag',
+                                defaults={
+                                    'slug': slugify(tag_name),
+                                    'sector': task.sector,
+                                    'account': task.account,
+                                }
+                            )
+                            content_record.taxonomy_terms.add(tag_obj)
+                        except Exception as e:
+                            logger.warning(f"Failed to add tag '{tag_name}': {e}")
+        
+        # Process categories from AI response
+        if categories_from_response and isinstance(categories_from_response, list):
+            from django.utils.text import slugify
+            for category_name in categories_from_response:
+                if category_name and isinstance(category_name, str):
+                    category_name = category_name.strip()
+                    if category_name:
+                        try:
+                            # Get or create category taxonomy term
+                            category_obj, _ = ContentTaxonomy.objects.get_or_create(
+                                site=task.site,
+                                name=category_name,
+                                taxonomy_type='category',
+                                defaults={
+                                    'slug': slugify(category_name),
+                                    'sector': task.sector,
+                                    'account': task.account,
+                                }
+                            )
+                            content_record.taxonomy_terms.add(category_obj)
+                        except Exception as e:
+                            logger.warning(f"Failed to add category '{category_name}': {e}")
         
         # STAGE 3: Update task status to completed
         task.status = 'completed'

backend/igny8_core/ai/tasks.py

@@ -707,6 +707,25 @@ def process_image_generation_queue(self, image_ids: list, account_id: int = None
             })
             failed += 1
     
+    # Check if all images for the content are generated and update status to 'review'
+    if content_id and completed > 0:
+        try:
+            from igny8_core.business.content.models import Content, Images
+            
+            content = Content.objects.get(id=content_id)
+            
+            # Check if all images for this content are now generated
+            all_images = Images.objects.filter(content=content)
+            pending_images = all_images.filter(status='pending').count()
+            
+            # If no pending images and content is still in draft, move to review
+            if pending_images == 0 and content.status == 'draft':
+                content.status = 'review'
+                content.save(update_fields=['status'])
+                logger.info(f"[process_image_generation_queue] Content #{content_id} status updated to 'review' (all images generated)")
+        except Exception as e:
+            logger.error(f"[process_image_generation_queue] Error updating content status: {str(e)}", exc_info=True)
+    
     # Final state
     logger.info("=" * 80)
     logger.info(f"process_image_generation_queue COMPLETED")