Update functions.py

functions.py

@@ -109,27 +109,34 @@ def generate_all_sections_0(data):
     return sections_content  # Return the content dictionary
     
 
-def generate_report_section(section_to_generate, data):
+def generate_all_sections(data):
     """
-    Generate content for a selected report section.
+    Generate all sections for the specified report type and update the session tracker.
     """
-    # Get report sections based on the report type
+    sections_content = {}
+    
+    # Get the report type from data
     report_type = data.get("report_type")
+    
+    # Retrieve sections based on report type
     report_sections = get_report_sections(report_type)
+    
+    # Loop through each section and generate content
+    for section_name, function_name in report_sections:
+        if function_name == "generate_all_sections":
+            continue  # Skip "All Sections" itself
 
-    # Dynamically call the function based on the selected section
-    section_function_name = dict(report_sections).get(section_to_generate)
-
-    # Update session tracker to log the generated section (optional if session tracking is needed)
-    update_session_tracker(section_to_generate)
-
-    # Call the corresponding section generation function
-    if section_function_name:
+        # Update the session tracker for each section
+        update_session_tracker(section_name)
+        
+        # Dynamically call the function to generate the content
+        section_function_name = dict(report_sections)[section_name]
         section_content = call_dynamic_function(section_function_name, data)
-        return section_content
-    else:
-        return "Error: Section function not found."
-
+        
+        # Store the generated content
+        sections_content[section_name] = section_content
+    
+    return sections_content
 
 
 # Define the actual generation functions, like: