Update app.py

app.py

@@ -1,628 +1,340 @@
 import gradio as gr
 import requests
-import re
 import json
-import csv
-import io
-from urllib.parse import urlparse, urljoin
-import time
-import random
+from typing import List, Dict, Any
+import pandas as pd
 
-def extract_domain(url):
-    """Extract domain from URL"""
-    try:
-        if not url.startswith(('http://', 'https://')):
-            url = 'https://' + url
-        parsed = urlparse(url)
-        domain = parsed.netloc.lower()
-        # Remove www. if present
-        if domain.startswith('www.'):
-            domain = domain[4:]
-        return domain
-    except:
-        return ""
+# Professional color scheme inspired by corporate/real estate companies
+theme = gr.themes.Soft(
+    primary_hue="blue",
+    secondary_hue="slate",
+    neutral_hue="slate",
+    text_size="md",
+    spacing_size="md",
+    radius_size="md"
+).set(
+    body_background_fill="#f8fafc",
+    body_background_fill_dark="#1e293b",
+    block_background_fill="#ffffff",
+    block_background_fill_dark="#334155",
+    block_border_color="#e2e8f0",
+    block_border_color_dark="#475569",
+    input_background_fill="#ffffff",
+    input_background_fill_dark="#334155",
+    input_border_color="#cbd5e1",
+    input_border_color_dark="#64748b",
+    button_primary_background_fill="#1e40af",
+    button_primary_background_fill_hover="#1d4ed8",
+    button_primary_text_color="#ffffff",
+    button_secondary_background_fill="#f1f5f9",
+    button_secondary_background_fill_hover="#e2e8f0",
+    button_secondary_text_color="#334155"
+)
 
-def simulate_website_scraping(url):
-    """Fetch contacts using your Hugging Face Space API"""
-    try:
-        response = requests.get(
-            "https://jaybene1-testapicontacts.hf.space/contacts",
-            params={"website": url},
-            timeout=10
-        )
-        response.raise_for_status()
-        data = response.json()
-        
-        # Wrap in list if it's a single dict
-        if isinstance(data, dict):
-            data = [data]
-        
-        # Normalize structure to match the app's expectations
-        contacts = []
-        for item in data:
-            contact = {
-                "first_name": item.get("name", "").split()[0],
-                "last_name": " ".join(item.get("name", "").split()[1:]),
-                "email": item.get("email", ""),
-                "phone": item.get("phone", ""),
-                "job_title": item.get("position", ""),
-                "company": item.get("company", ""),
-                "website": item.get("website", "")
-            }
-            contacts.append(contact)
-        
-        return contacts
+def search_contacts(url: str, api_endpoint: str = None) -> tuple[str, str]:
+    """
+    Search for contacts associated with a given URL using the Hugging Face API.
     
-    except Exception as e:
-        print(f"Error contacting API for {url}: {e}")
-        return []
-
-def is_valid_url(url):
-    """Check if a string looks like a valid URL"""
-    url = url.strip()
-    if not url:
-        return False
+    Args:
+        url: The website URL to search for contacts
+        api_endpoint: Your Hugging Face API endpoint URL
+        
+    Returns:
+        Tuple of (formatted_results, raw_json)
+    """
     
-    # Check if it contains common URL patterns
-    url_patterns = [
-        r'^\w+\.\w+',  # domain.com
-        r'^\w+\.\w+\.\w+',  # subdomain.domain.com
-        r'^https?://',  # starts with http/https
-        r'www\.',  # contains www
-    ]
+    if not url.strip():
+        return "❌ Please enter a website URL", ""
     
-    for pattern in url_patterns:
-        if re.search(pattern, url.lower()):
-            return True
+    if not api_endpoint or not api_endpoint.strip():
+        return "❌ Please provide your Hugging Face API endpoint URL", ""
     
-    return False
-
-def parse_csv_file(file_obj):
-    """Parse CSV file and extract website URLs from column H (or auto-detect)"""
-    websites = []
-    debug_info = []
+    # Clean and validate URL
+    url = url.strip()
+    if not url.startswith(('http://', 'https://')):
+        url = 'https://' + url
     
     try:
