rairo commited on
Commit
da9b96e
Β·
verified Β·
1 Parent(s): a2bd41d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -185,9 +185,10 @@ def generate_report(buf: bytes, name: str, ctx: str, key: str):
185
  "columns": list(df.columns),
186
  "user_ctx": ctx or "General business analysis",
187
  "full_dataframe": df.to_dict('records'),
188
- "data_types": df.dtypes.to_dict(),
189
- "missing_values": df.isnull().sum().to_dict(),
190
- "numeric_summary": df.describe().to_dict() if len(df.select_dtypes(include=['number']).columns) > 0 else {}
 
191
  }
192
 
193
  cols = ", ".join(ctx_dict["columns"][:6])
@@ -292,6 +293,7 @@ def generate_report(buf: bytes, name: str, ctx: str, key: str):
292
  }
293
 
294
 
 
295
  # ─── ANIMATION HELPERS ─────────────────────────────────────────────────────
296
  def animate_image_fade(img_cv2: np.ndarray, dur: float, out: Path, fps: int = FPS) -> str:
297
  frames = max(int(dur * fps), fps)
@@ -504,8 +506,9 @@ def generate_video(buf: bytes, name: str, ctx: str, key: str):
504
  "columns": list(df.columns),
505
  "user_ctx": ctx or "General business analysis",
506
  "full_dataframe": df.to_dict('records'),
507
- "data_types": df.dtypes.to_dict(),
508
- "numeric_summary": df.describe().to_dict() if len(df.select_dtypes(include=['number']).columns) > 0 else {}
 
509
  }
510
 
511
  script = llm.invoke(build_story_prompt(ctx_dict)).content
 
185
  "columns": list(df.columns),
186
  "user_ctx": ctx or "General business analysis",
187
  "full_dataframe": df.to_dict('records'),
188
+ "data_types": {col: str(dtype) for col, dtype in df.dtypes.to_dict().items()},
189
+ "missing_values": {col: int(count) for col, count in df.isnull().sum().to_dict().items()},
190
+ "numeric_summary": {col: {stat: float(val) for stat, val in stats.items()}
191
+ for col, stats in df.describe().to_dict().items()} if len(df.select_dtypes(include=['number']).columns) > 0 else {}
192
  }
193
 
194
  cols = ", ".join(ctx_dict["columns"][:6])
 
293
  }
294
 
295
 
296
+
297
  # ─── ANIMATION HELPERS ─────────────────────────────────────────────────────
298
  def animate_image_fade(img_cv2: np.ndarray, dur: float, out: Path, fps: int = FPS) -> str:
299
  frames = max(int(dur * fps), fps)
 
506
  "columns": list(df.columns),
507
  "user_ctx": ctx or "General business analysis",
508
  "full_dataframe": df.to_dict('records'),
509
+ "data_types": {col: str(dtype) for col, dtype in df.dtypes.to_dict().items()},
510
+ "numeric_summary": {col: {stat: float(val) for stat, val in stats.items()}
511
+ for col, stats in df.describe().to_dict().items()} if len(df.select_dtypes(include=['number']).columns) > 0 else {}
512
  }
513
 
514
  script = llm.invoke(build_story_prompt(ctx_dict)).content