Update app.py

app.py

@@ -19,6 +19,7 @@ class AutonomousEmailAgent:
         self.company_info = None
         self.role_description = None
         self.company_url = None
+        self.attempts = 0  # Counter for iterations
 
     # Fetch LinkedIn data via Proxycurl
     def fetch_linkedin_data(self):
@@ -118,6 +119,10 @@ class AutonomousEmailAgent:
         instruction = reasoning_output.lower().strip()
 
         if "scrape" in instruction:
+            if self.attempts >= 5:
+                print("Max attempts reached. Proceeding with fallback option.")
+                return self.generate_fallback_email()
+            self.attempts += 1
             self.fetch_company_url()
             if self.company_url:
                 self.fetch_company_info_with_firecrawl(self.company_url)
@@ -126,7 +131,7 @@ class AutonomousEmailAgent:
         elif "generate_email" in instruction:
             return self.generate_email()
 
-        elif "fallback" in instruction:
+        elif "fallback" in instruction or self.attempts >= 5:
             print("Action: Using fallback values for missing data.")
             if not self.company_info:
                 self.company_info = "A leading company in its field."
@@ -138,7 +143,7 @@ class AutonomousEmailAgent:
             print("Error: Unrecognized instruction from LLM. Proceeding with available data.")
             return self.generate_email()
 
-    # Fetch company URL using SERP API (same as before)
+    # Fetch company URL using SERP API
     def fetch_company_url(self):
         serp_api_key = os.getenv("SERP_API_KEY")
         print(f"Fetching company URL for {self.company_name} using SERP API...")
@@ -156,7 +161,7 @@ class AutonomousEmailAgent:
         else:
             print(f"Error fetching company URL: {response.status_code}")
 
-    # Fetch company information via Firecrawl API using company URL (same as before)
+    # Fetch company information via Firecrawl API using company URL
     def fetch_company_info_with_firecrawl(self, company_url):
         firecrawl_api_key = os.getenv("FIRECRAWL_API_KEY")
         print(f"Fetching company info for {company_url} using Firecrawl.")
@@ -173,7 +178,7 @@ class AutonomousEmailAgent:
             print(f"Error: Unable to fetch company info via Firecrawl. Status code: {response.status_code}")
             self.company_info = "A leading company in its field."
 
-    # Final Action: Generate the email using Groq Cloud LLM (same as before)
+    # Final Action: Generate the email using Groq Cloud LLM
     def generate_email(self):
         print("Action: Generating the email using Groq Cloud LLM with the gathered information.")