Update app.py

app.py

@@ -4,8 +4,8 @@ import json
 from typing import List, Dict, Any
 import pandas as pd
 
-# Hardcoded API endpoint
-API_ENDPOINT = "https://huggingface.co/spaces/JayBene1/testapicontacts"
+# Hardcoded API endpoint - Updated to correct Hugging Face Spaces API format
+API_ENDPOINT = "https://jaybene1-testapicontacts.hf.space/api/predict"
 
 # Professional color scheme inspired by corporate/real estate companies
 theme = gr.themes.Soft(
@@ -54,17 +54,14 @@ def search_contacts(url: str) -> tuple[str, str]:
         url = 'https://' + url
     
     try:
-        # Prepare the API request
+        # Prepare the API request for Hugging Face Spaces
         headers = {
             'Content-Type': 'application/json'
         }
         
+        # Hugging Face Spaces API format
         payload = {
-            "inputs": url,
-            "parameters": {
-                "task": "contact_search",
-                "url": url
-            }
+            "data": [url]
         }
         
         # Make the API request to the hardcoded endpoint
@@ -79,8 +76,14 @@ def search_contacts(url: str) -> tuple[str, str]:
             try:
                 result = response.json()
                 
+                # 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
+                
                 # Format the results for display
-                formatted_output = format_contact_results(result, url)
+                formatted_output = format_contact_results(actual_result, url)
                 raw_json = json.dumps(result, indent=2)
                 
                 return formatted_output, raw_json