Upload content_optimizer.py

agents/content_optimizer.py

@@ -41,14 +41,22 @@ class ContentOptimizationAgent:
             num_beams=4,
             early_stopping=True
         )
-        return self.meta_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+        try:
+            return self.meta_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+        except Exception as e:
+            logger.error(f"Meta gen failed: {e}")
+            return "Could not generate meta description."
 
     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[0], skip_special_tokens=True)
+        try:
+            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[0], skip_special_tokens=True)
+        except Exception as e:
+            logger.error(f"Title gen failed: {e}")
+            return "Could not generate title."
 
     def analyze_page(self, url):
         logger.info(f"Optimizing content for: {url}")