Update agents.py

agents.py

@@ -1,19 +1,25 @@
 from crewai import Agent
-from langchain.llms import OpenAI
 from langchain.llms import HuggingFaceEndpoint
-
 import os
 from dotenv import load_dotenv
+from tools.image_generation_tools import ImageGenerationTools
+
 load_dotenv()
 secret_token = os.getenv("content")
-#from tools.browser_tools import BrowserTools
-#from tools.search_tools import SearchTools
+
 repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
-mistral = HuggingFaceEndpoint(repo_id=repo_id, huggingfacehub_api_token = secret_token, max_new_tokens=28000,temperature=0.3,repetition_penalty=1.1)
+mistral = HuggingFaceEndpoint(repo_id=repo_id, huggingfacehub_api_token=secret_token, max_new_tokens=28000, temperature=0.3, repetition_penalty=1.1)
 
 class ContentAgents:
     def __init__(self):
-            self.llm = mistral
+        self.llm = mistral
+        self.image_generator = ImageGenerationTools()
+
+    def generate_image_for_section(self, section_title, topic):
+        prompt = f"{section_title} illustration for {topic}"
+        save_path = f"images/{section_title.replace(' ', '_')}.png"
+        self.image_generator.generate_image(prompt, save_path)
+        return save_path
 
     def planner_agent(self, topic):
         return Agent(