Better return statements

controllers/log_controller.py

@@ -404,7 +404,7 @@ def classify_log_activities(current_user):
              logger.error("Failed to save initial Log object.")
              return jsonify({'message': 'Failed to create log entry'}), 500
         log_id = new_log._id
-        logger.info(f"Log object created successfully with ID: {log_id}")
+        logger.info(f"Log object created and saved successfully with ID: {log_id}")
         
         logger.info(f"Extracting activities with LLM for log {log_id}...")
         activities_json = extract_activities(extracted_text)
@@ -509,12 +509,22 @@ def classify_log_activities(current_user):
             else:
                  logger.error(f"Could not find Log {log_id} to update with incident IDs.")
 
+        # --- Log right before return --- 
+        try:
+            log_dict_to_return = new_log.to_dict()
+            logger.info(f"Preparing to return Log object: {log_dict_to_return}")
+        except Exception as to_dict_err:
+             logger.error(f"Error calling new_log.to_dict() before return: {to_dict_err}")
+             # Optionally return an error here if to_dict fails
+             return jsonify({'message': 'Internal error preparing log data for response.'}), 500
+        # --- End Log right before return --- 
+
         logger.info("Successfully processed upload & classification request. Returning 200 OK.")
         return jsonify({
             'message': 'Log created, activities extracted and classified, incidents created.',
-            'log': new_log.to_dict(), # Return the created log object details
-            'classified_activities': classified_activities_output, # Activities with classification & incident IDs
-            'extracted_text': extracted_text # Keep sending this for context if needed
+            'log': log_dict_to_return, # Use the logged dictionary
+            'classified_activities': classified_activities_output, 
+            'extracted_text': extracted_text
         }), 200
         
     except Exception as e: