Update sozo_gen.py

sozo_gen.py

@@ -259,6 +259,15 @@ def concat_media(file_paths: List[str], output_path: Path):
 
 # --- Main Business Logic Functions for Flask ---
 
+# ADD THIS NEW HELPER FUNCTION SOMEWHERE NEAR THE TOP OF THE FILE
+def sanitize_for_firebase_key(text: str) -> str:
+    """Replaces Firebase-forbidden characters in a string with underscores."""
+    forbidden_chars = ['.', '$', '#', '[', ']', '/']
+    for char in forbidden_chars:
+        text = text.replace(char, '_')
+    return text
+
+# REPLACE THE OLD generate_report_draft WITH THIS CORRECTED VERSION
 def generate_report_draft(buf, name: str, ctx: str, uid: str, project_id: str, bucket):
     logging.info(f"Generating report draft for project {project_id}")
     df = load_dataframe_safely(buf, name)
@@ -279,20 +288,31 @@ def generate_report_draft(buf, name: str, ctx: str, uid: str, project_id: str, b
     chart_descs = extract_chart_tags(md)[:MAX_CHARTS]
     chart_urls = {}
     chart_generator = ChartGenerator(llm, df)
+
     for desc in chart_descs:
+        # Create a safe key for Firebase
+        safe_desc = sanitize_for_firebase_key(desc)
+
+        # Replace the original description in the markdown with the safe one
+        md = md.replace(f'<generate_chart: "{desc}">', f'<generate_chart: "{safe_desc}">')
+        md = md.replace(f'<generate_chart: {desc}>', f'<generate_chart: "{safe_desc}">') # Handle no quotes case
+
         with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
             img_path = Path(temp_file.name)
             try:
-                chart_spec = chart_generator.generate_chart_spec(desc)
+                chart_spec = chart_generator.generate_chart_spec(desc) # Still generate spec from original desc
                 if execute_chart_spec(chart_spec, df, img_path):
                     blob_name = f"sozo_projects/{uid}/{project_id}/charts/{uuid.uuid4().hex}.png"
                     blob = bucket.blob(blob_name)
                     blob.upload_from_filename(str(img_path))
-                    chart_urls[desc] = blob.public_url
-                    logging.info(f"Uploaded chart '{desc}' to {blob.public_url}")
+                    
+                    # Use the safe key in the dictionary
+                    chart_urls[safe_desc] = blob.public_url
+                    logging.info(f"Uploaded chart '{desc}' to {blob.public_url} with safe key '{safe_desc}'")
             finally:
                 if os.path.exists(img_path):
                     os.unlink(img_path)
+
     return {"raw_md": md, "chartUrls": chart_urls}
 
 def generate_single_chart(df: pd.DataFrame, description: str, uid: str, project_id: str, bucket):