SOC2

app/DocumentGeneration/pdf_generator.py

@@ -44,12 +44,19 @@ def create_pdf_document(request: ExportRequest):
         raise Exception("fpdf2 is not installed. Please install it with: pip install fpdf2")
     
     # Filter out empty sections to prevent blank pages
-    filtered_sections = filter_empty_sections(request.sections)
+    try:
+        filtered_sections = filter_empty_sections(request.sections)
+    except Exception as e:
+        raise Exception(f"Failed to filter sections: {str(e)}")
     if not filtered_sections:
         raise Exception("No valid sections found after filtering. All sections appear to be empty.")
     
     # Generate table of contents
-    toc_items = generate_table_of_contents(filtered_sections, request.businessIdea)
+    try:
+        toc_items = generate_table_of_contents(filtered_sections, request.businessIdea)
+    except Exception as e:
+        toc_items = []
+        # Continue with a fallback; TOC page builder already has fallback handling
     # logging.info(f"Generated TOC with {len(toc_items)} items: {[item['title'] for item in toc_items[:3]]}...")
     
     # Create PDF with professional settings
@@ -254,8 +261,15 @@ def create_pdf_document(request: ExportRequest):
             pdf.set_xy(25, pdf.get_y())  # Use current Y position for consistent spacing
             
             # Process content
-            content_text = clean_text_for_pdf(section.content)
-            content_text = convert_currency_symbols_to_iso(content_text)
+            try:
+                content_text = clean_text_for_pdf(section.content)
+            except Exception as e:
+                content_text = section.content or ""
+            try:
+                content_text = convert_currency_symbols_to_iso(content_text)
+            except Exception:
+                # If currency conversion fails, keep original text
+                pass
             
             # Remove chart data markers
             content_text = re.sub(r'<!--CHARTDATASTART-->.*?<!--CHARTDATAEND-->', '', content_text, flags=re.DOTALL)
@@ -408,7 +422,8 @@ def create_pdf_document(request: ExportRequest):
 
                         charts_added += 1
                         total_charts_placed += 1
-                    except Exception as chart_error:
+                    except Exception:
+                        # Skip faulty chart and continue rendering others
                         continue
             else:
                 # Debug: Check if there are any charts that might match this section

app/DocumentGeneration/word_generator.py

@@ -41,7 +41,10 @@ def create_word_document(request: ExportRequest):
         raise Exception("python-docx is not installed. Please install it with: pip install python-docx")
     
     # Filter out empty sections to prevent blank pages
-    filtered_sections = filter_empty_sections(request.sections)
+    try:
+        filtered_sections = filter_empty_sections(request.sections)
+    except Exception as e:
+        raise Exception(f"Failed to filter sections: {str(e)}")
     if not filtered_sections:
         raise Exception("No valid sections found after filtering. All sections appear to be empty.")
     
@@ -138,7 +141,11 @@ def create_word_document(request: ExportRequest):
     doc.add_paragraph()
     
     # Sort sections according to predefined order (do this before TOC generation)
-    sorted_sections = sort_sections_by_order(filtered_sections)
+    try:
+        sorted_sections = sort_sections_by_order(filtered_sections)
+    except Exception as e:
+        # Fallback: use filtered order if sorting fails
+        sorted_sections = filtered_sections
     
     # Generate TOC entries with actual page numbers
     # Since each section starts on a new page, we can calculate page numbers
@@ -205,8 +212,14 @@ def create_word_document(request: ExportRequest):
             doc.add_heading(section.title, level=1)
             
             # Clean and process content
-            clean_content = clean_text_for_pdf(section.content)
-            clean_content = convert_currency_symbols_to_iso(clean_content)
+            try:
+                clean_content = clean_text_for_pdf(section.content)
+            except Exception:
+                clean_content = section.content or ""
+            try:
+                clean_content = convert_currency_symbols_to_iso(clean_content)
+            except Exception:
+                pass
             
             # Remove chart data markers
             clean_content = re.sub(r'<!--CHARTDATASTART-->.*?<!--CHARTDATAEND-->', '', clean_content, flags=re.DOTALL)

