Update app.py

app.py

@@ -22,6 +22,10 @@ from fpdf import FPDF
 from dotenv import load_dotenv
 from retrying import retry
 
+
+
+
+
 # Declare the exa search API
 exa = Exa(api_key=os.getenv("EXA_API_KEY"))
 
@@ -66,7 +70,6 @@ def get_docx_bytes(doc):
 
 # Generic function to dynamically call report generation functions
 def call_dynamic_function(function_name, data):
-    # Use globals to call the function by name
     function = globals()[function_name]
     return function(data)
 
@@ -83,28 +86,50 @@ def collect_basic_info():
 
     # Create data dictionary by dynamically generating input fields
     data = {field_key: st.text_area(field_label) for field_key, field_label in report_fields}
-
-    # Select section to generate
+    
+    # Add an option to generate all sections
     section_to_generate = st.selectbox("Choose section to generate", [section[0] for section in report_sections])
 
-    if st.button('Generate Report Section'):
-        # Dynamically get the function name and call it
-        section_function_name = dict(report_sections)[section_to_generate]
-        update_session_tracker(section_to_generate)
+    if section_to_generate == "All Sections":
+        if st.button('Generate All Sections'):
+            # Generate all sections content at once
+            all_sections_content = generate_all_sections(data)
+
+            # Create a new document for all sections combined
+            combined_doc = create_document()
+
+            for section_name, section_content in all_sections_content.items():
+                st.subheader(section_name)
+                st.write(section_content)
+
+                # Add each section's content to the combined document
+                combined_doc = add_section_to_doc(combined_doc, section_name, section_content)
+
+            # Get bytes for the combined document and create a download button
+            combined_doc_bytes = get_docx_bytes(combined_doc)
+            sanitized_topic = sanitize_filename(data["research_topic"])
+            filename = f"{report_type}_{sanitized_topic}_all_sections.docx"
+            st.download_button(f"Download All Sections as DOCX", combined_doc_bytes, filename)
+
+    else:
+        if st.button('Generate Report Section'):
+            # Dynamically get the function name and call it for individual sections
+            section_function_name = dict(report_sections)[section_to_generate]
+            update_session_tracker(section_to_generate)
 
-        # Generate content and display it
-        section_content = call_dynamic_function(section_function_name, data)
-        st.subheader(section_to_generate)
-        st.write(section_content)
+            # Generate content and display it
+            section_content = call_dynamic_function(section_function_name, data)
+            st.subheader(section_to_generate)
+            st.write(section_content)
 
-        # Create and download the document
-        doc = create_document()
-        doc = add_section_to_doc(doc, section_to_generate, section_content)
-        doc_bytes = get_docx_bytes(doc)
+            # Create and download the document for this specific section only
+            doc = create_document()
+            doc = add_section_to_doc(doc, section_to_generate, section_content)
+            doc_bytes = get_docx_bytes(doc)
 
-        sanitized_topic = sanitize_filename(data["research_topic"])
-        filename = f"{report_type}_{sanitized_topic}.docx"
-        st.download_button(f"Download {section_to_generate} as DOCX", doc_bytes, filename)
+            sanitized_topic = sanitize_filename(data["research_topic"])
+            filename = f"{report_type}_{sanitized_topic}.docx"
+            st.download_button(f"Download {section_to_generate} as DOCX", doc_bytes, filename)
 
 # Run the main function
-collect_basic_info()
+collect_basic_info()