Update app.py

app.py

@@ -214,7 +214,11 @@ def extract_domain(url):
     """Extract domain from URL"""
     try:
         parsed = urlparse(url)
-        return parsed.netlify.lower()
+        domain = parsed.netloc.lower()
+        # Remove www. if present
+        if domain.startswith('www.'):
+            domain = domain[4:]
+        return domain
     except:
         return ""
 
@@ -224,26 +228,30 @@ def find_contacts_by_website(website_url):
     if not target_domain:
         return []
     
+    print(f"Searching for domain: {target_domain}")  # Debug info
+    
     matching_contacts = []
     for contact in CONTACTS_DB:
         contact_domain = extract_domain(contact['website'])
-        if target_domain in contact_domain or contact_domain in target_domain:
+        print(f"Comparing with: {contact_domain}")  # Debug info
+        
+        # Exact domain match or subdomain match
+        if target_domain == contact_domain or target_domain in contact_domain or contact_domain in target_domain:
             matching_contacts.append(contact)
     
+    print(f"Found {len(matching_contacts)} matching contacts")  # Debug info
     return matching_contacts
 
 def simulate_website_scraping(url):
     """Simulate scraping a website and finding contact information"""
     # Add some delay to simulate real scraping
-    time.sleep(random.uniform(1, 3))
+    time.sleep(random.uniform(1, 2))
     
     # Find matching contacts from our database
     contacts = find_contacts_by_website(url)
     
-    if not contacts:
-        # If no exact match, return a random subset as if found on the website
-        contacts = random.sample(CONTACTS_DB, min(3, len(CONTACTS_DB)))
-    
+    # Only return contacts if we found exact matches
+    # Don't return random contacts if no match found
     return contacts
 
 def search_website_contacts(website_url, max_results=10):
@@ -260,7 +268,7 @@ def search_website_contacts(website_url, max_results=10):
         contacts = simulate_website_scraping(website_url)
         
         if not contacts:
-            return "No contacts found on this website.", ""
+            return f"No contacts found on {website_url}. \n\nThis website is not in our contact database. Try one of the sample websites listed below, or the website might not have publicly available contact information.", ""
         
         # Limit results
         contacts = contacts[:max_results]