Upload ui_element_client.py with huggingface_hub

ui_element_client.py

@@ -20,10 +20,16 @@ class UIElementDetectionClient:
     def health_check(self):
         """Check if API is running."""
         try:
-            response = requests.get(f"{self.api_url}/health", timeout=5)
+            # Try new API endpoint first
+            response = requests.get(f"{self.api_url}/api/health", timeout=5)
             return response.json()
-        except Exception as e:
-            return {"error": str(e)}
+        except Exception:
+            # Fallback to old endpoint
+            try:
+                response = requests.get(f"{self.api_url}/health", timeout=5)
+                return response.json()
+            except Exception as e:
+                return {"error": str(e)}
     
     def analyze_image(self, image_path):
         """
@@ -39,15 +45,28 @@ class UIElementDetectionClient:
             raise FileNotFoundError(f"Image not found: {image_path}")
         
         print(f"[Client] Analyzing: {image_path}")
-        print(f"[Client] Uploading to {self.api_url}/analyze...")
+        
+        # Try new API endpoint first
+        api_endpoint = f"{self.api_url}/api/analyze"
+        print(f"[Client] Uploading to {api_endpoint}...")
         
         with open(image_path, 'rb') as f:
             files = {'file': f}
-            response = requests.post(
-                f"{self.api_url}/analyze",
-                files=files,
-                timeout=300
-            )
+            try:
+                response = requests.post(
+                    api_endpoint,
+                    files=files,
+                    timeout=300
+                )
+            except Exception:
+                # Fallback to old endpoint
+                api_endpoint = f"{self.api_url}/analyze"
+                print(f"[Client] Retrying with {api_endpoint}...")
+                response = requests.post(
+                    api_endpoint,
+                    files=files,
+                    timeout=300
+                )
         
         if response.status_code != 200:
             raise Exception(f"API error: {response.status_code} - {response.text}")
@@ -143,7 +162,7 @@ def main():
             result = client.analyze_image(args.image)
             element = None
             for elem in result['analysis']['elements']:
-                if elem['template_id'] == args.element:
+                if elem.get('element_id') == args.element or elem.get('template_id') == args.element:
                     element = elem
                     break
             
@@ -161,7 +180,10 @@ def main():
             elements = client.find_elements_in_region(args.image, *args.region)
             print(f"\n[Found {len(elements)} elements in region {args.region}]")
             for elem in elements:
-                print(f"  - {elem['template_id']} @ ({elem['center']['x']}, {elem['center']['y']})")
+                elem_id = elem.get('element_id') or elem.get('template_id', 'unknown')
+                x = elem.get('x') or elem.get('center', {}).get('x', 0)
+                y = elem.get('y') or elem.get('center', {}).get('y', 0)
+                print(f"  - {elem_id} @ ({x}, {y})")
         
         else:
             # Full analysis
@@ -169,7 +191,11 @@ def main():
             
             print(f"\n[Top 5 Elements by Confidence]")
             for i, elem in enumerate(result['analysis']['elements'][:5], 1):
-                print(f"  {i}. {elem['template_id']} @ ({elem['center']['x']}, {elem['center']['y']}) - {elem['confidence']:.4f}")
+                elem_id = elem.get('element_id') or elem.get('template_id', f"elem_{i}")
+                x = elem.get('x') or elem.get('center', {}).get('x', 0)
+                y = elem.get('y') or elem.get('center', {}).get('y', 0)
+                conf = elem.get('confidence', 0)
+                print(f"  {i}. {elem_id} @ ({x}, {y}) - {conf:.4f}")
     
     except Exception as e:
         print(f"[ERROR] {str(e)}")