Update sozo_gen.py

sozo_gen.py

@@ -570,6 +570,33 @@ def enhance_data_context(df: pd.DataFrame, ctx_dict: Dict) -> Dict[str, Any]:
     
     return enhanced
 
+def create_chart_safe_context(enhanced_ctx: Dict) -> Dict:
+    """
+    Create a chart-generator-safe version of enhanced context
+    by ensuring all values are JSON serializable
+    """
+    def make_json_safe(obj):
+        if isinstance(obj, bool):
+            return bool(obj)
+        elif isinstance(obj, (np.integer, np.floating)):
+            return float(obj)
+        elif isinstance(obj, np.ndarray):
+            return obj.tolist()
+        elif isinstance(obj, np.bool_):
+            return bool(obj)
+        elif isinstance(obj, dict):
+            return {k: make_json_safe(v) for k, v in obj.items()}
+        elif isinstance(obj, (list, tuple)):
+            return [make_json_safe(item) for item in obj]
+        elif pd.isna(obj):
+            return None
+        elif hasattr(obj, 'item'):  # numpy scalars
+            return obj.item()
+        else:
+            return obj
+    
+    return make_json_safe(enhanced_ctx)
+
 def generate_report_draft(buf, name: str, ctx: str, uid: str, project_id: str, bucket):
     """
     Enhanced autonomous report generation with intelligent narrative creation
@@ -595,7 +622,19 @@ def generate_report_draft(buf, name: str, ctx: str, uid: str, project_id: str, b
     # Extract and process charts
     chart_descs = extract_chart_tags(md)[:MAX_CHARTS]
     chart_urls = {}
-    chart_generator = ChartGenerator(llm, df)
+    
+    # Create a chart-safe context
+    chart_safe_ctx = create_chart_safe_context(enhanced_ctx)
+    
+    # Try to pass the safe context to ChartGenerator
+    try:
+        chart_generator = ChartGenerator(llm, df, chart_safe_ctx)
+    except TypeError:
+        # Fallback: if ChartGenerator doesn't accept enhanced_ctx parameter
+        chart_generator = ChartGenerator(llm, df)
+        # If it has an enhanced_ctx attribute, set it safely
+        if hasattr(chart_generator, 'enhanced_ctx'):
+            chart_generator.enhanced_ctx = chart_safe_ctx
     
     for desc in chart_descs:
         # Create a safe key for Firebase