-        # Handle different file input types
-        if hasattr(file_obj, 'name'):
-            # This is a file path (NamedString from Gradio)
-            with open(file_obj.name, 'r', encoding='utf-8') as f:
-                content = f.read()
-        elif isinstance(file_obj, str):
-            # This is already a string
-            content = file_obj
-        else:
-            # This might be bytes
-            content = file_obj.decode('utf-8')
-        
-        # Parse CSV
-        csv_reader = csv.reader(io.StringIO(content))
-        rows = list(csv_reader)
-        
-        if not rows:
-            debug_info.append("CSV file is empty")
-            return [], debug_info
-        
-        debug_info.append(f"Total rows in CSV: {len(rows)}")
-        
-        # Analyze the first few rows to understand structure
-        if len(rows) > 0:
-            debug_info.append(f"First row has {len(rows[0])} columns: {rows[0]}")
-        if len(rows) > 1:
-            debug_info.append(f"Second row has {len(rows[1])} columns: {rows[1]}")
-        
-        # Check if first row looks like headers
-        first_row = rows[0]
-        has_headers = any(col.lower() in ['website', 'url', 'domain', 'site', 'web'] for col in first_row)
-        
-        # Try to find website column
-        website_column_index = None
+        # Prepare the API request
+        headers = {
+            'Content-Type': 'application/json',
+            'Authorization': f'Bearer {api_endpoint.split("/")[-1]}'  # Adjust based on your API auth
+        }
+        
+        payload = {
+            "inputs": url,
+            "parameters": {
+                "task": "contact_search",
+                "url": url
+            }
+        }
+        
+        # Make the API request
+        response = requests.post(
+            api_endpoint,
+            headers=headers,
+            json=payload,
+            timeout=30
+        )
         
-        # First, try column H (index 7)
-        if len(first_row) >= 8:
-            website_column_index = 7
-            debug_info.append(f"Using column H (index 7) as specified")
+        if response.status_code == 200:
+            try:
+                result = response.json()
+                
+                # Format the results for display
+                formatted_output = format_contact_results(result, url)
+                raw_json = json.dumps(result, indent=2)
+                
+                return formatted_output, raw_json
+                
+            except json.JSONDecodeError:
+                return f"❌ Error: Invalid JSON response from API", response.text
+                
         else:
-            # Auto-detect website column
-            for i, col in enumerate(first_row):
-                if col.lower() in ['website', 'url', 'domain', 'site', 'web']:
-                    website_column_index = i
-                    debug_info.append(f"Auto-detected website column at index {i}: '{col}'")
-                    break
-        
-        if website_column_index is None:
-            # If no obvious column found, scan all columns for URLs
-            debug_info.append("No obvious website column found, scanning all columns for URLs...")
-            for row_idx, row in enumerate(rows[1:] if has_headers else rows, start=1):
-                for col_idx, cell in enumerate(row):
-                    if is_valid_url(cell):
-                        website_column_index = col_idx
-                        debug_info.append(f"Found URLs in column {col_idx} (row {row_idx}): '{cell}'")
-                        break
-                if website_column_index is not None:
-                    break
-        
-        if website_column_index is None:
-            debug_info.append("ERROR: Could not find any column with website URLs")
-            return [], debug_info
-        
-        # Extract websites from the identified column
-        start_row = 1 if has_headers else 0
-        for row_idx, row in enumerate(rows[start_row:], start=start_row + 1):
-            if len(row) > website_column_index:
-                website_url = row[website_column_index].strip()
-                if website_url and is_valid_url(website_url):
-                    websites.append(website_url)
-                    debug_info.append(f"Found website in row {row_idx}: {website_url}")
-                elif website_url:
-                    debug_info.append(f"Row {row_idx}: '{website_url}' doesn't look like a valid URL")
-                else:
-                    debug_info.append(f"Row {row_idx}: Empty cell in website column")
-            else:
-                debug_info.append(f"Row {row_idx}: Has only {len(row)} columns, need at least {website_column_index + 1}")
-        
-        debug_info.append(f"Total websites extracted: {len(websites)}")
-        if websites:
-            debug_info.append(f"Sample websites: {websites[:5]}")
-        
-        return websites, debug_info
-        
+            return f"❌ API Error ({response.status_code}): {response.text}", ""
+            
+    except requests.exceptions.Timeout:
+        return "❌ Request timeout. Please try again.", ""
+    except requests.exceptions.ConnectionError:
+        return "❌ Connection error. Please check your API endpoint.", ""
     except Exception as e:
-        debug_info.append(f"Error parsing CSV: {e}")
-        return [], debug_info
+        return f"❌ Error: {str(e)}", ""
 
-def search_csv_websites(csv_file, max_results=10):
-    """Search for contacts from websites listed in CSV file"""
-    if csv_file is None:
-        return "Please upload a CSV file", ""
+def format_contact_results(results: Dict[Any, Any], url: str) -> str:
+    """
+    Format the API results into a readable format.
     
-    try:
-        # Parse CSV file
-        websites, debug_info = parse_csv_file(csv_file)
-        
-        debug_text = "\n".join(debug_info)
-        
-        if not websites:
-            error_msg = "No websites found in the CSV file.\n\n"
-            error_msg += "DEBUG INFORMATION:\n" + debug_text + "\n\n"
-            error_msg += "TROUBLESHOOTING:\n"
-            error_msg += "1. Ensure your CSV has website URLs in column H (8th column)\n"
-            error_msg += "2. Or have a column header named 'website', 'url', 'domain', etc.\n"
-            error_msg += "3. Check that URLs are properly formatted (e.g., example.com or https://example.com)\n"
-            error_msg += "4. Verify the CSV file is not corrupted\n"
-            return error_msg, ""
-        
-        all_contacts = []
-        processed_websites = []
-        
-        # Search each website
-        for website in websites[:20]:  # Limit to first 20 websites
-            print(f"Processing website: {website}")
-            contacts = simulate_website_scraping(website)
-            if contacts:
-                all_contacts.extend(contacts)
-                processed_websites.append(website)
-                print(f"Found {len(contacts)} contacts for {website}")
-            else:
-                print(f"No contacts found for {website}")
-        
-        # Remove duplicates based on email
-        unique_contacts = []
-        seen_emails = set()
-        for contact in all_contacts:
-            if contact['email'] not in seen_emails:
-                unique_contacts.append(contact)
-                seen_emails.add(contact['email'])
-        
-        # Limit results
-        unique_contacts = unique_contacts[:max_results]
+    Args:
+        results: The JSON response from the API
+        url: The searched URL
         
-        if not unique_contacts:
-            result_msg = f"No contacts found for the {len(websites)} websites from the CSV file.\n\n"
-            result_msg += "DEBUG INFORMATION:\n" + debug_text + "\n\n"
-            result_msg += f"Websites processed: {', '.join(websites[:10])}\n"
-            result_msg += "This might be because the websites are not in our sample database."
-            return result_msg, ""
-        
-        # Format results
-        results_text = f"CONTACT DISCOVERY REPORT\n"
-        results_text += f"CSV Processing Details:\n"
-        results_text += f"Total Websites in CSV: {len(websites)}\n"
-        results_text += f"Websites Processed: {len(processed_websites)}\n"
-        results_text += f"Websites with Contacts: {len(processed_websites)}\n"
-        results_text += f"Unique Contacts Found: {len(unique_contacts)}\n"
-        results_text += f"Processed Websites: {', '.join(processed_websites)}\n"
-        results_text += f"{'='*60}\n\n"
-        
-        # Add debug info
-        results_text += "DEBUG INFORMATION:\n" + debug_text + "\n\n"
-        results_text += f"{'='*60}\n\n"
-        
-        for i, contact in enumerate(unique_contacts, 1):
-            results_text += f"CONTACT #{i}\n"
-            results_text += f"Name: {contact['first_name']} {contact['last_name']}\n"
-            results_text += f"Position: {contact['job_title']}\n"
-            results_text += f"Email: {contact['email']}\n"
-            results_text += f"Phone: {contact['phone']}\n"
-            results_text += f"Company: {contact['company']}\n"
-            results_text += f"Website: {contact['website']}\n\n"
-        
-        # Create CSV output
-        csv_output = "First Name,Last Name,Job Title,Email,Phone,Company,Website\n"
-        for contact in unique_contacts:
-            csv_output += f"{contact['first_name']},{contact['last_name']},{contact['job_title']},{contact['email']},{contact['phone']},{contact['company']},{contact['website']}\n"
-        
-        return results_text, csv_output
-        
-    except Exception as e:
-        return f"Error processing CSV file: {str(e)}", ""
-
-def search_website_contacts(website_url, max_results=10):
-    """Main function to search for contacts on a website"""
-    if not website_url:
-        return "Please enter a website URL", ""
+    Returns:
+        Formatted string with contact information
+    """
     