app/main.py

@@ -1457,36 +1457,32 @@ async def export_business_plan(request: ExportRequest):
     FLOW: Server generates document → Returns base64 data → Frontend uploads to UploadThing → User gets download link
     """
     try:
-
-        
-
-        # Generate unique request ID for tracking
-        request_id = str(id(request))
-        
-        # Add request fingerprint for duplicate detection
-        request_fingerprint = f"{request.businessIdea}_{request.format}_{len(request.sections)}_{hash(str(request.sections))}"
-        
-        # Check if this is a duplicate request (within 5 seconds)
-        current_time = time.time()
-        if hasattr(export_business_plan, 'last_requests'):
-            # Clean old requests (older than 5 seconds)
-            export_business_plan.last_requests = {
-                fp: timestamp for fp, timestamp in export_business_plan.last_requests.items()
-                if current_time - timestamp < 5
-            }
-            
-            # Check for duplicate
-            if request_fingerprint in export_business_plan.last_requests:
-                raise HTTPException(
-                    status_code=429, 
-                    detail="Duplicate export request detected. Please wait a few seconds before trying again."
-                )
-        else:
-            export_business_plan.last_requests = {}
-        
-        # Record this request
-        export_business_plan.last_requests[request_fingerprint] = current_time
-        
+        # Duplicate detection (best-effort; non-fatal if bookkeeping fails)
+        try:
+            request_id = str(id(request))
+            request_fingerprint = f"{request.businessIdea}_{request.format}_{len(request.sections)}_{hash(str(request.sections))}"
+            current_time = time.time()
+            if hasattr(export_business_plan, 'last_requests'):
+                # Clean old requests (older than 5 seconds)
+                export_business_plan.last_requests = {
+                    fp: timestamp for fp, timestamp in export_business_plan.last_requests.items()
+                    if current_time - timestamp < 5
+                }
+                # Check for duplicate
+                if request_fingerprint in export_business_plan.last_requests:
+                    raise HTTPException(
+                        status_code=429,
+                        detail="Duplicate export request detected. Please wait a few seconds before trying again."
+                    )
+            else:
+                export_business_plan.last_requests = {}
+            # Record this request
+            export_business_plan.last_requests[request_fingerprint] = current_time
+        except HTTPException:
+            raise
+        except Exception:
+            # Do not block export on dedupe bookkeeping errors
+            pass
 
         
         # Validate request format
@@ -1503,10 +1499,13 @@ async def export_business_plan(request: ExportRequest):
         if len(request.sections) == 0:
             raise HTTPException(status_code=400, detail="At least one section is required for export.")
         
-        # Check section content
+        # Check section content and shape
         for i, section in enumerate(request.sections):
+            title = getattr(section, 'title', None)
+            if not title or not str(title).strip():
+                raise HTTPException(status_code=400, detail=f"Section at index {i} is missing a title.")
             if not section.content or not section.content.strip():
-                raise HTTPException(status_code=400, detail=f"Section '{section.title}' has no content.")
+                raise HTTPException(status_code=400, detail=f"Section '{title}' has no content.")
         
         # logging.info(f"Exporting business plan for: {request.businessIdea} in {request.format.upper()} format")
         # logging.info(f"Request has {len(request.sections)} sections and summary length: {len(request.summary)}")
@@ -1539,6 +1538,11 @@ async def export_business_plan(request: ExportRequest):
                 doc_bytes, filename = create_pdf_document(request)
             else:  # word
                 doc_bytes, filename = create_word_document(request)
+            # Basic sanity check on output
+            if not doc_bytes or (isinstance(doc_bytes, (bytes, bytearray)) and len(doc_bytes) < 1000):
+                raise HTTPException(status_code=500, detail=f"Generated {request.format.upper()} appears empty or invalid.")
+        except HTTPException:
+            raise
         except ImportError as import_error:
             raise HTTPException(
                 status_code=500,