Update main.py

main.py

@@ -125,6 +125,16 @@ logger = logging.getLogger(__name__)
 # 4. SOZO BUSINESS STUDIO API ENDPOINTS WITH LOGGING
 # -----------------------------------------------------------------------------
 
+import logging
+
+# Configure console logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+# -----------------------------------------------------------------------------
+# 4. SOZO BUSINESS STUDIO API ENDPOINTS WITH LOGGING
+# -----------------------------------------------------------------------------
+
 @app.route('/api/sozo/projects', methods=['POST'])
 def create_sozo_project():
     logger.info("POST /api/sozo/projects - Creating new project")
@@ -245,21 +255,46 @@ def generate_sozo_report(project_id):
             
             # 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)}")
+            logger.info(f"Chart URLs type: {type(chart_urls)}, count: {len(chart_urls)}")
+            logger.info(f"Chart URLs content: {chart_urls}")
+            
+            # Convert chart_urls to a simple serializable format
+            if isinstance(chart_urls, dict):
+                # Convert dict to list of values or keep as dict but ensure all values are serializable
+                serializable_chart_urls = {}
+                for key, value in chart_urls.items():
+                    # Ensure key is string and value is serializable
+                    clean_key = str(key).replace('\x00', '').replace('\ufeff', '')
+                    clean_value = str(value).replace('\x00', '').replace('\ufeff', '') if value else ''
+                    serializable_chart_urls[clean_key] = clean_value
+                chart_urls = serializable_chart_urls
+            elif isinstance(chart_urls, list):
+                # Clean list items
+                chart_urls = [str(item).replace('\x00', '').replace('\ufeff', '') for item in chart_urls if item]
             else:
                 logger.warning(f"Chart URLs is not a list/dict: {type(chart_urls)}")
                 chart_urls = []
             
-            update_data = {
+            # Try to serialize the data to JSON first to catch any issues
+            import json
+            test_data = {
                 'status': 'draft',
                 'rawMarkdown': raw_markdown,
                 'chartUrls': chart_urls
             }
             
+            # Test JSON serialization
+            json_str = json.dumps(test_data, ensure_ascii=False)
+            logger.info(f"JSON serialization test passed, length: {len(json_str)}")
+            
             logger.info(f"Updating project {project_id} with cleaned data")
-            project_ref.update(update_data)
+            project_ref.update(test_data)
             
+        except json.JSONEncodeError as json_error:
+            logger.error(f"JSON serialization error for project {project_id}: {str(json_error)}")
+            logger.error(f"Problematic data - Raw markdown preview: {raw_markdown[:100]}...")
+            logger.error(f"Problematic data - Chart URLs: {chart_urls}")
+            raise json_error
         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'}")