Perform Sales Lead Enrichment

import os, requests, csv
API_KEY = os.getenv("LEADX_API_KEY")
BASE_URL = "https://api.leadx.com/v1"
headers = {"X-API-KEY": API_KEY, "Content-Type": "application/json"}

# Inline CSV data for demonstration; adjust to import a CSV
csv_data = [
    {"linkedin_url": "https://www.linkedin.com/in/john-doe"},
    {"linkedin_url": "https://www.linkedin.com/in/jane-doe"}
]

output_csv = "sales_leads.csv"

def get_mobile_number(linkedin_url):
    """Fetch mobile number using the /mobile-numbers/enrich endpoint."""
    try:
        payload = {"linkedin_url": linkedin_url}
        response = requests.post(f"{BASE_URL}/mobile-numbers/enrich", headers=HEADERS, json=payload)
        response.raise_for_status()  
        data = response.json()
        return data.get("mobile_number", "N/A")
    except requests.exceptions.RequestException as e:
        print(f"Error fetching mobile number for {linkedin_url}: {e}")
        return "Error"

def get_contact_details(linkedin_url):
    """Fetch contact details using the /contacts endpoint."""
    try:
        payload = {"linkedin_url": linkedin_url}
        response = requests.post(f"{BASE_URL}/contacts", headers=HEADERS, json=payload)
        response.raise_for_status()  
        if data.get("records"):
            contact = data["records"][0]
            return {
                "name": contact.get("name", "N/A"),
                "title": contact.get("title", "N/A"),
                "company": contact.get("company", "N/A"),
                "location": contact.get("location", "N/A"),
                "employment_timeline": contact.get("employment_timeline", "N/A"),
            }
        return {}
    except requests.exceptions.RequestException as e:
        print(f"Error fetching contact details for {linkedin_url}: {e}")
        return {}

def process_csv(data, output_csv):
    """Process the inline CSV data and generate an output CSV with enriched data."""
    try:
        with open(output_csv, "w", newline="") as outfile:
            fieldnames = [
                "linkedin_url", "mobile_number", "name", "title", "company", "location", "employment_timeline"
            ]
            writer = csv.DictWriter(outfile, fieldnames=fieldnames)
            writer.writeheader()

            for row in data:
                linkedin_url = row["linkedin_url"]
                print(f"Processing: {linkedin_url}")

                # Fetch data
                mobile_number = get_mobile_number(linkedin_url)
                contact_details = get_contact_details(linkedin_url)

                # Write to output CSV
                writer.writerow({
                    "linkedin_url": linkedin_url,
                    "mobile_number": mobile_number,
                    "name": contact_details.get("name", "N/A"),
                    "title": contact_details.get("title", "N/A"),
                    "company": contact_details.get("company", "N/A"),
                    "location": contact_details.get("location", "N/A"),
                    "employment_timeline": contact_details.get("employment_timeline", "N/A"),
                })
    except FileNotFoundError as e:
        print(f"Error writing to file {output_csv}: {e}")
    except IOError as e:
        print(f"File I/O error: {e}")

if __name__ == "__main__":
    try:
        process_csv(csv_data, output_csv)
        print(f"Sales leads enriched and saved to {output_csv}")
    except Exception as e:
        print(f"An unexpected error occurred: {e}")