Update app.py

app.py

@@ -1,17 +1,25 @@
 import yaml
 import os
 
-# Monkey patch for Gradio compatibility
+# Monkey patch for Gradio compatibility - remove unsupported kwargs from components
 import gradio as gr
-_original_blocks_init__ = gr.Blocks.__init__
-
-def patched_blocks_init(self, *args, **kwargs):
-    # Remove unsupported kwargs
-    kwargs.pop('theme', None)
-    kwargs.pop('fill_height', None)
-    _original_blocks_init__(self, *args, **kwargs)
-
-gr.Blocks.__init__ = patched_blocks_init
+from functools import wraps
+
+def create_wrapper(original_init):
+    @wraps(original_init)
+    def wrapper(self, *args, **kwargs):
+        # List of kwargs to remove for compatibility
+        unsupported_kwargs = ['theme', 'fill_height', 'type', 'scale', 'min_width']
+        for kwarg in unsupported_kwargs:
+            kwargs.pop(kwarg, None)
+        return original_init(self, *args, **kwargs)
+    return wrapper
+
+# Patch Blocks, Chatbot, Textbox, and other components
+for component_name in ['Blocks', 'Chatbot', 'Textbox', 'Button', 'Slider', 'Dropdown', 'Radio', 'Checkbox', 'Number', 'Image', 'Audio', 'Video']:
+    if hasattr(gr, component_name):
+        component = getattr(gr, component_name)
+        component.__init__ = create_wrapper(component.__init__)
 from smolagents import GradioUI, CodeAgent, InferenceClientModel
 
 # Get current directory path