Update charts.py

charts.py

@@ -4,35 +4,27 @@ import plotly.express as px
 import pandas as pd
 import numpy as np
 
-DARK_BG     = "#FFFFE3"
-CARD_BG     = "#FFFFFF"
-BORDER      = "#BDDDFC"
-TEXT_MAIN   = "#2d2d2d"
-TEXT_DIM    = "#555555"
-PRIMARY     = "#269ccc"
-CYAN        = "#269ccc"
-
-POS_COLOR   = "#88BDF2"
-NEG_COLOR   = "#6A89A7"
-NEU_COLOR   = "#CBCBCB"
-
-ALGAE       = "#9ed2c5"
-GREEN       = "#16a34a"
-RED         = "#dc2626"
-AMBER       = "#d97706"
-PURPLE      = "#7c3aed"
-BLUE        = "#2563eb"
-
-_FONT       = dict(family="'DM Sans', 'Nunito', sans-serif", color=TEXT_MAIN, size=12)
-_TITLE_FONT = dict(family="'DM Sans', sans-serif", color=TEXT_MAIN, size=13)
-_TICK_FONT  = dict(family="'DM Sans', sans-serif", color=TEXT_MAIN, size=11)
-_LEGEND_FONT= dict(family="'DM Sans', sans-serif", color=TEXT_MAIN, size=11)
+CREAM        = "#FFFFE3"
+CARD_BG      = "#FFFFFF"
+BORDER       = "#BDDDFC"
+TEXT_MAIN    = "#4A4A4A"
+TEXT_DIM     = "#7b7b7b"
+INK_DARK     = "#384959"
+PRIMARY      = "#269ccc"
+STORMY_SKY   = "#88BDF2"
+STORMY_SLATE = "#6A89A7"
+INK_GREY     = "#CBCBCB"
+GREEN        = "#2e9e6b"
+RED          = "#c0392b"
+AMBER        = "#d4841a"
+PURPLE       = "#7C6DB5"
+BLUE         = "#269ccc"
 
 PLOTLY_LAYOUT = dict(
-    paper_bgcolor="rgba(255,255,227,0)",
+    paper_bgcolor="rgba(0,0,0,0)",
     plot_bgcolor="rgba(189,221,252,0.13)",
-    font=_FONT,
-    margin=dict(l=20, r=20, t=44, b=20),
+    font=dict(family="'DM Mono', monospace", color=TEXT_MAIN, size=12),
+    margin=dict(l=20, r=20, t=40, b=20),
 )
 
 
