Update app.py

app.py

@@ -37,30 +37,46 @@ def decode_image(image_base64: str):
 
 def clean_ppstructure_result(result):
     """
-    Recursively converts numpy arrays in the PPStructure result to standard lists.
-    Also optionally removes the 'img' field (raw numpy image crop) to keep standard JSON payloads light.
+    Recursively converts numpy arrays and custom objects in the PPStructure result
+    to standard Python lists and dictionaries that are JSON-safe.
     """
     if isinstance(result, list):
+        # Case 1: List - Recurse through all elements
         return [clean_ppstructure_result(item) for item in result]
-    elif isinstance(result, dict):
+
+    # Case 2: Dictionary or Custom Object
+    # Custom objects (with a __dict__) need to be converted to dicts before processing.
+    if isinstance(result, dict) or hasattr(result, '__dict__'):
+        if hasattr(result, '__dict__'):
+            # It's a custom class instance (e.g., from PPStructure), get its attributes as a dict
+            items = vars(result).items()
+        else:
+            # It's a standard dictionary
+            items = result.items()
+
         cleaned_dict = {}
-        for key, value in result.items():
-            # PPStructure often includes the raw image crop as a numpy array in the 'img' key.
-            # Returning this in standard JSON is usually a bad idea (too heavy).
-            # Comment out the next two lines if you actually NEED raw pixel data in your JSON.
+        for key, value in items:
+            # Exclude raw image data, as it's too large and unnecessary for JSON
             if key == 'img':
                  continue
             
             cleaned_dict[key] = clean_ppstructure_result(value)
         return cleaned_dict
+    
+    # Case 3: Numpy array - Convert to standard Python list
     elif isinstance(result, np.ndarray):
-        return result.tolist() # crucial: converts numpy array to standard list
+        return result.tolist()
+        
+    # Case 4: Numpy scalar types - Convert to standard Python primitives
+    # These base classes cover all int and float variants (int16, float32, etc.)
     elif isinstance(result, np.floating):
-        return float(result) # crucial: converts ALL numpy floats (32, 64, etc) to standard python floats
+        return float(result)
     elif isinstance(result, np.integer):
-        return int(result) # crucial: converts ALL numpy ints (int8, int16, uint8, etc) to standard python ints
+        return int(result)
     elif isinstance(result, np.bool_):
         return bool(result)
+        
+    # Case 5: Standard Python primitive or unhandled type - return as is
     else:
         return result