Cleaned up

functions/__init__.py

@@ -5,6 +5,6 @@ This package contains modules for data acquisition, processing, and analysis
 of LinkedIn profiles, GitHub profiles, and job postings.
 """
 
-from .data_acquisition import get_linkedin_profile_html
+from .context_acquisition import get_linkedin_profile_html
 
 __all__ = ['get_linkedin_profile_html']

functions/context_acquisition.py

@@ -5,14 +5,15 @@ Functions for acquiring context from various sources including LinkedIn profiles
 GitHub profiles, and job postings using browser automation.
 """
 
+import time
+import logging
+
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException, WebDriverException
-import time
-import logging
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -35,14 +36,14 @@ def get_linkedin_profile_html(profile_url: str, wait_time: int = 10) -> str:
         WebDriverException: If there's an issue with the browser automation
         TimeoutException: If the page takes too long to load
     """
-    
+
     # Validate LinkedIn URL
     if not profile_url or not isinstance(profile_url, str):
         raise ValueError("Profile URL must be a non-empty string")
-    
+
     if "linkedin.com/in/" not in profile_url:
         raise ValueError("URL must be a valid LinkedIn profile URL (containing 'linkedin.com/in/')")
-    
+
     # Configure Chrome options for headless browsing
     chrome_options = Options()
     chrome_options.add_argument("--headless")  # Run in background
@@ -50,53 +51,66 @@ def get_linkedin_profile_html(profile_url: str, wait_time: int = 10) -> str:
     chrome_options.add_argument("--disable-dev-shm-usage")
     chrome_options.add_argument("--disable-gpu")
     chrome_options.add_argument("--window-size=1920,1080")
-    chrome_options.add_argument("--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
-    
+    chrome_options.add_argument("--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) " +
+        "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
+
     driver = None
     try:
         # Initialize the Chrome driver
         logger.info("Initializing browser for URL: %s", profile_url)
         driver = webdriver.Chrome(options=chrome_options)
         driver.set_page_load_timeout(30)
-        
+
         # Navigate to the LinkedIn profile
         logger.info("Navigating to LinkedIn profile...")
         driver.get(profile_url)
-        
+
         # Wait for the page to load
         # Look for common LinkedIn profile elements
         wait = WebDriverWait(driver, wait_time)
-        
+
         try:
             # Wait for either the main content or login prompt
             wait.until(
                 EC.any_of(
-                    EC.presence_of_element_located((By.CSS_SELECTOR, ".pv-top-card")),  # Profile header
-                    EC.presence_of_element_located((By.CSS_SELECTOR, ".profile-section")),  # Profile section
-                    EC.presence_of_element_located((By.CSS_SELECTOR, ".authwall")),  # Auth wall
-                    EC.presence_of_element_located((By.CSS_SELECTOR, ".public-profile")),  # Public profile
+                    EC.presence_of_element_located(( # Profile header
+                        By.CSS_SELECTOR,
+                        ".pv-top-card"
+                    )),
+                    EC.presence_of_element_located(( # Profile section
+                        By.CSS_SELECTOR,
+                        ".profile-section"
+                    )),
+                    EC.presence_of_element_located(( # Auth wall
+                        By.CSS_SELECTOR,
+                        ".authwall"
+                    )),
+                    EC.presence_of_element_located(( # Public profile
+                        By.CSS_SELECTOR,
+                        ".public-profile"
+                    )),
                 )
             )
         except TimeoutException:
             logger.warning("Standard LinkedIn elements not found, proceeding with current page state")
-        
+
         # Additional wait to ensure dynamic content loads
         time.sleep(2)
-        
+
         # Get the page HTML
         html_content = driver.page_source
-        
+
         logger.info("Successfully retrieved HTML content (%d characters)", len(html_content))
         return html_content
-        
+
     except WebDriverException as e:
         logger.error("WebDriver error occurred: %s", str(e))
         raise WebDriverException(f"Browser automation failed: {str(e)}") from e
-    
+
     except Exception as e:
         logger.error("Unexpected error occurred: %s", str(e))
         raise RuntimeError(f"Failed to retrieve LinkedIn profile: {str(e)}") from e
-    
+
     finally:
         # Always clean up the driver
         if driver:
@@ -123,5 +137,5 @@ def setup_chrome_driver_options() -> Options:
     chrome_options.add_argument("--disable-blink-features=AutomationControlled")
     chrome_options.add_experimental_option("excludeSwitches", ["enable-automation"])
     chrome_options.add_experimental_option('useAutomationExtension', False)
-    
+
     return chrome_options

resumate.py

@@ -37,7 +37,7 @@ def process_inputs(linkedin_url, github_url, job_post_url):
         try:
             result += "Attempting to retrieve LinkedIn profile...\n"
             html_content = get_linkedin_profile_html(linkedin_url)
-            result += f"✅ Successfully retrieved LinkedIn profile HTML ({len(html_content)} characters)\n"
+            result += f"LinkedIn profile HTML ({len(html_content)} characters)\n"
         except Exception as e: # pylint: disable=broad-exception-caught
             result += f"❌ Failed to retrieve LinkedIn profile: {str(e)}\n"