Update app.py

app.py

@@ -80,8 +80,10 @@ class Agent:
             company_info = fetch_company_info(self.user_data['company_url'])
             return linkedin_info, company_info
         elif self.name == "Email Generation Agent":
-            linkedin_info, company_info = self.user_data
-            prompt = structure_email(self.user_data[0], linkedin_info, company_info)
+            user_data = self.user_data['user_data']
+            linkedin_info = self.user_data['linkedin_info']
+            company_info = self.user_data['company_info']
+            prompt = structure_email(user_data, linkedin_info, company_info)
             email_content = generate_email_content(OPENAI_API_KEY, prompt)
             return email_content
 
@@ -122,8 +124,15 @@ def run_agent(name, email, phone, linkedin_url, company_url, role):
     if isinstance(linkedin_info, str):  # If an error message is returned
         return linkedin_info
 
+    # Create a structured dictionary for the Email Generation Agent
+    agent_data = {
+        "user_data": user_data,
+        "linkedin_info": linkedin_info,
+        "company_info": company_info
+    }
+
     # Pass the collected data to the Email Generation Agent
-    email_agent = Agent("Email Generation Agent", "Generate the email content", (user_data, linkedin_info, company_info))
+    email_agent = Agent("Email Generation Agent", "Generate the email content", agent_data)
     email_content = email_agent.act()
 
     # Validate and refine the email using a ReAct pattern with a maximum of 3 iterations