Upload folder using huggingface_hub

app.py

@@ -1,6 +1,6 @@
 import inspect
 import time
-from typing import Iterable
+from typing import Iterable, Optional
 
 from gradio_client.documentation import document_fn
 
@@ -50,7 +50,6 @@ def get_doc_theme_var_groups():
     return groups, flat_variables
 
 
-
 variable_groups, flat_variables = get_doc_theme_var_groups()
 
 css = """
@@ -504,9 +503,14 @@ with gr.Blocks(  # noqa: SIM117
             + mono_is_google
             + theme_var_input
         )
-
-        def load_theme(theme_name, custom_theme: gr.themes.Base = None):
-            print("loading theme", theme_name, custom_theme.__dict__ if custom_theme else None)
+        def load_theme(theme_name: Optional[str], custom_theme: Optional[gr.themes.ThemeClass] = None):
+            '''
+            Load a theme based on the theme name or theme object.
+            
+            Parameters:
+                theme_name (str): The name of the theme to load.
+                custom_theme (gr.themes.Base, optional): Custom theme object to load. Defaults to None if loading a default theme.
+            '''
             if custom_theme:
                 theme = custom_theme
             else:
@@ -521,7 +525,7 @@ with gr.Blocks(  # noqa: SIM117
             spacing_size = parameters["spacing_size"].default
             radius_size = parameters["radius_size"].default
 
-            
+
 
             font = theme._font[:4]
             font_mono = theme._font_mono[:4]
@@ -557,9 +561,10 @@ with gr.Blocks(  # noqa: SIM117
             )
 
         def load_custom_theme(custom_theme_name):
-            print("loading custom theme", custom_theme_name)
-            theme = gr.Theme.from_hub(custom_theme_name)
-            print("loaded custom theme", theme.name)
+            try:
+                theme = gr.Theme.from_hub(custom_theme_name)
+            except Exception as e:
+                raise gr.Error(f"Error loading custom theme:{e}")
             return load_theme(theme.name, theme)
         
 
@@ -644,21 +649,7 @@ with gr.Blocks(  # noqa: SIM117
                     if getattr(source_obj, attr) != final_attr_values[final_theme_attr]:
                         diff = True
                 if diff:
-                    new_final_attr_values = {}
-                    # We need to update the theme keys to match the color and size attribute names
-                    for key, val in final_attr_values.items():
-                        if key.startswith(("primary_", "secondary_", "neutral_")):
-                            color_key = "c" + key.split("_")[-1]
-                            new_final_attr_values[color_key] = val
-                        elif key.startswith(("text_", "spacing_", "radius_")):
-                            size_key = key.split("_")[-1]
-                            new_final_attr_values[size_key] = val
-                        else:
-                            new_final_attr_values[key] = val
-                    specific_core_diffs[value_name] = (
-                        source_class,
-                        new_final_attr_values,
-                    )
+                    specific_core_diffs[value_name] = (source_class, final_attr_values)
 
             font_diffs = {}
 
@@ -927,7 +918,8 @@ with gr.Blocks(theme=theme) as demo:
         )
 
         attach_rerender(
-            load_custom_theme_btn.click(load_custom_theme, custom_theme_name, theme_inputs, show_api=False).then
+            load_custom_theme_btn.click(load_custom_theme, custom_theme_name,
+                                         theme_inputs, show_api=False).then
         )
 
         for theme_box in (