Update app.py

app.py

@@ -46,10 +46,10 @@ theme = gr.themes.Soft(
     # Button colors
     button_primary_background_fill="#1e40af",
     button_primary_background_fill_hover="#2563eb",
-    button_primary_text_color="#2d2d2d",
+    button_primary_text_color="#ffffff",
     button_secondary_background_fill="#404040",
     button_secondary_background_fill_hover="#555555",
-    button_secondary_text_color="#2d2d2d"
+    button_secondary_text_color="#ffffff"
 )
 
 def normalize_url(url: str) -> str:
@@ -165,7 +165,7 @@ def search_contacts(url: str) -> tuple[str, str]:
     """
     
     if not url.strip():
-        return "Please enter a website URL", ""
+        return "❌ Please enter a website URL", ""
     
     # Clean and validate URL
     url = url.strip()
@@ -204,10 +204,10 @@ def search_contacts(url: str) -> tuple[str, str]:
                     # If we filtered out everything or got very few results compared to original,
                     # it's likely the API returned all contacts instead of URL-specific ones
                     if len(filtered_result) == 0:
-                        return f"No contacts found for: {original_url}\n\nThis URL may not be in our database.", json.dumps(result, indent=2)
+                        return f"❌ No contacts found for: {original_url}\n\nThis URL may not be in our database.", json.dumps(result, indent=2)
                     elif original_count > 50 and len(filtered_result) < 5:
                         # Likely got all contacts, filtered to very few - show warning
-                        formatted_output = f"**Warning:** API returned {original_count} total contacts. After filtering for '{original_url}', found {len(filtered_result)} matches.\n\n"
+                        formatted_output = f"⚠️ **Warning:** API returned {original_count} total contacts. After filtering for '{original_url}', found {len(filtered_result)} matches.\n\n"
                         formatted_output += "These results may not be accurate. The URL might not be in our database.\n\n"
                         formatted_output += format_contact_results(filtered_result, url)
                     else:
@@ -220,9 +220,9 @@ def search_contacts(url: str) -> tuple[str, str]:
                     filtered_contacts = filter_contacts_by_url(actual_result['contacts'], url)
                     
                     if len(filtered_contacts) == 0:
-                        return f"No contacts found for: {original_url}\n\nThis URL may not be in our database.", json.dumps(result, indent=2)
+                        return f"❌ No contacts found for: {original_url}\n\nThis URL may not be in our database.", json.dumps(result, indent=2)
                     elif original_count > 50 and len(filtered_contacts) < 5:
-                        formatted_output = f"**Warning:** API returned {original_count} total contacts. After filtering for '{original_url}', found {len(filtered_contacts)} matches.\n\n"
+                        formatted_output = f"⚠️ **Warning:** API returned {original_count} total contacts. After filtering for '{original_url}', found {len(filtered_contacts)} matches.\n\n"
                         formatted_output += "These results may not be accurate. The URL might not be in our database.\n\n"
                         formatted_output += format_contact_results({'contacts': filtered_contacts}, url)
                     else:
@@ -250,17 +250,17 @@ def search_contacts(url: str) -> tuple[str, str]:
                 return formatted_output, raw_json
                 
             except json.JSONDecodeError:
-                return f"Error: Invalid JSON response from API", response.text
+                return f"❌ Error: Invalid JSON response from API", response.text
                 
         else:
-            return f"API Error ({response.status_code}): {response.text}", ""
+            return f"❌ API Error ({response.status_code}): {response.text}", ""
             
     except requests.exceptions.Timeout:
-        return "Request timeout. Please try again.", ""
+        return "❌ Request timeout. Please try again.", ""
     except requests.exceptions.ConnectionError:
-        return "Connection error. Please check your API endpoint.", ""
+        return "❌ Connection error. Please check your API endpoint.", ""
     except Exception as e:
-        return f"Error: {str(e)}", ""
+        return f"❌ Error: {str(e)}", ""
 
 def format_contact_results(results: Dict[Any, Any], url: str) -> str:
     """
@@ -340,6 +340,34 @@ def format_contact_results(results: Dict[Any, Any], url: str) -> str:
     
     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="Kwekel Companies Contact Search Tool", css="""
     /* Root and body styles */
@@ -607,4 +635,4 @@ if __name__ == "__main__":
         server_port=7860,
         show_error=True,
         debug=True
-    )
+    )