Update app.py

app.py

@@ -98,7 +98,7 @@ def collect_basic_info():
             # Create a new document for all sections combined
             combined_doc = create_document()
 
-            for section_name, section_content in all_sections_content.items():
+            for section_name, section_content in all_sections_content: #.items():
                st.subheader(section_name)
                st.write(section_content)  # Display the generated content
 
@@ -111,6 +111,39 @@ def collect_basic_info():
             filename = f"{report_type}_{sanitized_topic}_all_sections.docx"
             st.download_button(f"Download All Sections as DOCX", combined_doc_bytes, filename)
 
+   if section_to_generate == "All Sections_0":
+        for section_name, generate_prompt_func in sections_to_process:
+            prompt = generate_prompt_func(data)
+            section_content = call_llm(prompt)
+            st.subheader(section_name)
+            st.write(section_content)
+            
+            # Update document and create download link
+            doc = add_section_to_doc(doc, section_name, section_content)
+            doc_bytes = get_docx_bytes(doc)
+
+            st.download_button(
+                label=f"Download {section_name} as DOCX",
+                data=doc_bytes,
+                file_name=f"{section_name.replace(' ', '_').lower()}.docx",
+                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+            )
+
+            # Save section content in a file
+            #filename = f"{company_name}_{section_name}.docx"
+            # Sanitize section name and limit file name length
+            sanitized_section_name = sanitize_filename(section_name, max_length=50)
+            
+            # Generate final file name and ensure it is not too long
+            filename = f"{sanitized_company_name}_{sanitized_section_name}.docx"
+            
+            with open(filename, 'wb') as f:
+                f.write(doc_bytes.getbuffer())
+
+            # Send email with section content and attachment
+            #send_email_with_attachment(email, f"{section_name} of Your Business Proposal", "Please find attached the section of your business proposal.", filename, section_content)
+
+    
     else:
         if st.button('Generate Report Section'):
             # Dynamically get the function name and call it for individual sections