Upload content_optimizer.py

agents/content_optimizer.py

@@ -41,14 +41,14 @@ class ContentOptimizationAgent:
             num_beams=4,
             early_stopping=True
         )
-        return self.meta_tokenizer.decode(summary_ids, skip_special_tokens=True)
+        return self.meta_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 
     def optimize_title(self, content, keyword):
         if not hasattr(self, 'title_model'): return ""
         prompt = f"Write an SEO-optimized page title under 60 characters for content about '{keyword}': {content[:500]}"
         inputs = self.title_tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
         outputs = self.title_model.generate(inputs["input_ids"], max_length=60, num_beams=5)
-        return self.title_tokenizer.decode(outputs, skip_special_tokens=True)
+        return self.title_tokenizer.decode(outputs[0], skip_special_tokens=True)
 
     def analyze_page(self, url):
         logger.info(f"Optimizing content for: {url}")