Update agents.py

agents.py

@@ -12,13 +12,17 @@ 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)
 
 from langchain_groq import ChatGroq
+# Initialize the Groq LLM
+groq_llm = ChatGroq(temperature=0.6, groq_api_key=os.environ.get("groq_api_key"), model_name="llama3-8b-8192")
+
 # Initialize the Tools
+image_generation_tools = ImageGenerationTools()
 web_search_tools = WebSearchTools()
 
 class ContentAgents:
     def __init__(self):
-        self.llm = ChatGroq(temperature=0.6, groq_api_key=os.environ.get("groq_api_key"), model_name="llama3-8b-8192")
-        self.image_generator = ImageGenerationTools()
+        self.llm = groq_llm
+        self.image_generator = image_generation_tools
 
     def generate_image_for_section(self, section_title, topic):
         prompt = f"{section_title} illustration for {topic}"