Update functions.py

functions.py

@@ -74,21 +74,21 @@ def get_report_sections(report_type):
 
 def generate_all_sections(data):
     """
-    Generate all sections for the specified report type and display them on Streamlit.
+    Generate all sections for the specified report type.
     """
+    sections_content = {}
+    
     # Retrieve sections based on report type
     report_sections = get_report_sections(data.get("report_type"))
 
-    # Loop through each section and generate content
-    for section_name, function_name in report_sections:
         # Call the respective function dynamically
         section_content = call_dynamic_function(function_name, data)
         
-        # Display the generated content on Streamlit
-        st.subheader(section_name)  # Display section title
-        st.write(section_content)    # Display section content
+        # Store the generated content in a dictionary
+        sections_content[section_name] = section_content
+    
+    return sections_content
 
-    # No need to return anything; just display results
 
     
 # Define the actual generation functions, like: