update

patch_gradio.py

@@ -1,20 +1,31 @@
 import gradio_client.utils
 
-# Store the original function
-original_get_type = gradio_client.utils.get_type
+# Store the original function to maintain its behavior
+original_json_schema_to_python_type = gradio_client.utils._json_schema_to_python_type
 
 # Define the patched function
-def patched_get_type(schema):
+def patched_json_schema_to_python_type(schema, defs=None):
     try:
-        # Check if schema is a dict before using 'in' operator
-        if not isinstance(schema, dict):
-            return str(schema)  # Convert to string as fallback
+        # Check if schema is a boolean - handle this special case
+        if isinstance(schema, bool):
+            return "any"  # Return a sensible default type for boolean schemas
         
-        # Original logic continues
-        return original_get_type(schema)
+        # Continue with original function for non-boolean schemas
+        return original_json_schema_to_python_type(schema, defs)
     except Exception as e:
         print(f"Error parsing schema: {schema}, Error: {e}")
-        return "unknown"  # Return a default type for invalid schemas
+        return "any"  # Return a default type for invalid schemas
 
 # Apply the patch
-gradio_client.utils.get_type = patched_get_type
+gradio_client.utils._json_schema_to_python_type = patched_json_schema_to_python_type
+
+# Also patch the wrapper function that calls the internal function
+def patched_json_schema_to_python_type_wrapper(schema, defs=None):
+    try:
+        return patched_json_schema_to_python_type(schema, schema.get("$defs"))
+    except Exception as e:
+        print(f"Error in wrapper: {e}")
+        return "any"
+
+# Apply the wrapper patch
+gradio_client.utils.json_schema_to_python_type = patched_json_schema_to_python_type_wrapper