Update app.py

app.py

@@ -140,7 +140,6 @@ def generate_financial_report(model, aggregated_data, start_date, end_date, stat
     """
     st.info(f"Preparing to generate {statement_type} with pre-aggregated data...")
 
-    # This is the final, simplified, high-level prompt with no special characters.
     prompt = f"""You are an expert financial analyst. Your task is to generate a professional Income Statement in Markdown format using the pre-aggregated JSON data provided below.
 
 Here is the financial data:
@@ -171,8 +170,8 @@ Separate the major sections with a horizontal rule.
 
 def create_pdf_report(report_text):
     """
-    Creates a PDF from markdown text. Includes the critical fix for the
-    'bytearray' object has no attribute 'encode' error.
+    Creates a PDF from markdown text. This version includes fixes for both the
+    ValueError from incorrect int conversion and the table rendering logic.
     """
     if not report_text:
         st.warning("Report text is empty, skipping PDF generation.")
@@ -191,8 +190,9 @@ def create_pdf_report(report_text):
         
         for element in soup.find_all(True):
             if element.name in ['h1', 'h2', 'h3']:
-                level = int(element.name)
-                font_size = {1: 16, 2: 14, 3: 12}.get(level)
+                # --- BUG FIX 1: Correctly extract the number from the tag ---
+                level = int(element.name[1])
+                font_size = {1: 16, 2: 14, 3: 12}.get(level, 11)
                 pdf.set_font('helvetica', 'B', font_size)
                 pdf.multi_cell(0, 10, element.get_text().strip())
                 pdf.ln(level * 2)
@@ -233,9 +233,11 @@ def create_pdf_report(report_text):
                     if is_total_row:
                         pdf.set_font('helvetica', 'B', 10)
 
-                    pdf.cell(col_widths, 7, row, border=1)
-                    pdf.cell(col_widths, 7, row, border=1, align='R')
-                    pdf.ln()
+                    # --- BUG FIX 2: Correctly iterate through cells and widths ---
+                    if len(row) == len(col_widths): # Ensure row has the correct number of cells
+                        pdf.cell(col_widths[0], 7, row[0], border=1)
+                        pdf.cell(col_widths[1], 7, row[1], border=1, align='R')
+                        pdf.ln()
                     
                     if is_total_row:
                         pdf.set_font('helvetica', '', 10)