mcli-framework 7.5.1__py3-none-any.whl → 7.6.1__py3-none-any.whl

mcli/ml/dashboard/components/charts.py

@@ -1,10 +1,11 @@
 """Reusable chart components for Streamlit dashboards"""
 
-import plotly.graph_objects as go
-import plotly.express as px
+from typing import Any, Dict, List, Optional
+
 import pandas as pd
+import plotly.express as px
+import plotly.graph_objects as go
 import streamlit as st
-from typing import Optional, Dict, List, Any
 
 
 def create_timeline_chart(
@@ -13,24 +14,13 @@ def create_timeline_chart(
     y_col: str,
     title: str = "Timeline",
     color_col: Optional[str] = None,
-    height: int = 400
+    height: int = 400,
 ) -> go.Figure:
     """Create a timeline chart with Plotly"""
 
-    fig = px.line(
-        data,
-        x=x_col,
-        y=y_col,
-        color=color_col,
-        title=title,
-        markers=True
-    )
+    fig = px.line(data, x=x_col, y=y_col, color=color_col, title=title, markers=True)
 
-    fig.update_layout(
-        height=height,
-        hovermode='x unified',
-        showlegend=True if color_col else False
-    )
+    fig.update_layout(height=height, hovermode="x unified", showlegend=True if color_col else False)
 
     return fig
 
@@ -39,7 +29,7 @@ def create_status_pie_chart(
     data: pd.DataFrame,
     status_col: str,
     title: str = "Status Distribution",
-    color_map: Optional[Dict[str, str]] = None
+    color_map: Optional[Dict[str, str]] = None,
 ) -> go.Figure:
     """Create a pie chart for status distribution"""
 
@@ -47,28 +37,28 @@ def create_status_pie_chart(
 
     if color_map is None:
         color_map = {
-            'completed': '#10b981',
-            'running': '#3b82f6',
-            'pending': '#f59e0b',
-            'failed': '#ef4444',
-            'cancelled': '#6b7280'
+            "completed": "#10b981",
+            "running": "#3b82f6",
+            "pending": "#f59e0b",
+            "failed": "#ef4444",
+            "cancelled": "#6b7280",
         }
 
-    colors = [color_map.get(status.lower(), '#6b7280') for status in status_counts.index]
-
-    fig = go.Figure(data=[go.Pie(
-        labels=status_counts.index,
-        values=status_counts.values,
-        marker=dict(colors=colors),
-        hole=0.4
-    )])
-
-    fig.update_layout(
-        title=title,
-        height=350,
-        showlegend=True
+    colors = [color_map.get(status.lower(), "#6b7280") for status in status_counts.index]
+
+    fig = go.Figure(
+        data=[
+            go.Pie(
+                labels=status_counts.index,
+                values=status_counts.values,
+                marker=dict(colors=colors),
+                hole=0.4,
+            )
+        ]
     )
 
+    fig.update_layout(title=title, height=350, showlegend=True)
+
     return fig
 
 
@@ -77,21 +67,23 @@ def create_metric_trend_chart(
     time_col: str,
     metric_col: str,
     title: str,
-    target_value: Optional[float] = None
+    target_value: Optional[float] = None,
 ) -> go.Figure:
     """Create a metric trend chart with optional target line"""
 
     fig = go.Figure()
 
     # Add metric line
-    fig.add_trace(go.Scatter(
-        x=data[time_col],
-        y=data[metric_col],
-        mode='lines+markers',
-        name=metric_col,
-        line=dict(width=3),
-        marker=dict(size=8)
-    ))
+    fig.add_trace(
+        go.Scatter(
+            x=data[time_col],
+            y=data[metric_col],
+            mode="lines+markers",
+            name=metric_col,
+            line=dict(width=3),
+            marker=dict(size=8),
+        )
+    )
 
     # Add target line if specified
     if target_value is not None:
@@ -99,15 +91,11 @@ def create_metric_trend_chart(
             y=target_value,
             line_dash="dash",
             line_color="red",
-            annotation_text=f"Target: {target_value}"
+            annotation_text=f"Target: {target_value}",
         )
 
     fig.update_layout(
-        title=title,
-        xaxis_title=time_col,
-        yaxis_title=metric_col,
-        height=400,
-        hovermode='x unified'
+        title=title, xaxis_title=time_col, yaxis_title=metric_col, height=400, hovermode="x unified"
     )
 
     return fig
@@ -119,23 +107,13 @@ def create_heatmap(
     y_col: str,
     value_col: str,
     title: str = "Heatmap",
-    color_scale: str = "Blues"
+    color_scale: str = "Blues",
 ) -> go.Figure:
     """Create a heatmap visualization"""
 
-    pivot_data = data.pivot_table(
-        index=y_col,
-        columns=x_col,
-        values=value_col,
-        aggfunc='mean'
-    )
+    pivot_data = data.pivot_table(index=y_col, columns=x_col, values=value_col, aggfunc="mean")
 
-    fig = px.imshow(
-        pivot_data,
-        color_continuous_scale=color_scale,
-        title=title,
-        aspect="auto"
-    )
+    fig = px.imshow(pivot_data, color_continuous_scale=color_scale, title=title, aspect="auto")
 
     fig.update_layout(height=400)
 
@@ -148,17 +126,12 @@ def create_gantt_chart(
     start_col: str,
     end_col: str,
     status_col: Optional[str] = None,
-    title: str = "Timeline"
+    title: str = "Timeline",
 ) -> go.Figure:
     """Create a Gantt chart for job/task scheduling"""
 
     fig = px.timeline(
-        data,
-        x_start=start_col,
-        x_end=end_col,
-        y=task_col,
-        color=status_col,
-        title=title
+        data, x_start=start_col, x_end=end_col, y=task_col, color=status_col, title=title
     )
 
     fig.update_yaxes(categoryorder="total ascending")
@@ -168,9 +141,7 @@ def create_gantt_chart(
 
 
 def create_multi_metric_gauge(
-    values: Dict[str, float],
-    max_values: Dict[str, float],
-    title: str = "Metrics"
+    values: Dict[str, float], max_values: Dict[str, float], title: str = "Metrics"
 ) -> go.Figure:
     """Create multiple gauge charts in a grid"""
 
@@ -183,8 +154,8 @@ def create_multi_metric_gauge(
     fig = make_subplots(
         rows=rows,
         cols=cols,
-        specs=[[{'type': 'indicator'}] * cols] * rows,
-        subplot_titles=list(values.keys())
+        specs=[[{"type": "indicator"}] * cols] * rows,
+        subplot_titles=list(values.keys()),
     )
 
     for idx, (metric, value) in enumerate(values.items()):
@@ -196,63 +167,51 @@ def create_multi_metric_gauge(
             go.Indicator(
                 mode="gauge+number+delta",
                 value=value,
-                title={'text': metric},
+                title={"text": metric},
                 gauge={
-                    'axis': {'range': [None, max_val]},
-                    'bar': {'color': "darkblue"},
-                    'steps': [
-                        {'range': [0, max_val * 0.5], 'color': "lightgray"},
-                        {'range': [max_val * 0.5, max_val * 0.8], 'color': "gray"}
+                    "axis": {"range": [None, max_val]},
+                    "bar": {"color": "darkblue"},
+                    "steps": [
+                        {"range": [0, max_val * 0.5], "color": "lightgray"},
+                        {"range": [max_val * 0.5, max_val * 0.8], "color": "gray"},
                     ],
-                    'threshold': {
-                        'line': {'color': "red", 'width': 4},
-                        'thickness': 0.75,
-                        'value': max_val * 0.9
-                    }
-                }
+                    "threshold": {
+                        "line": {"color": "red", "width": 4},
+                        "thickness": 0.75,
+                        "value": max_val * 0.9,
+                    },
+                },
             ),
             row=row,
-            col=col
+            col=col,
         )
 
-    fig.update_layout(
-        title=title,
-        height=300 * rows
-    )
+    fig.update_layout(title=title, height=300 * rows)
 
     return fig
 
 
 def create_waterfall_chart(
-    categories: List[str],
-    values: List[float],
-    title: str = "Waterfall Chart"
+    categories: List[str], values: List[float], title: str = "Waterfall Chart"
 ) -> go.Figure:
     """Create a waterfall chart for step-by-step changes"""
 
-    fig = go.Figure(go.Waterfall(
-        name="",
-        orientation="v",
-        measure=["relative"] * (len(values) - 1) + ["total"],
-        x=categories,
-        y=values,
-        connector={"line": {"color": "rgb(63, 63, 63)"}},
-    ))
-
-    fig.update_layout(
-        title=title,
-        showlegend=False,
-        height=400
+    fig = go.Figure(
+        go.Waterfall(
+            name="",
+            orientation="v",
+            measure=["relative"] * (len(values) - 1) + ["total"],
+            x=categories,
+            y=values,
+            connector={"line": {"color": "rgb(63, 63, 63)"}},
+        )
     )
 
+    fig.update_layout(title=title, showlegend=False, height=400)
+
     return fig
 
 
 def render_chart(fig: go.Figure, key: Optional[str] = None):
     """Helper to render Plotly chart with consistent configuration"""
-    st.plotly_chart(
-        fig,
-        width="stretch",
-        config={"responsive": True},
-        key=key
-    )
+    st.plotly_chart(fig, width="stretch", config={"responsive": True}, key=key)

mcli/ml/dashboard/components/metrics.py

@@ -1,8 +1,9 @@
 """Reusable metric display components"""
 
-import streamlit as st
 from typing import Optional, Union
 
+import streamlit as st
+
 
 def display_metric_card(
     label: str,
@@ -10,42 +11,33 @@ def display_metric_card(
     delta: Optional[Union[int, float, str]] = None,
     delta_color: str = "normal",
     help_text: Optional[str] = None,
-    icon: Optional[str] = None
+    icon: Optional[str] = None,
 ):
     """Display a metric card with optional delta and icon"""
 
     if icon:
         label = f"{icon} {label}"
 
-    st.metric(
-        label=label,
-        value=value,
-        delta=delta,
-        delta_color=delta_color,
-        help=help_text
-    )
+    st.metric(label=label, value=value, delta=delta, delta_color=delta_color, help=help_text)
 
 
-def display_status_badge(
-    status: str,
-    size: str = "medium"
-) -> str:
+def display_status_badge(status: str, size: str = "medium") -> str:
     """Return a colored status badge"""
 
     status_colors = {
-        'completed': '🟢',
-        'success': '🟢',
-        'running': '🔵',
-        'in_progress': '🔵',
-        'pending': '🟡',
-        'waiting': '🟡',
-        'failed': '🔴',
-        'error': '🔴',
-        'cancelled': '⚪',
-        'unknown': '⚫'
+        "completed": "🟢",
+        "success": "🟢",
+        "running": "🔵",
+        "in_progress": "🔵",
+        "pending": "🟡",
+        "waiting": "🟡",
+        "failed": "🔴",
+        "error": "🔴",
+        "cancelled": "⚪",
+        "unknown": "⚫",
     }
 
-    icon = status_colors.get(status.lower(), '⚫')
+    icon = status_colors.get(status.lower(), "⚫")
 
     if size == "small":
         return f"{icon} {status}"
@@ -68,21 +60,17 @@ def display_kpi_row(metrics: dict, columns: Optional[int] = None):
             if isinstance(value, dict):
                 display_metric_card(
                     label=label,
-                    value=value.get('value', '-'),
-                    delta=value.get('delta'),
-                    delta_color=value.get('delta_color', 'normal'),
-                    help_text=value.get('help'),
-                    icon=value.get('icon')
+                    value=value.get("value", "-"),
+                    delta=value.get("delta"),
+                    delta_color=value.get("delta_color", "normal"),
+                    help_text=value.get("help"),
+                    icon=value.get("icon"),
                 )
             else:
                 st.metric(label=label, value=value)
 
 
-def display_progress_bar(
-    label: str,
-    progress: float,
-    show_percentage: bool = True
-):
+def display_progress_bar(label: str, progress: float, show_percentage: bool = True):
     """Display a progress bar with label"""
 
     st.text(label)
@@ -92,11 +80,7 @@ def display_progress_bar(
         st.caption(f"{progress * 100:.1f}%")
 
 
-def display_health_indicator(
-    component: str,
-    is_healthy: bool,
-    details: Optional[str] = None
-):
+def display_health_indicator(component: str, is_healthy: bool, details: Optional[str] = None):
     """Display a health status indicator"""
 
     if is_healthy:
@@ -105,11 +89,7 @@ def display_health_indicator(
         st.error(f"❌ {component}: Unhealthy" + (f" ({details})" if details else ""))
 
 
-def display_alert(
-    message: str,
-    alert_type: str = "info",
-    icon: Optional[str] = None
-):
+def display_alert(message: str, alert_type: str = "info", icon: Optional[str] = None):
     """Display an alert message"""
 
     if icon:

mcli/ml/dashboard/components/tables.py

@@ -1,8 +1,9 @@
 """Reusable table components"""
 
-import streamlit as st
+from typing import Any, Callable, List, Optional
+
 import pandas as pd
-from typing import Optional, List, Callable, Any
+import streamlit as st
 
 
 def display_dataframe_with_search(
@@ -10,7 +11,7 @@ def display_dataframe_with_search(
     search_columns: Optional[List[str]] = None,
     default_sort_column: Optional[str] = None,
     page_size: int = 20,
-    key_prefix: str = "table"
+    key_prefix: str = "table",
 ) -> pd.DataFrame:
     """Display a dataframe with search and pagination"""
 
@@ -23,7 +24,7 @@ def display_dataframe_with_search(
         search_term = st.text_input(
             "🔍 Search",
             key=f"{key_prefix}_search",
-            placeholder=f"Search in: {', '.join(search_columns)}"
+            placeholder=f"Search in: {', '.join(search_columns)}",
         )
 
         if search_term:
@@ -44,11 +45,7 @@ def display_dataframe_with_search(
     if len(df) > page_size:
         total_pages = (len(df) - 1) // page_size + 1
         page = st.number_input(
-            "Page",
-            min_value=1,
-            max_value=total_pages,
-            value=1,
-            key=f"{key_prefix}_page"
+            "Page", min_value=1, max_value=total_pages, value=1, key=f"{key_prefix}_page"
         )
         start_idx = (page - 1) * page_size
         end_idx = start_idx + page_size
@@ -63,9 +60,7 @@ def display_dataframe_with_search(
 
 
 def display_filterable_dataframe(
-    df: pd.DataFrame,
-    filter_columns: Optional[dict] = None,
-    key_prefix: str = "filter"
+    df: pd.DataFrame, filter_columns: Optional[dict] = None, key_prefix: str = "filter"
 ) -> pd.DataFrame:
     """Display a dataframe with column-specific filters"""
 
@@ -88,31 +83,32 @@ def display_filterable_dataframe(
                             col_name,
                             options=unique_values,
                             default=unique_values,
-                            key=f"{key_prefix}_{col_name}"
+                            key=f"{key_prefix}_{col_name}",
                         )
                         if selected:
                             df = df[df[col_name].isin(selected)]
 
                     elif filter_type == "text":
-                        search_text = st.text_input(
-                            col_name,
-                            key=f"{key_prefix}_{col_name}"
-                        )
+                        search_text = st.text_input(col_name, key=f"{key_prefix}_{col_name}")
                         if search_text:
-                            df = df[df[col_name].astype(str).str.contains(search_text, case=False, na=False)]
+                            df = df[
+                                df[col_name]
+                                .astype(str)
+                                .str.contains(search_text, case=False, na=False)
+                            ]
 
                     elif filter_type == "date_range":
                         if pd.api.types.is_datetime64_any_dtype(df[col_name]):
                             min_date = df[col_name].min()
                             max_date = df[col_name].max()
                             date_range = st.date_input(
-                                col_name,
-                                value=(min_date, max_date),
-                                key=f"{key_prefix}_{col_name}"
+                                col_name, value=(min_date, max_date), key=f"{key_prefix}_{col_name}"
                             )
                             if len(date_range) == 2:
-                                df = df[(df[col_name] >= pd.Timestamp(date_range[0])) &
-                                       (df[col_name] <= pd.Timestamp(date_range[1]))]
+                                df = df[
+                                    (df[col_name] >= pd.Timestamp(date_range[0]))
+                                    & (df[col_name] <= pd.Timestamp(date_range[1]))
+                                ]
 
     st.dataframe(df, width="stretch")
 
@@ -120,10 +116,7 @@ def display_filterable_dataframe(
 
 
 def display_table_with_actions(
-    df: pd.DataFrame,
-    actions: List[dict],
-    row_id_column: str = "id",
-    key_prefix: str = "action"
+    df: pd.DataFrame, actions: List[dict], row_id_column: str = "id", key_prefix: str = "action"
 ):
     """Display a table with action buttons for each row
 
@@ -145,14 +138,13 @@ def display_table_with_actions(
             # Action buttons
             for action_idx, action in enumerate(actions):
                 with data_cols[action_idx + 1]:
-                    icon = action.get('icon', '')
-                    button_label = f"{icon} {action['label']}" if icon else action['label']
+                    icon = action.get("icon", "")
+                    button_label = f"{icon} {action['label']}" if icon else action["label"]
 
                     if st.button(
-                        button_label,
-                        key=f"{key_prefix}_{row[row_id_column]}_{action_idx}"
+                        button_label, key=f"{key_prefix}_{row[row_id_column]}_{action_idx}"
                     ):
-                        action['callback'](row)
+                        action["callback"](row)
 
             st.divider()
 
@@ -162,7 +154,7 @@ def display_expandable_table(
     summary_columns: List[str],
     detail_callback: Callable[[Any], None],
     row_id_column: str = "id",
-    key_prefix: str = "expand"
+    key_prefix: str = "expand",
 ):
     """Display a table where each row can be expanded for details"""
 
@@ -183,7 +175,7 @@ def export_dataframe(
     df: pd.DataFrame,
     filename: str = "data",
     formats: List[str] = ["csv", "json"],
-    key_prefix: str = "export"
+    key_prefix: str = "export",
 ):
     """Provide export buttons for a dataframe"""
 
@@ -195,34 +187,34 @@ def export_dataframe(
     for idx, fmt in enumerate(formats):
         with cols[idx]:
             if fmt == "csv":
-                csv = df.to_csv(index=False).encode('utf-8')
+                csv = df.to_csv(index=False).encode("utf-8")
                 st.download_button(
                     label="📥 Download CSV",
                     data=csv,
                     file_name=f"{filename}.csv",
                     mime="text/csv",
-                    key=f"{key_prefix}_csv"
+                    key=f"{key_prefix}_csv",
                 )
             elif fmt == "json":
-                json_str = df.to_json(orient='records', indent=2)
+                json_str = df.to_json(orient="records", indent=2)
                 st.download_button(
                     label="📥 Download JSON",
                     data=json_str,
                     file_name=f"{filename}.json",
                     mime="application/json",
-                    key=f"{key_prefix}_json"
+                    key=f"{key_prefix}_json",
                 )
             elif fmt == "excel":
                 # Requires openpyxl
                 try:
                     buffer = BytesIO()
-                    df.to_excel(buffer, index=False, engine='openpyxl')
+                    df.to_excel(buffer, index=False, engine="openpyxl")
                     st.download_button(
                         label="📥 Download Excel",
                         data=buffer.getvalue(),
                         file_name=f"{filename}.xlsx",
                         mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-                        key=f"{key_prefix}_excel"
+                        key=f"{key_prefix}_excel",
                     )
                 except ImportError:
                     st.warning("Excel export requires openpyxl package")