Spaces:

umarbalak
/

QueryMind

Sleeping

QueryMind / patch_gradio.py

update

0e3d491 over 1 year ago

1.28 kB

import gradio_client.utils

# 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_json_schema_to_python_type(schema, defs=None):
    try:
        # Check if schema is a boolean - handle this special case
        if isinstance(schema, bool):
            return "any"  # Return a sensible default type for boolean schemas
        
        # 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 "any"  # Return a default type for invalid schemas

# Apply the patch
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

	import gradio_client.utils

	# 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_json_schema_to_python_type(schema, defs=None):
	try:
	# Check if schema is a boolean - handle this special case
	if isinstance(schema, bool):
	return "any" # Return a sensible default type for boolean schemas

	# 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 "any" # Return a default type for invalid schemas

	# Apply the patch
	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