import importlib import os import warnings from types import ModuleType from typing import Any, Dict, List, Optional import gradio from gradio.themes import Size from ffff import logger, metadata, state_manager, wording from ffff.exit_helper import hard_exit from ffff.filesystem import resolve_relative_path from ffff.uis import overrides from ffff.uis.typing import Component, ComponentName os.environ['GRADIO_ANALYTICS_ENABLED'] = '0' warnings.filterwarnings('ignore', category = UserWarning, module = 'gradio') gradio.processing_utils.encode_array_to_base64 = overrides.encode_array_to_base64 gradio.processing_utils.encode_pil_to_base64 = overrides.encode_pil_to_base64 UI_COMPONENTS: Dict[ComponentName, Component] = {} UI_LAYOUT_MODULES : List[ModuleType] = [] UI_LAYOUT_METHODS =\ [ 'pre_check', 'pre_render', 'render', 'listen', 'run' ] def load_ui_layout_module(ui_layout : str) -> Any: try: ui_layout_module = importlib.import_module('ffff.uis.layouts.' + ui_layout) for method_name in UI_LAYOUT_METHODS: if not hasattr(ui_layout_module, method_name): raise NotImplementedError except ModuleNotFoundError as exception: logger.error(wording.get('ui_layout_not_loaded').format(ui_layout = ui_layout), __name__) logger.debug(exception.msg, __name__) hard_exit(1) except NotImplementedError: logger.error(wording.get('ui_layout_not_implemented').format(ui_layout = ui_layout), __name__) hard_exit(1) return ui_layout_module def get_ui_layouts_modules(ui_layouts : List[str]) -> List[ModuleType]: global UI_LAYOUT_MODULES if not UI_LAYOUT_MODULES: for ui_layout in ui_layouts: ui_layout_module = load_ui_layout_module(ui_layout) UI_LAYOUT_MODULES.append(ui_layout_module) return UI_LAYOUT_MODULES def get_ui_component(component_name : ComponentName) -> Optional[Component]: if component_name in UI_COMPONENTS: return UI_COMPONENTS[component_name] return None def get_ui_components(component_names : List[ComponentName]) -> Optional[List[Component]]: ui_components = [] for component_name in component_names: component = get_ui_component(component_name) if component: ui_components.append(component) return ui_components def register_ui_component(component_name : ComponentName, component: Component) -> None: UI_COMPONENTS[component_name] = component def launch() -> None: ui_layouts_total = len(state_manager.get_item('ui_layouts')) with gradio.Blocks(theme = get_theme(), css = get_css(), title = metadata.get('name') + ' ' + metadata.get('version'), fill_width = True) as ui: for ui_layout in state_manager.get_item('ui_layouts'): ui_layout_module = load_ui_layout_module(ui_layout) if ui_layout_module.pre_render(): if ui_layouts_total > 1: with gradio.Tab(ui_layout): ui_layout_module.render() ui_layout_module.listen() else: ui_layout_module.render() ui_layout_module.listen() for ui_layout in state_manager.get_item('ui_layouts'): ui_layout_module = load_ui_layout_module(ui_layout) ui_layout_module.run(ui) def get_theme() -> gradio.Theme: return gradio.themes.Base( primary_hue = gradio.themes.colors.red, secondary_hue = gradio.themes.colors.neutral, radius_size = Size( xxs = '0.375rem', xs = '0.375rem', sm = '0.375rem', md = '0.375rem', lg = '0.375rem', xl = '0.375rem', xxl = '0.375rem', ), font = gradio.themes.GoogleFont('Open Sans') ).set( background_fill_primary = '*neutral_100', block_background_fill = 'white', block_border_width = '0', block_label_background_fill = '*neutral_100', block_label_background_fill_dark = '*neutral_700', block_label_border_width = 'none', block_label_margin = '0.5rem', block_label_radius = '*radius_md', block_label_text_color = '*neutral_700', block_label_text_size = '*text_sm', block_label_text_color_dark = 'white', block_label_text_weight = '600', block_title_background_fill = '*neutral_100', block_title_background_fill_dark = '*neutral_700', block_title_padding = '*block_label_padding', block_title_radius = '*block_label_radius', block_title_text_color = '*neutral_700', block_title_text_size = '*text_sm', block_title_text_weight = '600', block_padding = '0.5rem', border_color_primary = 'transparent', border_color_primary_dark = 'transparent', button_large_padding = '2rem 0.5rem', button_large_text_weight = 'normal', button_primary_background_fill = '*primary_500', button_primary_text_color = 'white', button_secondary_background_fill = 'white', button_secondary_border_color = 'transparent', button_secondary_border_color_dark = 'transparent', button_secondary_border_color_hover = 'transparent', button_secondary_border_color_hover_dark = 'transparent', button_secondary_text_color = '*neutral_800', button_small_padding = '0.75rem', checkbox_background_color = '*neutral_200', checkbox_background_color_selected = '*primary_600', checkbox_background_color_selected_dark = '*primary_700', checkbox_border_color_focus = '*primary_500', checkbox_border_color_focus_dark = '*primary_600', checkbox_border_color_selected = '*primary_600', checkbox_border_color_selected_dark = '*primary_700', checkbox_label_background_fill = '*neutral_50', checkbox_label_background_fill_hover = '*neutral_50', checkbox_label_background_fill_selected = '*primary_500', checkbox_label_background_fill_selected_dark = '*primary_600', checkbox_label_text_color_selected = 'white', input_background_fill = '*neutral_50', shadow_drop = 'none', slider_color = '*primary_500', slider_color_dark = '*primary_600' ) def get_css() -> str: # Retorna una cadena vacía para no cargar CSS adicional return "" def launch() -> None: ui_layouts_total = len(state_manager.get_item('ui_layouts')) with gradio.Blocks(theme='ParityError/Interstellar', css="") as ui: # Aplica el tema for ui_layout in state_manager.get_item('ui_layouts'): ui_layout_module = load_ui_layout_module(ui_layout) if ui_layout_module.pre_render(): if ui_layouts_total > 1: with gradio.Tab(ui_layout): ui_layout_module.render() ui_layout_module.listen() else: ui_layout_module.render() ui_layout_module.listen() for ui_layout in state_manager.get_item('ui_layouts'): ui_layout_module = load_ui_layout_module(ui_layout) ui_layout_module.run(ui)