@@ -48,26 +40,26 @@ def misinfo_gauge(score: float, label: str) -> go.Figure:
     fig = go.Figure(go.Indicator(
         mode="gauge+number+delta",
         value=pct,
-        number={"suffix": "%", "font": {"size": 32, "color": bar_color, "family": "'Nunito', sans-serif"}},
+        number={"suffix": "%", "font": {"size": 32, "color": bar_color, "family": "'DM Mono', monospace"}},
         delta={"reference": 50, "increasing": {"color": RED}, "decreasing": {"color": GREEN}},
-        title={"text": label, "font": {"size": 13, "color": TEXT_MAIN}},
+        title={"text": label, "font": {"size": 13, "color": TEXT_DIM}},
         gauge={
             "axis": {
                 "range": [0, 100],
                 "tickwidth": 1,
                 "tickcolor": BORDER,
-                "tickfont": {"color": TEXT_MAIN, "size": 10},
+                "tickfont": {"color": TEXT_DIM, "size": 10},
             },
             "bar": {"color": bar_color, "thickness": 0.3},
             "bgcolor": CARD_BG,
             "borderwidth": 0,
             "steps": [
-                {"range": [0, 35],  "color": "#0d1f18"},
-                {"range": [35, 65], "color": "#1f1a0d"},
-                {"range": [65, 100],"color": "#1f0d0d"},
+                {"range": [0, 35],   "color": "#e8f7ef"},
+                {"range": [35, 65],  "color": "#fdf3e3"},
+                {"range": [65, 100], "color": "#fdecea"},
             ],
             "threshold": {
-                "line": {"color": TEXT_MAIN, "width": 2},
+                "line": {"color": INK_DARK, "width": 2},
                 "thickness": 0.75,
                 "value": pct,
             },
@@ -78,35 +70,34 @@ def misinfo_gauge(score: float, label: str) -> go.Figure:
 
 
 def sentiment_donut(summary: Dict) -> go.Figure:
-    labels  = ["Positively Engagement", "Neutral", "Negatively Engagement"]
-    values  = [summary["Positively Engagement"], summary["Neutral"], summary["Negatively Engagement"]]
-    colors  = [POS_COLOR, NEU_COLOR, NEG_COLOR]
+    labels = ["Positively Engagement", "Neutral", "Negatively Engagement"]
+    values = [summary["POSITIVE"], summary["NEUTRAL"], summary["NEGATIVE"]]
+    colors = [STORMY_SKY, INK_GREY, STORMY_SLATE]
 
     fig = go.Figure(go.Pie(
         labels=labels,
         values=values,
         hole=0.62,
-        marker=dict(colors=colors, line=dict(color=DARK_BG, width=3)),
+        marker=dict(colors=colors, line=dict(color=CARD_BG, width=3)),
         textinfo="label+percent",
-        textfont=dict(family="'DM Sans', sans-serif", size=11, color=TEXT_MAIN),
+        textfont=dict(family="'DM Mono', monospace", size=11, color=TEXT_MAIN),
         hovertemplate="<b>%{label}</b><br>%{value} comments (%{percent})<extra></extra>",
         rotation=90,
     ))
 
     avg = summary.get("avg_compound", 0)
-    overall = "😊 Positively Engaged" if avg > 0.05 else ("😟 Negatively Engaged" if avg < -0.05 else "😐 Mixed")
+    overall = "😊 Positive" if avg > 0.05 else ("😟 Negative" if avg < -0.05 else "😐 Mixed")
     fig.add_annotation(
         text=f"<b>{overall}</b><br><span style='font-size:11px;color:{TEXT_DIM}'>{summary['total']} comments</span>",
         x=0.5, y=0.5,
         showarrow=False,
-        font=dict(size=13, color=TEXT_MAIN, family="'DM Sans', sans-serif"),
+        font=dict(size=13, color=TEXT_MAIN, family="'DM Mono', monospace"),
         align="center",
     )
     fig.update_layout(
         **PLOTLY_LAYOUT,
         height=300,
-        legend=dict(orientation="h", y=-0.08, font=_LEGEND_FONT),
-        title=dict(text="Sentiment Breakdown", font=_TITLE_FONT, x=0),
+        legend=dict(orientation="h", y=-0.08, font=dict(size=11, color=TEXT_MAIN)),
     )
     return fig
 
@@ -114,7 +105,7 @@ def sentiment_donut(summary: Dict) -> go.Figure:
 def keyword_bar(
     keywords: List[Tuple[str, float]],
     title: str = "Top Keywords",
-    color: str = CYAN,
+    color: str = PRIMARY,
 ) -> go.Figure:
     if not keywords:
         return _empty_fig(title)
@@ -129,33 +120,25 @@ def keyword_bar(
         orientation="h",
         marker=dict(
             color=norm,
-            colorscale=[[0, f"{PRIMARY}33"], [1, PRIMARY]],
-            line=dict(color=DARK_BG, width=1.5),
+            colorscale=[[0, f"{color}44"], [1, color]],
+            line=dict(width=0),
         ),
         text=[f"{w:.0f}" for w in weights],
         textposition="inside",
-        textfont=dict(size=10, color="#FFFFFF", family="'DM Sans', sans-serif"),
+        textfont=dict(size=10, color=CARD_BG),
         hovertemplate="<b>%{y}</b><br>Weight: %{text}<extra></extra>",
     ))
     fig.update_layout(
         **PLOTLY_LAYOUT,
-        title=dict(text=title, font=_TITLE_FONT, x=0),
+        title=dict(text=title, font=dict(size=13, color=INK_DARK), x=0),
         height=380,
         yaxis=dict(
             autorange="reversed",
-            tickfont=_TICK_FONT,
-            title_font=_TITLE_FONT,
+            tickfont=dict(size=11, color=TEXT_MAIN),
             gridcolor=BORDER,
-            showgrid=True,
-            gridwidth=1,
         ),
-        xaxis=dict(
-            showticklabels=False,
-            gridcolor=BORDER,
-            showgrid=False,
-        ),
-        bargap=0.3,
-        plot_bgcolor="rgba(189,221,252,0.13)",
+        xaxis=dict(showticklabels=False, gridcolor=BORDER),
+        bargap=0.35,
     )
     return fig
 
@@ -169,7 +152,7 @@ def stream_trust_bars(stream_details: Dict) -> go.Figure:
         x=values,
         y=[l.replace("_", " ").title() for l in labels],
         orientation="h",
-        marker=dict(color=colors, line=dict(color=DARK_BG, width=1.5)),
+        marker=dict(color=colors, line=dict(color=CARD_BG, width=1)),
         text=[f"{v}%" for v in values],
         textposition="outside",
         textfont=dict(size=11, color=TEXT_MAIN),
@@ -177,10 +160,10 @@ def stream_trust_bars(stream_details: Dict) -> go.Figure:
     ))
     fig.update_layout(
         **PLOTLY_LAYOUT,
-        title=dict(text="Per-Stream Analysis", font=_TITLE_FONT, x=0),
+        title=dict(text="Per-Stream Analysis", font=dict(size=13, color=INK_DARK), x=0),
         height=220,
-        xaxis=dict(range=[0, 110], showticklabels=False, gridcolor=BORDER, tickfont=_TICK_FONT),
-        yaxis=dict(tickfont=_TICK_FONT),
+        xaxis=dict(range=[0, 120], showticklabels=False, gridcolor=BORDER),
+        yaxis=dict(tickfont=dict(size=11, color=TEXT_MAIN)),
         bargap=0.4,
     )
     return fig
