changes made to app.py

app.py

@@ -6,28 +6,29 @@ import os
 import gradio_client.utils as client_utils
 import sys
 
-# ================== FIX START ==================
-# Store original reference first
-_original_json_schema_handler = client_utils._json_schema_to_python_type
-
+# ===============================
+# Recursion Handling Fix
+# ===============================
 def _patched_json_schema_to_python_type(schema, defs=None, depth=0):
-    if depth > 100:  # Safety net
+    # Safety check to prevent infinite recursion
+    if depth > 100:
         return "Any"
+    
+    # Handle boolean cases
     if isinstance(schema, bool):
         return "Any" if schema else "None"
-    return _original_json_schema_handler(schema, defs)
+    
+    # Call the original function with increased depth
+    try:
+        return client_utils._json_schema_to_python_type(schema, defs)
+    except RecursionError:
+        return "Any"
 
+# Modify the utilities to use the patched function
 client_utils._json_schema_to_python_type = _patched_json_schema_to_python_type
-sys.setrecursionlimit(10000)  # Temporarily increase recursion depth
-# ================== FIX END ====================
 
-# Workaround for Gradio schema generation issue
-def _patched_json_schema_to_python_type(schema, defs=None):
-    if isinstance(schema, bool):
-        return "Any" if schema else "None"
-    return client_utils._json_schema_to_python_type(schema, defs)
-
-client_utils._json_schema_to_python_type = _patched_json_schema_to_python_type
+# Increase recursion limit as a backup
+sys.setrecursionlimit(10000)
 
 # Set up device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")