-    # Clean up URL
-    if not website_url.startswith(('http://', 'https://')):
-        website_url = 'https://' + website_url
+    output = f"# 🔍 Contact Search Results for: {url}\n\n"
     
-    try:
-        # Simulate finding contacts
-        contacts = simulate_website_scraping(website_url)
-        
-        if not contacts:
-            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]
-        
-        # Format results
-        results_text = f"CONTACT INTELLIGENCE REPORT\n"
-        results_text += f"Website: {website_url}\n"
-        results_text += f"Contacts Found: {len(contacts)}\n"
-        results_text += f"{'='*60}\n\n"
-        
-        for i, contact in enumerate(contacts, 1):
-            results_text += f"CONTACT #{i}\n"
-            results_text += f"First Name: {contact['first_name']}\n"
-            results_text += f"Last Name: {contact['last_name']}\n"
-            results_text += f"Position: {contact['job_title']}\n"
-            results_text += f"Email: {contact['email']}\n"
-            results_text += f"Phone: {contact['phone']}\n"
-            results_text += f"Company: {contact['company']}\n\n"
-        
-        # Create a simple table format for the second output
-        table_text = "First Name,Last Name,Job Title,Email,Phone,Company\n"
-        for contact in contacts:
-            table_text += f"{contact['first_name']},{contact['last_name']},{contact['job_title']},{contact['email']},{contact['phone']},{contact['company']}\n"
-        
-        return results_text, table_text
-        
-    except Exception as e:
-        return f"Error searching website: {str(e)}", ""
-
-def get_all_available_websites():
-    """Return a static list of example websites for display"""
-    return "\n".join([
-        "kwekelcompanies.com",
-        "techflowsolutions.com",
-        "greenleafconsult.com",
-        "blueskymarketing.net",
-        "quantumdynamics.org",
-        "stellarlogistics.biz",
-        "nexusfinancial.pro",
-        "horizonhealth.care",
-        "phoenixmfg.com"
-    ])
-
-# Custom CSS
-custom_css = """
-.gradio-container {
-    background: linear-gradient(135deg, #f8f9fa 0%, #e9ecef 100%);
-    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-}
-
-.main-header {
-    background: linear-gradient(135deg, #1e3a8a 0%, #3b82f6 50%, #1e40af 100%);
-    color: white;
-    padding: 40px 20px;
-    text-align: center;
-    border-radius: 15px;
-    margin-bottom: 30px;
-    box-shadow: 0 10px 30px rgba(30, 58, 138, 0.3);
-}
-
-.main-header h1 {
-    font-size: 2.5em;
-    margin: 0;
-    font-weight: 700;
-    text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
-}
-
-.main-header p {
-    font-size: 1.2em;
-    margin: 15px 0 0 0;
-    opacity: 0.9;
-}
-
-.corporate-card {
-    background: white;
-    border: 1px solid #d1d5db;
-    border-radius: 12px;
-    padding: 25px;
-    margin: 15px 0;
-    box-shadow: 0 4px 15px rgba(0,0,0,0.1);
-    border-left: 5px solid #1e40af;
-}
-
-.tips-section {
-    background: linear-gradient(135deg, #f1f5f9 0%, #e2e8f0 100%);
-    border: 2px solid #cbd5e1;
-    border-radius: 15px;
-    padding: 20px;
-    margin: 10px 0;
-}
-
-.tips-section h3 {
-    color: #1e40af;
-    margin-top: 0;
-    font-weight: 600;
-}
-
-.primary-btn {
-    background: linear-gradient(135deg, #1e40af 0%, #3b82f6 100%);
-    color: white;
-    border: none;
-    border-radius: 8px;
-    padding: 15px 30px;
-    font-weight: 600;
-    font-size: 16px;
-    transition: all 0.3s ease;
-}
+    # Handle different possible result structures
+    if isinstance(results, dict):
+        if 'contacts' in results:
+            contacts = results['contacts']
+        elif 'results' in results:
+            contacts = results['results']
+        elif 'data' in results:
+            contacts = results['data']
+        else:
+            contacts = results
+            
+        if isinstance(contacts, list) and len(contacts) > 0:
+            output += f"**Found {len(contacts)} contact(s):**\n\n"
+            
+            for i, contact in enumerate(contacts, 1):
+                output += f"## 👤 Contact {i}\n"
+                
+                # Handle different contact field names
+                name = contact.get('name') or contact.get('full_name') or contact.get('contact_name') or "N/A"
+                email = contact.get('email') or contact.get('email_address') or "N/A"
+                title = contact.get('title') or contact.get('job_title') or contact.get('position') or "N/A"
+                company = contact.get('company') or contact.get('organization') or "N/A"
+                phone = contact.get('phone') or contact.get('phone_number') or "N/A"
+                linkedin = contact.get('linkedin') or contact.get('linkedin_url') or "N/A"
+                
+                output += f"- **Name:** {name}\n"
+                output += f"- **Email:** {email}\n"
+                output += f"- **Title:** {title}\n"
+                output += f"- **Company:** {company}\n"
+                output += f"- **Phone:** {phone}\n"
+                output += f"- **LinkedIn:** {linkedin}\n\n"
+                
+                # Add any additional fields
+                additional_fields = {k: v for k, v in contact.items() 
+                                   if k not in ['name', 'full_name', 'contact_name', 'email', 'email_address', 
+                                              'title', 'job_title', 'position', 'company', 'organization', 
+                                              'phone', 'phone_number', 'linkedin', 'linkedin_url']}
+                
+                if additional_fields:
+                    output += "**Additional Information:**\n"
+                    for key, value in additional_fields.items():
+                        output += f"- **{key.replace('_', ' ').title()}:** {value}\n"
+                    output += "\n"
+                    
+                output += "---\n\n"
+                
+        else:
+            output += "❌ No contacts found for this URL.\n\n"
+            
+    elif isinstance(results, list):
+        if len(results) > 0:
+            output += f"**Found {len(results)} contact(s):**\n\n"
+            for i, contact in enumerate(results, 1):
+                output += f"## 👤 Contact {i}\n"
+                output += f"{contact}\n\n"
+        else:
+            output += "❌ No contacts found for this URL.\n\n"
+    else:
+        output += f"**Result:** {results}\n\n"
+    
+    output += f"*Search completed at: {pd.Timestamp.now().strftime('%Y-%m-%d %H:%M:%S')}*"
+    
+    return output
 
-.primary-btn:hover {
-    background: linear-gradient(135deg, #1e3a8a 0%, #2563eb 100%);
-    transform: translateY(-2px);
-    box-shadow: 0 6px 20px rgba(30, 64, 175, 0.4);
-}
+def create_sample_data():
+    """Create sample data for demonstration"""
+    return '''# 🔍 Contact Search Results for: example.com
 
-.secondary-btn {
-    background: white;
-    color: #374151;
-    border: 2px solid #d1d5db;
-    border-radius: 6px;
-    padding: 8px 16px;
-    font-weight: 500;
-    transition: all 0.3s ease;
-}
+**Found 2 contact(s):**
 
-.secondary-btn:hover {
-    border-color: #1e40af;
-    color: #1e40af;
-    background: #f8fafc;
-}
+## 👤 Contact 1
+- **Name:** John Smith
+- **Email:** john.smith@example.com
+- **Title:** CEO
+- **Company:** Example Corp
+- **Phone:** (555) 123-4567
+- **LinkedIn:** linkedin.com/in/johnsmith
 
-.custom-input {
-    border: 2px solid #d1d5db;
-    border-radius: 8px;
-    padding: 12px;
-    font-size: 16px;
-    transition: border-color 0.3s ease;
-}
+---
 
-.custom-input:focus {
-    border-color: #3b82f6;
-    box-shadow: 0 0 0 3px rgba(59, 130, 246, 0.1);
-}
+## 👤 Contact 2
+- **Name:** Jane Doe
+- **Email:** jane.doe@example.com
+- **Title:** VP of Sales
+- **Company:** Example Corp
+- **Phone:** (555) 123-4568
+- **LinkedIn:** linkedin.com/in/janedoe
 
-.results-container {
-    background: white;
-    border: 1px solid #e5e7eb;
-    border-radius: 10px;
-    padding: 20px;
-    margin: 15px 0;
-    box-shadow: 0 2px 10px rgba(0,0,0,0.05);
-}
+---
 
-.section-header {
-    background: linear-gradient(135deg, #64748b 0%, #475569 100%);
-    color: white;
-    padding: 15px 20px;
-    border-radius: 10px;
-    margin: 20px 0 15px 0;
-    font-weight: 600;
-}
-"""
+*Search completed at: 2025-01-15 10:30:00*'''
 
-# Create Gradio interface
-with gr.Blocks(css=custom_css, title="Contact Discovery Platform", theme=gr.themes.Base()) as app:
+# Create the Gradio interface
+with gr.Blocks(theme=theme, title="Contact Search - Kwekel Companies", css="""
+    .gradio-container {
+        max-width: 1200px !important;
+        margin: 0 auto !important;
+    }
+    .header {
+        background: linear-gradient(135deg, #1e40af 0%, #3b82f6 100%);
+        color: white;
+        padding: 2rem;
+        border-radius: 0.5rem;
+        margin-bottom: 2rem;
+        text-align: center;
+    }
+    .header h1 {
+        margin: 0;
+        font-size: 2.5rem;
+        font-weight: 700;
+        margin-bottom: 0.5rem;
+    }
+    .header p {
+        margin: 0;
+        font-size: 1.2rem;
+        opacity: 0.9;
+    }
+    .info-box {
+        background: #f0f9ff;
+        border: 1px solid #0ea5e9;
+        border-radius: 0.5rem;
+        padding: 1rem;
+        margin: 1rem 0;
+    }
+    .footer {
+        text-align: center;
+        margin-top: 2rem;
+        padding-top: 2rem;
+        border-top: 1px solid #e2e8f0;
+        color: #64748b;
+    }
+""") as demo:
+    
+    # Header
     gr.HTML("""
-    <div class="main-header">
-        <h1>Contact Discovery Platform</h1>
-        <p>Professional Contact Discovery & Lead Generation Tool</p>
-        <p style="font-size: 0.95em; opacity: 0.8;">Advanced website analysis for contact intelligence gathering</p>
+    <div class="header">
+        <h1>🔍 Contact Search Tool</h1>
+        <p>Professional contact discovery powered by AI</p>
     </div>
     """)
     
-    with gr.Tabs():
-        # Single Website Search Tab
-        with gr.TabItem("Single Website Search"):
-            with gr.Row():
-                with gr.Column(scale=2):
-                    gr.HTML('<div class="section-header">Search Parameters</div>')
-                    
-                    website_input = gr.Textbox(
-                        label="Target Website URL",
-                        placeholder="Enter company website (e.g., techflowsolutions.com)",
-                        value="",
-                        elem_classes=["custom-input"]
-                    )
-                    
-                    with gr.Row():
-                        max_results = gr.Slider(
-                            minimum=1,
-                            maximum=20,
-                            value=8,
-                            step=1,
-                            label="Maximum Results",
-                            elem_classes=["custom-input"]
-                        )
-                        
-                        search_btn = gr.Button(
-                            "Execute Search", 
-                            variant="primary", 
-                            size="lg",
-                            elem_classes=["primary-btn"]
-                        )
-            
-            gr.HTML('<div class="section-header">Search Results</div>')
-            
-            with gr.Row():
-                results_display = gr.Textbox(
-                    label="Contact Intelligence Report",
-                    lines=18,
-                    max_lines=35,
-                    show_copy_button=True,
-                    elem_classes=["results-container"]
-                )
-                
-                csv_output = gr.Textbox(
-                    label="Export Data (CSV Format)",
-                    lines=18,
-                    max_lines=35,
-                    show_copy_button=True,
-                    elem_classes=["results-container"]
-                )
-        
-        # CSV Upload Tab
-        with gr.TabItem("CSV Bulk Search"):
-            with gr.Row():
-                with gr.Column(scale=2):
-                    gr.HTML('<div class="section-header">CSV Upload</div>')
-                    
-                    csv_file = gr.File(
-                        label="Upload CSV File",
-                        file_types=[".csv"],
-                        elem_classes=["custom-input"]
-                    )
-                    
-                    gr.HTML("""
-                    <div style="background: #f8fafc; padding: 15px; border-radius: 8px; border-left: 4px solid #1e40af; margin: 10px 0;">
-                        <strong>CSV Format - Multiple Options:</strong><br>
-                        <strong>Option 1:</strong> Website URLs in <strong>Column H (8th column)</strong><br>
-                        <strong>Option 2:</strong> Column header named 'website', 'url', 'domain', 'site', or 'web'<br>
-                        <strong>Option 3:</strong> System will auto-detect columns with valid URLs<br><br>
-                        <strong>Examples:</strong> techflowsolutions.com, https://example.com, www.company.com<br>
-                        <strong>Note:</strong> The system will show detailed debugging information about your CSV structure
-                    </div>
-                    """)
-                    
-                    with gr.Row():
-                        csv_max_results = gr.Slider(
-                            minimum=1,
-                            maximum=50,
-                            value=20,
-                            step=1,
-                            label="Maximum Results",
-                            elem_classes=["custom-input"]
-                        )
-                        
-                        csv_search_btn = gr.Button(
-                            "Process CSV", 
-                            variant="primary", 
-                            size="lg",
-                            elem_classes=["primary-btn"]
-                        )
+    # Info box
+    gr.HTML("""
+    <div class="info-box">
+        <h3>📋 Instructions:</h3>
+        <ol>
+            <li>Enter your Hugging Face API endpoint URL in the first field</li>
+            <li>Enter the website URL you want to search for contacts</li>
+            <li>Click "Search Contacts" to get results</li>
+            <li>View formatted results and raw JSON response</li>
+        </ol>
+    </div>
+    """)
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            api_endpoint = gr.Textbox(
+                label="🔗 Hugging Face API Endpoint",
+                placeholder="https://your-model-endpoint.hf.space/api/v1/predict",
+                info="Enter your Hugging Face API endpoint URL",
+                lines=1
+            )
             
-            gr.HTML('<div class="section-header">CSV Results</div>')
+            url_input = gr.Textbox(
+                label="🌐 Website URL",
+                placeholder="example.com or https://example.com",
+                info="Enter the website URL to search for contacts",
+                lines=1
+            )
             
             with gr.Row():
-                csv_results_display = gr.Textbox(
-                    label="CSV Processing Report",
-                    lines=18,
-                    max_lines=35,
-                    show_copy_button=True,
-                    elem_classes=["results-container"]
-                )
-                
-                csv_export_output = gr.Textbox(
-                    label="Export Data (CSV Format)",
-                    lines=18,
-                    max_lines=35,
-                    show_copy_button=True,
-                    elem_classes=["results-container"]
-                )
-    
-    # Sample websites section
-    with gr.Accordion("Sample Websites Database", open=False):
-        gr.HTML('<div style="background: #f8fafc; padding: 15px; border-radius: 8px; border-left: 4px solid #1e40af;">')
-        sample_websites = gr.Textbox(
-            label="Available Websites in Database",
-            value=get_all_available_websites(),
-            lines=8,
-            interactive=False,
-            elem_classes=["custom-input"]
-        )
-        gr.HTML('</div>')
-    
-    # Quick search buttons
-    gr.HTML('<div class="section-header">Quick Access Sample Websites</div>')
+                search_btn = gr.Button("🔍 Search Contacts", variant="primary", scale=2)
+                demo_btn = gr.Button("📋 Show Demo", variant="secondary", scale=1)
+                clear_btn = gr.Button("🗑️ Clear", variant="secondary", scale=1)
     
     with gr.Row():
-        quick_btn1 = gr.Button("TechFlow Solutions", size="sm", elem_classes=["secondary-btn"])
-        quick_btn2 = gr.Button("GreenLeaf Consulting", size="sm", elem_classes=["secondary-btn"])
-        quick_btn3 = gr.Button("BlueSky Marketing", size="sm", elem_classes=["secondary-btn"])
-        quick_btn4 = gr.Button("Quantum Dynamics", size="sm", elem_classes=["secondary-btn"])
-    
-    with gr.Row():
-        quick_btn5 = gr.Button("Stellar Logistics", size="sm", elem_classes=["secondary-btn"])
-        quick_btn6 = gr.Button("Nexus Financial", size="sm", elem_classes=["secondary-btn"])
-        quick_btn7 = gr.Button("Horizon Health", size="sm", elem_classes=["secondary-btn"])
-        quick_btn8 = gr.Button("Phoenix Manufacturing", size="sm", elem_classes=["secondary-btn"])
+        with gr.Column(scale=1):
+            results_output = gr.Markdown(
+                label="📊 Contact Results",
+                value="Enter a URL and click 'Search Contacts' to see results here.",
+                height=400
+            )
+        
+        with gr.Column(scale=1):
+            json_output = gr.Code(
+                label="📝 Raw JSON Response",
+                language="json",
+                value="",
+                height=400
+            )
     
     # Event handlers
     search_btn.click(
-        fn=search_website_contacts,
-        inputs=[website_input, max_results],
-        outputs=[results_display, csv_output]
+        fn=search_contacts,
+        inputs=[url_input, api_endpoint],
+        outputs=[results_output, json_output]
     )
     
-    csv_search_btn.click(
-        fn=search_csv_websites,
-        inputs=[csv_file, csv_max_results],
-        outputs=[csv_results_display, csv_export_output]
+    demo_btn.click(
+        fn=lambda: (create_sample_data(), '{"contacts": [{"name": "John Smith", "email": "john.smith@example.com", "title": "CEO"}]}'),
+        outputs=[results_output, json_output]
     )
     
-    # Quick search button handlers
-    quick_btn1.click(lambda: "techflowsolutions.com", outputs=website_input)
-    quick_btn2.click(lambda: "greenleafconsult.com", outputs=website_input)
-    quick_btn3.click(lambda: "blueskymarketing.net", outputs=website_input)
-    quick_btn4.click(lambda: "quantumdynamics.org", outputs=website_input)
-    quick_btn5.click(lambda: "stellarlogistics.biz", outputs=website_input)
-    quick_btn6.click(lambda: "nexusfinancial.pro", outputs=website_input)
-    quick_btn7.click(lambda: "horizonhealth.care", outputs=website_input)
-    quick_btn8.click(lambda: "phoenixmfg.com", outputs=website_input)
-    
-    # Examples
-    gr.Examples(
-        examples=[
-            ["techflowsolutions.com", 5],
-            ["greenleafconsult.com", 3],
-            ["blueskymarketing.net", 4],
-            ["quantumdynamics.org", 6]
-        ],
-        inputs=[website_input, max_results],
-        label="Sample Searches"
+    clear_btn.click(
+        fn=lambda: ("", "", ""),
+        outputs=[url_input, results_output, json_output]
     )
     
     # Footer
     gr.HTML("""
-    <div style="text-align: center; padding: 30px 20px; background: linear-gradient(135deg, #64748b 0%, #475569 100%); color: white; border-radius: 15px; margin-top: 30px;">
-        <h3 style="margin: 0 0 10px 0;">Contact Intelligence Platform</h3>
-        <p style="margin: 0; opacity: 0.9;">Professional-grade contact discovery and lead generation technology</p>
-        <p style="margin: 10px 0 0 0; font-size: 0.9em; opacity: 0.7;">Powered by advanced web intelligence algorithms</p>
+    <div class="footer">
+        <p>© 2025 Contact Search Tool - Professional contact discovery solution</p>
     </div>
     """)
 
+# Launch the app
 if __name__ == "__main__":
-    app.launch()
+    demo.launch(
+        share=True,
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        debug=True
+    )