Update app.py

app.py

@@ -4,9 +4,10 @@ import json
 from typing import List, Dict, Any
 import pandas as pd
 
+# Hardcoded API endpoint - Updated to correct endpoint
 API_ENDPOINT = "https://jaybene1-testapicontacts.hf.space/contacts"
 
-# Updated dark theme with white text
+# Professional color scheme inspired by corporate/real estate companies
 theme = gr.themes.Soft(
     primary_hue="blue",
     secondary_hue="slate",
@@ -15,54 +16,96 @@ theme = gr.themes.Soft(
     spacing_size="md",
     radius_size="md"
 ).set(
-    body_background_fill="#0f172a",
-    body_background_fill_dark="#0f172a",
-    block_background_fill="#1e293b",
-    block_background_fill_dark="#1e293b",
-    block_border_color="#334155",
-    block_border_color_dark="#334155",
-    input_background_fill="#1e293b",
-    input_background_fill_dark="#1e293b",
-    input_border_color="#475569",
-    input_border_color_dark="#475569",
+    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="#334155",
-    button_secondary_background_fill_hover="#475569",
-    button_secondary_text_color="#f8fafc",
-    text_color="#f8fafc",
-    text_color_dark="#f8fafc"
+    button_secondary_background_fill="#f1f5f9",
+    button_secondary_background_fill_hover="#e2e8f0",
+    button_secondary_text_color="#334155"
 )
 
 def search_contacts(url: str) -> tuple[str, str]:
+    """
+    Search for contacts associated with a given URL using the Hugging Face API.
+    
+    Args:
+        url: The website URL to search for contacts
+        
+    Returns:
+        Tuple of (formatted_results, raw_json)
+    """
+    
     if not url.strip():
         return "❌ Please enter a website URL", ""
+    
+    # Clean and validate URL
     url = url.strip()
     if not url.startswith(('http://', 'https://')):
         url = 'https://' + url
-
+    
     try:
-        response = requests.get(API_ENDPOINT, params={'url': url}, timeout=30)
+        # Use GET request with URL parameter
+        params = {
+            'url': url
+        }
+        
+        # Make the API request to the hardcoded endpoint using GET
+        response = requests.get(
+            API_ENDPOINT,
+            params=params,
+            timeout=30
+        )
+        
         if response.status_code == 200:
-            result = response.json()
-            request_url = f"{API_ENDPOINT}?url={url}"
-            actual_result = result.get('data', [])[0] if isinstance(result, dict) and 'data' in result else result
-            if isinstance(actual_result, list):
-                actual_result = filter_contacts_by_url(actual_result, url)
-            elif isinstance(actual_result, dict) and 'contacts' in actual_result:
-                actual_result['contacts'] = filter_contacts_by_url(actual_result['contacts'], url)
-            formatted_output = format_contact_results(actual_result, url)
-            formatted_output += (
-                f"\n\n**Debug Info:**\n"
-                f"- Request URL: {API_ENDPOINT}\n"
-                f"- Search URL: {url}\n"
-                f"- Response contains {len(actual_result) if isinstance(actual_result, list) else 'N/A'} items\n"
-                f"- Client-side filtering applied\n"
-            )
-            return formatted_output, json.dumps(result, indent=2)
+            try:
+                result = response.json()
+                
+                # Add debugging info
+                request_url = f"{API_ENDPOINT}?url={url}"
+                
+                # Hugging Face Spaces API returns data in a different format
+                if isinstance(result, dict) and 'data' in result:
+                    actual_result = result['data'][0] if result['data'] else {}
+                else:
+                    actual_result = result
+                
+                # Apply client-side filtering
+                if isinstance(actual_result, list):
+                    actual_result = filter_contacts_by_url(actual_result, url)
+                elif isinstance(actual_result, dict) and 'contacts' in actual_result:
+                    actual_result['contacts'] = filter_contacts_by_url(actual_result['contacts'], url)
+                
+                # Format the results for display
+                formatted_output = format_contact_results(actual_result, url)
+                
+                # Add debugging info to the formatted output
+                debug_info = f"\n\n**Debug Info:**\n"
+                debug_info += f"- Request URL: {API_ENDPOINT}\n"
+                debug_info += f"- Search URL: {url}\n"
+                debug_info += f"- Response contains {len(actual_result) if isinstance(actual_result, list) else 'N/A'} items\n"
+                debug_info += f"- Client-side filtering applied\n"
+                
+                formatted_output += debug_info
+                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:
             return f"❌ API Error ({response.status_code}): {response.text}", ""
+            
     except requests.exceptions.Timeout:
         return "❌ Request timeout. Please try again.", ""
     except requests.exceptions.ConnectionError:
@@ -71,12 +114,21 @@ def search_contacts(url: str) -> tuple[str, str]:
         return f"❌ Error: {str(e)}", ""
 
 def filter_contacts_by_url(contacts, search_url):
+    """
+    Filter contacts based on the search URL.
+    This is a client-side filter in case the API doesn't filter properly.
+    """
     if not isinstance(contacts, list):
         return contacts
+    
+    # Extract domain from search URL
     search_domain = search_url.replace('https://', '').replace('http://', '').replace('www.', '').split('/')[0].lower()
+    
     filtered_contacts = []
     for contact in contacts:
+        # Check if contact is associated with the search domain
         if isinstance(contact, dict):
+            # Check various fields that might contain the domain
             fields_to_check = [
                 contact.get('website', ''),
                 contact.get('company_website', ''),
@@ -86,54 +138,126 @@ def filter_contacts_by_url(contacts, search_url):
                 str(contact.get('url', '')),
                 str(contact.get('source', ''))
             ]
+            
+            # Check if any field contains the search domain
             for field in fields_to_check:
                 if field and search_domain in str(field).lower():
                     filtered_contacts.append(contact)
                     break
+    
     return filtered_contacts if filtered_contacts else contacts
 
 def format_contact_results(results: Dict[Any, Any], url: str) -> str:
+    """
+    Format the API results into a readable format.
+    
+    Args:
+        results: The JSON response from the API
+        url: The searched URL
+        
+    Returns:
+        Formatted string with contact information
+    """
+    
     output = f"# Contact Search Results for: {url}\n\n"
+    
+    # Handle different possible result structures
     if isinstance(results, dict):
-        contacts = results.get('contacts') or results.get('results') or results.get('data') or results
-        if isinstance(contacts, list) and contacts:
+        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- **Email:** {email}\n- **Title:** {title}\n- **Company:** {company}\n- **Phone:** {phone}\n- **LinkedIn:** {linkedin}\n\n"
-                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']}
+                
+                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 results:
+        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{contact}\n\n"
+                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
 
-with gr.Blocks(theme=theme, title="Kwekel Companies Contact Search Tool", css="""
+def create_sample_data():
+    """Create sample data for demonstration"""
+    return '''# Contact Search Results for: example.com
+
+**Found 2 contact(s):**
+
+## Contact 1
+- **Name:** John Smith
+- **Email:** john.smith@example.com
+- **Title:** CEO
+- **Company:** Example Corp
+- **Phone:** (555) 123-4567
+- **LinkedIn:** linkedin.com/in/johnsmith
+
+---
+
+## 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
+
+---
+
+*Search completed at: 2025-01-15 10:30:00*'''
+
+# 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;
-        background-color: #0f172a !important;
-        color: #f8fafc !important;
     }
     .header {
         background: linear-gradient(135deg, #1e40af 0%, #3b82f6 100%);
@@ -147,68 +271,103 @@ with gr.Blocks(theme=theme, title="Kwekel Companies Contact Search Tool", css=""
         margin: 0;
         font-size: 2.5rem;
         font-weight: 700;
-        color: #ffffff;
+        margin-bottom: 0.5rem;
+    }
+    .header p {
+        margin: 0;
+        font-size: 1.2rem;
+        opacity: 0.9;
     }
     .info-box {
-        background: #1e293b;
-        border: 1px solid #3b82f6;
+        background: #f0f9ff;
+        border: 1px solid #0ea5e9;
         border-radius: 0.5rem;
         padding: 1rem;
         margin: 1rem 0;
-        color: #f8fafc;
-    }
-    label, h3, p, li, ol {
-        color: #f8fafc !important;
     }
     .footer {
         text-align: center;
         margin-top: 2rem;
         padding-top: 2rem;
-        border-top: 1px solid #475569;
-        color: #94a3b8 !important;
+        border-top: 1px solid #e2e8f0;
+        color: #64748b;
     }
 """) as demo:
     
+    # Header
     gr.HTML("""
     <div class="header">
-        <h1>Kwekel Companies Contact Search Tool</h1>
+        <h1>Contact Search Tool</h1>
+        <p>Professional contact discovery powered by AI</p>
     </div>
     """)
-
+    
+    # Info box
     gr.HTML(f"""
     <div class="info-box">
         <h3>Instructions:</h3>
         <ol>
-            <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>
+            <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>
         <p><strong>API Endpoint:</strong> {API_ENDPOINT}</p>
     </div>
     """)
-
+    
     with gr.Row():
         with gr.Column(scale=2):
-            url_input = gr.Textbox(label="Website URL", placeholder="example.com or https://example.com", lines=1)
+            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():
                 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():
         with gr.Column(scale=1):
-            results_output = gr.Markdown(label="Contact Results", value="Enter a URL and click 'Search Contacts' to see results here.")
+            results_output = gr.Markdown(
+                label="Contact Results",
+                value="Enter a URL and click 'Search Contacts' to see results here."
+            )
+        
         with gr.Column(scale=1):
-            json_output = gr.Code(label="Raw JSON Response", language="json", value="")
-
-    search_btn.click(fn=search_contacts, inputs=[url_input], outputs=[results_output, json_output])
-    clear_btn.click(fn=lambda: ("", ""), outputs=[url_input, results_output])
-
+            json_output = gr.Code(
+                label="Raw JSON Response",
+                language="json",
+                value=""
+            )
+    
+    # Event handlers
+    search_btn.click(
+        fn=search_contacts,
+        inputs=[url_input],
+        outputs=[results_output, json_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]
+    )
+    
+    clear_btn.click(
+        fn=lambda: ("", ""),
+        outputs=[url_input, results_output]
+    )
+    
+    # Footer
     gr.HTML("""
     <div class="footer">
-        <p>© 2025 Kwekel Companies Contact Search Tool</p>
+        <p>© 2025 Contact Search Tool - Professional contact discovery solution</p>
     </div>
     """)
 
+# Launch the app
 if __name__ == "__main__":
     demo.launch(
         share=True,
@@ -216,4 +375,4 @@ if __name__ == "__main__":
         server_port=7860,
         show_error=True,
         debug=True
-    )
+    )