Update app.py

app.py

@@ -4,10 +4,9 @@ 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"
 
-# Professional color scheme inspired by corporate/real estate companies
+# Custom theme inspired by image
 theme = gr.themes.Soft(
     primary_hue="blue",
     secondary_hue="slate",
@@ -16,96 +15,52 @@ theme = gr.themes.Soft(
     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",
+    body_background_fill="#0f172a",  # Very dark blue-gray
+    body_background_fill_dark="#0f172a",
+    block_background_fill="#1e293b",  # Slightly lighter for blocks
+    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",
     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"
+    button_secondary_background_fill="#334155",
+    button_secondary_background_fill_hover="#475569",
+    button_secondary_text_color="#f8fafc"
 )
 
 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:
-        # 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
-        )
-        
+        response = requests.get(API_ENDPOINT, params={'url': url}, timeout=30)
         if response.status_code == 200:
-            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
-                
+            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)
         else:
             return f"❌ API Error ({response.status_code}): {response.text}", ""
-            
     except requests.exceptions.Timeout:
         return "❌ Request timeout. Please try again.", ""
     except requests.exceptions.ConnectionError:
@@ -114,21 +69,12 @@ 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', ''),
@@ -138,126 +84,54 @@ 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):
-        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:
+        contacts = results.get('contacts') or results.get('results') or results.get('data') or results
+        if isinstance(contacts, list) and contacts:
             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']}
-                
+                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']}
                 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:
+        if results:
             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"
+                output += f"## Contact {i}\n{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
 
-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="""
+with gr.Blocks(theme=theme, title="Kwekel Companies Contact Search Tool", 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%);
@@ -271,103 +145,64 @@ with gr.Blocks(theme=theme, title="Contact Search - Kwekel Companies", css="""
         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;
+        background: #1e293b;
+        border: 1px solid #3b82f6;
         border-radius: 0.5rem;
         padding: 1rem;
         margin: 1rem 0;
+        color: #f8fafc;
     }
     .footer {
         text-align: center;
         margin-top: 2rem;
         padding-top: 2rem;
-        border-top: 1px solid #e2e8f0;
-        color: #64748b;
+        border-top: 1px solid #475569;
+        color: #94a3b8;
     }
 """) as demo:
     
-    # Header
     gr.HTML("""
     <div class="header">
-        <h1>Contact Search Tool</h1>
-        <p>Professional contact discovery powered by AI</p>
+        <h1>Kwekel Companies Contact Search Tool</h1>
     </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",
-                info="Enter the website URL to search for contacts",
-                lines=1
-            )
-            
+            url_input = gr.Textbox(label="Website URL", placeholder="example.com or https://example.com", 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=""
-            )
-    
-    # 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
+            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])
+
     gr.HTML("""
     <div class="footer">
-        <p>© 2025 Contact Search Tool - Professional contact discovery solution</p>
+        <p>© 2025 Kwekel Companies Contact Search Tool</p>
     </div>
     """)
 
-# Launch the app
 if __name__ == "__main__":
     demo.launch(
         share=True,
@@ -375,4 +210,4 @@ if __name__ == "__main__":
         server_port=7860,
         show_error=True,
         debug=True
-    )
+    )