sanjaystarc commited on
Commit
321cd22
Β·
verified Β·
1 Parent(s): ea42a15

Update core_agent.py

Browse files
Files changed (1) hide show
  1. core_agent.py +5 -5
core_agent.py CHANGED
@@ -28,9 +28,9 @@ warnings.filterwarnings("ignore")
28
  load_dotenv()
29
 
30
  # ─── Palette ─────────────────────────────────────────────────────────────────
31
- PALETTE = ["#111111", "#444444", "#777777", "#999999", "#BBBBBB", "#DDDDDD"]
32
- DARK_BG = "#FFFFFF"
33
- CARD_BG = "#F9F9F9"
34
 
35
 
36
  # ─── LLM Setup ───────────────────────────────────────────────────────────────
@@ -169,7 +169,7 @@ def make_plotly_chart(chart_type: str, df: pd.DataFrame, profile: dict,
169
  num_cols = profile["numeric_columns"]
170
  cat_cols = profile["categorical_columns"]
171
 
172
- template = "plotly_white"
173
 
174
  if chart_type == "correlation_heatmap" and len(num_cols) >= 2:
175
  corr = df[num_cols].corr().round(2)
@@ -276,7 +276,7 @@ def make_plotly_chart(chart_type: str, df: pd.DataFrame, profile: dict,
276
  fig.update_layout(
277
  paper_bgcolor=DARK_BG,
278
  plot_bgcolor=CARD_BG,
279
- font=dict(family="DM Sans, sans-serif", color="#111111"),
280
  margin=dict(l=40, r=40, t=60, b=40),
281
  )
282
  return fig
 
28
  load_dotenv()
29
 
30
  # ─── Palette ─────────────────────────────────────────────────────────────────
31
+ PALETTE = ["#6C63FF", "#FF6584", "#43E97B", "#F7971E", "#4FC3F7", "#CE93D8"]
32
+ DARK_BG = "#0F0F1A"
33
+ CARD_BG = "#1A1A2E"
34
 
35
 
36
  # ─── LLM Setup ───────────────────────────────────────────────────────────────
 
169
  num_cols = profile["numeric_columns"]
170
  cat_cols = profile["categorical_columns"]
171
 
172
+ template = "plotly_dark"
173
 
174
  if chart_type == "correlation_heatmap" and len(num_cols) >= 2:
175
  corr = df[num_cols].corr().round(2)
 
276
  fig.update_layout(
277
  paper_bgcolor=DARK_BG,
278
  plot_bgcolor=CARD_BG,
279
+ font=dict(family="DM Sans, sans-serif", color="#E0E0FF"),
280
  margin=dict(l=40, r=40, t=60, b=40),
281
  )
282
  return fig