Update app.py

app.py

@@ -68,7 +68,7 @@ def fetch_company_info(company_url):
         logging.error(f"Exception during company info fetch: {e}")
         return None
 
-# Function to structure the email dynamically based on inputs and fetched data
+# Function to structure the email dynamically with fallback for missing data
 def structure_email(user_data, linkedin_info, company_info):
     linkedin_role = sanitize_data(linkedin_info.get('current_role', user_data['role']))
     linkedin_skills = sanitize_data(linkedin_info.get('skills', 'relevant skills'))
@@ -77,6 +77,18 @@ def structure_email(user_data, linkedin_info, company_info):
     company_mission = sanitize_data(company_info.get('mission', f"{company_name}'s mission"))
     company_goal = sanitize_data(company_info.get('goal', 'achieving excellence'))
 
+    # If essential data is missing, fill with defaults to ensure email has some content
+    if not linkedin_role:
+        linkedin_role = user_data['role']
+    if not linkedin_skills:
+        linkedin_skills = "skills relevant to this position"
+    if not linkedin_industry:
+        linkedin_industry = "the industry"
+    if not company_mission:
+        company_mission = f"{company_name}'s mission"
+    if not company_goal:
+        company_goal = "the company's goals"
+
     # Construct the email with fully sanitized and available data
     email_body = (
         f"Dear Hiring Manager,\n\n"
@@ -92,11 +104,11 @@ def structure_email(user_data, linkedin_info, company_info):
 
     return email_body
 
-# Function to validate the generated email based on critical components
+# Function to validate the generated email based on critical components with improved flexibility
 def validate_email(email_content, user_data):
     logging.info("Validating email content...")
 
-    # Validate the presence of essential details in the email
+    # Basic components we want to check in the email
     required_keywords = [
         user_data['name'],
         user_data['role'],
@@ -105,8 +117,16 @@ def validate_email(email_content, user_data):
         "contribute",
         "Best regards"
     ]
-
-    return all(keyword in email_content for keyword in required_keywords)
+    
+    # Check if the email contains all the required elements, allow some flexibility
+    missing_elements = [keyword for keyword in required_keywords if keyword.lower() not in email_content.lower()]
+
+    if missing_elements:
+        logging.info(f"Missing elements: {missing_elements}")
+        return False
+    else:
+        logging.info("Email content validation passed.")
+        return True
 
 # Custom Agent class following ReAct pattern
 class Agent: