Update main.py

main.py

@@ -234,12 +234,38 @@ def generate_sozo_report(project_id):
         
         logger.info(f"Report draft generated for project {project_id}")
         
-        update_data = {
-            'status': 'draft',
-            'rawMarkdown': draft_data['raw_md'],
-            'chartUrls': draft_data['chartUrls']
-        }
-        project_ref.update(update_data)
+        # Clean and validate data before updating Firebase
+        try:
+            # Ensure rawMarkdown is a string and handle any encoding issues
+            raw_markdown = draft_data.get('raw_md', '')
+            if raw_markdown:
+                # Clean any problematic characters that might cause JSON parsing issues
+                raw_markdown = str(raw_markdown).replace('\x00', '').replace('\ufeff', '')
+                logger.info(f"Raw markdown length: {len(raw_markdown)}")
+            
+            # Ensure chartUrls is a proper list/dict
+            chart_urls = draft_data.get('chartUrls', [])
+            if isinstance(chart_urls, (list, dict)):
+                logger.info(f"Chart URLs count: {len(chart_urls)}")
+            else:
+                logger.warning(f"Chart URLs is not a list/dict: {type(chart_urls)}")
+                chart_urls = []
+            
+            update_data = {
+                'status': 'draft',
+                'rawMarkdown': raw_markdown,
+                'chartUrls': chart_urls
+            }
+            
+            logger.info(f"Updating project {project_id} with cleaned data")
+            project_ref.update(update_data)
+            
+        except Exception as update_error:
+            logger.error(f"Error preparing update data for project {project_id}: {str(update_error)}")
+            logger.error(f"Draft data keys: {list(draft_data.keys()) if draft_data else 'None'}")
+            logger.error(f"Raw markdown type: {type(draft_data.get('raw_md'))}")
+            logger.error(f"Chart URLs type: {type(draft_data.get('chartUrls'))}")
+            raise update_error
         
         logger.info(f"Project {project_id} updated with draft data")