Update seo_agent/seo_agent.py

seo_agent/seo_agent.py

@@ -1,36 +1,40 @@
-# seo_agent.py
-
-import os
-from string import Template
-from chat_wrapper import ChatRefiner
-
-class SEOAgent:
-    def __init__(self, model_name="gemini-1.5-flash"):
-        base     = os.path.dirname(__file__)
-        seo_path = os.path.join(base, "prompts", "seo_prompt.txt")
-        img_path = os.path.join(base, "prompts", "image_prompt.txt")
-
-        self.chat = ChatRefiner(model_name=model_name)
-
-        # load as dollar‐templates
-        with open(seo_path, encoding="utf-8") as f:
-            self.seo_template = Template(f.read())
-        with open(img_path, encoding="utf-8") as f:
-            self.img_template = Template(f.read())
-
-    def analyze_seo(self, html: str, text: str) -> str:
-        """
-        Returns the raw SEO analysis text from the chat model.
-        """
-        prompt = self.seo_template.safe_substitute(
-            html=html,
-            text=text,
-        )
-        return self.chat.answer(prompt)
-
-    def analyze_images(self, images: list[str], text: str) -> str:
-        prompt = self.img_template.safe_substitute(
-            images=images,      # pass the list directly
-            text=text,
-        )
-        return self.chat.answer(prompt)
+# seo_agent.py
+
+import os
+from string import Template
+from chat_wrapper import ChatRefiner
+
+class SEOAgent:
+    def __init__(self, model_name="gemini-1.5-flash"):
+        base     = os.path.dirname(__file__)
+        seo_path = os.path.join(base, "prompts", "seo_prompt.txt")
+        img_path = os.path.join(base, "prompts", "image_prompt.txt")
+
+        self.chat = ChatRefiner(model_name=model_name)
+
+        # load as dollar‐templates
+        with open(seo_path, encoding="utf-8") as f:
+            self.seo_template = Template(f.read())
+        with open(img_path, encoding="utf-8") as f:
+            self.img_template = Template(f.read())
+
+    def analyze_seo(self, html: str, text: str) -> str:
+        """
+        Returns the raw SEO analysis text from the chat model.
+        """
+        short_html = html[:5000] 
+        short_text = text[:5000]
+        prompt = self.seo_template.safe_substitute(
+            html=short_html,
+            text=short_text,
+        )
+        return self.chat.answer(prompt)
+
+    def analyze_images(self, images: list[str], text: str) -> str:
+        limited_images = list(dict.fromkeys(images))[:10]
+        short_text = text[:5000]
+        prompt = self.img_template.safe_substitute(
+            images=limited_images,      # pass the list directly
+            text=short_text,
+        )
+        return self.chat.answer(prompt)