@@ -206,12 +189,12 @@ def modality_misinfo_distribution(modality_analysis: Dict) -> go.Figure:
         y=misinfo_pcts,
         marker=dict(
             color=[RED, RED, RED],
-            opacity=0.88,
-            line=dict(color=DARK_BG, width=1.5),
+            opacity=0.90,
+            line=dict(color=CARD_BG, width=2),
         ),
         text=[f"{v:.1f}%" for v in misinfo_pcts],
         textposition="outside",
-        textfont=dict(size=11, color=RED),
+        textfont=dict(size=11, color=RED, family="'DM Mono', monospace"),
         customdata=logit_tips,
         hovertemplate=(
             "<b>%{x} — Misinformation</b><br>"
@@ -226,12 +209,12 @@ def modality_misinfo_distribution(modality_analysis: Dict) -> go.Figure:
         y=credible_pcts,
         marker=dict(
             color=[GREEN, GREEN, GREEN],
-            opacity=0.88,
-            line=dict(color=DARK_BG, width=1.5),
+            opacity=0.90,
+            line=dict(color=CARD_BG, width=2),
         ),
         text=[f"{v:.1f}%" for v in credible_pcts],
         textposition="outside",
-        textfont=dict(size=11, color=GREEN),
+        textfont=dict(size=11, color=GREEN, family="'DM Mono', monospace"),
         customdata=logit_tips,
         hovertemplate=(
             "<b>%{x} — Credible</b><br>"
@@ -244,34 +227,31 @@ def modality_misinfo_distribution(modality_analysis: Dict) -> go.Figure:
         **PLOTLY_LAYOUT,
         title=dict(
             text="Modality Misinformation Distribution",
-            font=_TITLE_FONT,
+            font=dict(size=13, color=INK_DARK),
             x=0,
         ),
         barmode="group",
-        height=280,
+        height=300,
         xaxis=dict(
-            title="Modality",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
+            title=dict(text="Modality", font=dict(color=TEXT_MAIN)),
+            tickfont=dict(size=12, color=TEXT_MAIN),
             gridcolor=BORDER,
         ),
         yaxis=dict(
-            title="Softmax Score (%)",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
-            range=[0, 115],
+            title=dict(text="Softmax Score (%)", font=dict(color=TEXT_MAIN)),
+            range=[0, 125],
             gridcolor=BORDER,
             ticksuffix="%",
+            tickfont=dict(color=TEXT_MAIN),
         ),
         legend=dict(
             orientation="h",
-            y=1.12,
-            font=_LEGEND_FONT,
-            bgcolor="rgba(255,255,227,0)",
+            y=1.14,
+            font=dict(size=11, color=TEXT_MAIN),
+            bgcolor="rgba(0,0,0,0)",
         ),
         bargap=0.22,
         bargroupgap=0.06,
-        plot_bgcolor="rgba(189,221,252,0.13)",
     )
     return fig
 
@@ -291,12 +271,12 @@ def trust_score_by_modality(modality_analysis: Dict) -> go.Figure:
         y=trust_vals,
         marker=dict(
             color=bar_colors,
-            opacity=0.88,
-            line=dict(color=DARK_BG, width=1.5),
+            opacity=0.90,
+            line=dict(color=CARD_BG, width=2),
         ),
         text=[f"{v:.1f}%" for v in trust_vals],
         textposition="outside",
-        textfont=dict(size=11, color=TEXT_MAIN),
+        textfont=dict(size=11, color=TEXT_MAIN, family="'DM Mono', monospace"),
         hovertemplate=(
             "<b>%{x}</b><br>"
             "Trust Level: %{y:.2f}%<br>"
@@ -308,7 +288,7 @@ def trust_score_by_modality(modality_analysis: Dict) -> go.Figure:
     for level, label, color in [(80, "High Trust", GREEN), (50, "Threshold", AMBER)]:
         fig.add_hline(
             y=level,
-            line=dict(color=color, width=1, dash="dot"),
+            line=dict(color=color, width=1.5, dash="dot"),
             annotation_text=label,
             annotation_position="right",
             annotation_font=dict(size=9, color=color),
@@ -318,26 +298,23 @@ def trust_score_by_modality(modality_analysis: Dict) -> go.Figure:
         **PLOTLY_LAYOUT,
         title=dict(
             text="Trust Score by Modality",
-            font=_TITLE_FONT,
+            font=dict(size=13, color=INK_DARK),
             x=0,
         ),
-        height=280,
+        height=300,
         xaxis=dict(
-            title="Modality",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
+            title=dict(text="Modality", font=dict(color=TEXT_MAIN)),
+            tickfont=dict(size=12, color=TEXT_MAIN),
             gridcolor=BORDER,
         ),
         yaxis=dict(
-            title="Trust Level (%)",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
-            range=[0, 115],
+            title=dict(text="Trust Level (%)", font=dict(color=TEXT_MAIN)),
+            range=[0, 120],
             gridcolor=BORDER,
             ticksuffix="%",
+            tickfont=dict(color=TEXT_MAIN),
         ),
         bargap=0.38,
-        plot_bgcolor="rgba(189,221,252,0.13)",
     )
     return fig
 
@@ -359,12 +336,12 @@ def uncertainty_analysis(modality_analysis: Dict) -> go.Figure:
         y=uncertainty_vals,
         marker=dict(
             color=bar_colors,
-            opacity=0.88,
-            line=dict(color=DARK_BG, width=1.5),
+            opacity=0.90,
+            line=dict(color=CARD_BG, width=2),
         ),
         text=[f"{v:.1f}%" for v in uncertainty_vals],
         textposition="outside",
-        textfont=dict(size=11, color=TEXT_MAIN),
+        textfont=dict(size=11, color=TEXT_MAIN, family="'DM Mono', monospace"),
         customdata=[[f"p_misinfo={m:.1f}%"] for m in misinfo_pcts],
         hovertemplate=(
             "<b>%{x}</b><br>"
@@ -377,14 +354,14 @@ def uncertainty_analysis(modality_analysis: Dict) -> go.Figure:
 
     fig.add_hline(
         y=100,
-        line=dict(color=RED, width=1, dash="dot"),
+        line=dict(color=RED, width=1.5, dash="dot"),
         annotation_text="Max Entropy (no signal)",
         annotation_position="right",
         annotation_font=dict(size=9, color=RED),
     )
     fig.add_hline(
         y=50,
-        line=dict(color=AMBER, width=1, dash="dot"),
+        line=dict(color=AMBER, width=1.5, dash="dot"),
         annotation_text="Mid Uncertainty",
         annotation_position="right",
         annotation_font=dict(size=9, color=AMBER),
@@ -394,26 +371,23 @@ def uncertainty_analysis(modality_analysis: Dict) -> go.Figure:
         **PLOTLY_LAYOUT,
         title=dict(
             text="Uncertainty Analysis (Shannon Entropy)",
-            font=_TITLE_FONT,
+            font=dict(size=13, color=INK_DARK),
             x=0,
         ),
-        height=280,
+        height=300,
         xaxis=dict(
-            title="Modality",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
+            title=dict(text="Modality", font=dict(color=TEXT_MAIN)),
+            tickfont=dict(size=12, color=TEXT_MAIN),
             gridcolor=BORDER,
         ),
         yaxis=dict(
-            title="Uncertainty (%)",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
-            range=[0, 120],
+            title=dict(text="Uncertainty (%)", font=dict(color=TEXT_MAIN)),
+            range=[0, 130],
             gridcolor=BORDER,
             ticksuffix="%",
+            tickfont=dict(color=TEXT_MAIN),
         ),
         bargap=0.38,
-        plot_bgcolor="rgba(189,221,252,0.13)",
     )
     return fig
 
@@ -423,20 +397,16 @@ def sentiment_timeline(comments_df: pd.DataFrame, sentiments: List[Dict]) -> go.
         return _empty_fig("Comment Sentiment Distribution")
 
     df = comments_df.copy()
-    df["compound"] = [s.get("compound", 0) for s in sentiments]
-    df["label"]    = [s.get("label", "Neutral") for s in sentiments]
-    df["color"]    = df["label"].map({
-        "Positively Engagement": POS_COLOR,
-        "Negatively Engagement": NEG_COLOR,
-        "Neutral":               NEU_COLOR,
-    })
+    df["compound"]   = [s.get("compound", 0) for s in sentiments]
+    df["label"]      = [s.get("label", "NEUTRAL") for s in sentiments]
+    df["color"]      = df["label"].map({"POSITIVE": STORMY_SKY, "NEGATIVE": STORMY_SLATE, "NEUTRAL": INK_GREY})
     df["text_short"] = df["text"].str[:80] + "…"
 
     fig = go.Figure()
-    for lbl, clr in [
-        ("Positively Engagement", POS_COLOR),
-        ("Negatively Engagement", NEG_COLOR),
-        ("Neutral",               NEU_COLOR),
+    for lbl, clr, disp in [
+        ("POSITIVE", STORMY_SKY,   "Positively Engagement"),
+        ("NEGATIVE", STORMY_SLATE, "Negatively Engagement"),
+        ("NEUTRAL",  INK_GREY,     "Neutral"),
     ]:
         sub = df[df["label"] == lbl]
         if sub.empty:
@@ -445,11 +415,11 @@ def sentiment_timeline(comments_df: pd.DataFrame, sentiments: List[Dict]) -> go.
             x=sub.index,
             y=sub["compound"],
             mode="markers",
-            name=lbl,
+            name=disp,
             marker=dict(
                 size=np.clip(np.log1p(sub["likes"].fillna(0)) * 4 + 4, 4, 20),
                 color=clr,
-                opacity=0.75,
+                opacity=0.80,
                 line=dict(width=0),
             ),
             text=sub["text_short"],
@@ -459,33 +429,11 @@ def sentiment_timeline(comments_df: pd.DataFrame, sentiments: List[Dict]) -> go.
     fig.add_hline(y=0, line=dict(color=BORDER, width=1, dash="dot"))
     fig.update_layout(
         **PLOTLY_LAYOUT,
-        title=dict(text="Comment Sentiment (size = likes)", font=_TITLE_FONT, x=0),
+        title=dict(text="Comment Sentiment (size = likes)", font=dict(size=13, color=INK_DARK), x=0),
         height=320,
-        xaxis=dict(
-            title="Comment index",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
-            gridcolor=BORDER,
-            showgrid=False,
-        ),
-        yaxis=dict(
-            title="Compound score",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
-            gridcolor=BORDER,
-            range=[-1.1, 1.1],
-            showgrid=True,
-            gridwidth=1,
-        ),
-        legend=dict(
-            orientation="h",
-            y=1.12,
-            font=_LEGEND_FONT,
-            bgcolor="rgba(255,255,255,0.85)",
-            bordercolor=BORDER,
-            borderwidth=1,
-        ),
-        plot_bgcolor="rgba(189,221,252,0.13)",
+        xaxis=dict(title="Comment index", gridcolor=BORDER, showgrid=False, tickfont=dict(color=TEXT_MAIN)),
+        yaxis=dict(title="Compound score", gridcolor=BORDER, range=[-1.1, 1.1], tickfont=dict(color=TEXT_MAIN)),
+        legend=dict(orientation="h", y=1.14, font=dict(size=11, color=TEXT_MAIN)),
     )
     return fig
 
@@ -509,15 +457,9 @@ def keyword_comparison(
         fig.add_trace(go.Bar(
             name="Positively Engagement",
             y=list(pw),
-            x=[v/max_p*100 for v in pv],
+            x=[v / max_p * 100 for v in pv],
             orientation="h",
-            marker=dict(
-                color=POS_COLOR,
-                line=dict(color=DARK_BG, width=1.5),
-            ),
-            text=[f"{v/max_p*100:.0f}" for v in pv],
-            textposition="outside",
-            textfont=dict(size=10, color=TEXT_MAIN),
+            marker_color=STORMY_SKY,
             hovertemplate="<b>%{y}</b><br>Score: %{x:.1f}<extra></extra>",
         ))
 
@@ -527,42 +469,27 @@ def keyword_comparison(
         fig.add_trace(go.Bar(
             name="Negatively Engagement",
             y=list(nw),
-            x=[-v/max_n*100 for v in nv],
+            x=[-v / max_n * 100 for v in nv],
             orientation="h",
-            marker=dict(
-                color=NEG_COLOR,
-                line=dict(color=DARK_BG, width=1.5),
-            ),
-            text=[f"{v/max_n*100:.0f}" for v in nv],
-            textposition="outside",
-            textfont=dict(size=10, color=TEXT_MAIN),
+            marker_color=STORMY_SLATE,
             hovertemplate="<b>%{y}</b><br>Score: %{x:.1f}<extra></extra>",
         ))
 
     fig.update_layout(
         **PLOTLY_LAYOUT,
-        title=dict(text="Sentiment-Weighted Keywords", font=_TITLE_FONT, x=0),
+        title=dict(text="Sentiment-Weighted Keywords", font=dict(size=13, color=INK_DARK), x=0),
         height=360,
         barmode="overlay",
         xaxis=dict(
             title="← Negatively Engagement  |  Positively Engagement →",
-            title_font=_TITLE_FONT,
-            tickfont=_TICK_FONT,
             gridcolor=BORDER,
             zeroline=True,
-            zerolinecolor=BORDER,
+            zerolinecolor=INK_DARK,
             zerolinewidth=2,
+            tickfont=dict(color=TEXT_MAIN),
         ),
-        yaxis=dict(tickfont=_TICK_FONT),
-        legend=dict(
-            orientation="h",
-            y=1.1,
-            font=_LEGEND_FONT,
-            bgcolor="rgba(255,255,255,0.85)",
-            bordercolor=BORDER,
-            borderwidth=1,
-        ),
-        plot_bgcolor="rgba(189,221,252,0.13)",
+        yaxis=dict(tickfont=dict(size=10, color=TEXT_MAIN)),
+        legend=dict(orientation="h", y=1.1, font=dict(size=11, color=TEXT_MAIN)),
     )
     return fig
 
@@ -573,12 +500,7 @@ def _empty_fig(title: str) -> go.Figure:
         text="No data available",
         x=0.5, y=0.5,
         showarrow=False,
-        font=dict(size=14, color=TEXT_MAIN),
-    )
-    fig.update_layout(
-        **PLOTLY_LAYOUT,
-        title=dict(text=title, font=_TITLE_FONT, x=0),
-        height=250,
-        plot_bgcolor="rgba(189,221,252,0.13)",
+        font=dict(size=14, color=TEXT_DIM),
     )
+    fig.update_layout(**PLOTLY_LAYOUT, title=dict(text=title, x=0, font=dict(color=INK_DARK)), height=250)
     return fig