Update app.py

app.py

@@ -141,18 +141,17 @@ 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.
+    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.
 
-JSON Data:
+Here is the financial data:
 {json.dumps(aggregated_data, indent=2)}
 
-Instructions:
-Your response must be a complete financial report in Markdown.
+Your instructions for the report are:
 The main title of the report is "Income Statement".
 The reporting period is from {start_date.strftime('%d %B %Y')} to {end_date.strftime('%d %B %Y')}.
 The currency is South African Rand (ZAR).
 The report must contain sections for Revenue, Operating Expenses, and Net Income or Loss. Each of these sections must be a clear table.
+The Revenue section must contain a single table showing only the 'Total Revenue'. Do NOT create a table that itemizes or lists individual income sources.
 The report must also include a "Key Highlights" section with bullet points and a final "Summary" paragraph.
 Use the provided JSON data for all financial figures.
 For the Net Income or Loss table, if the net position is negative, display the amount in parentheses.
@@ -192,7 +191,7 @@ def create_pdf_report(report_text):
         
         for element in soup.find_all(True):
             if element.name in ['h1', 'h2', 'h3']:
-                level = int(element.name[1])
+                level = int(element.name)
                 font_size = {1: 16, 2: 14, 3: 12}.get(level)
                 pdf.set_font('helvetica', 'B', font_size)
                 pdf.multi_cell(0, 10, element.get_text().strip())
@@ -212,7 +211,8 @@ def create_pdf_report(report_text):
                 pdf.ln(2)
                 for li in element.find_all('li'):
                     pdf.set_font('helvetica', '', 11)
-                    pdf.multi_cell(0, 5, f"  •  {li.get_text().strip()}")
+                    item_text = li.get_text().strip().replace('•', '-')
+                    pdf.multi_cell(0, 5, f"  -  {item_text}")
                     pdf.ln(1)
                 pdf.ln(4)
             elif element.name == 'table':
@@ -233,8 +233,8 @@ def create_pdf_report(report_text):
                     if is_total_row:
                         pdf.set_font('helvetica', 'B', 10)
 
-                    pdf.cell(col_widths[0], 7, row[0], border=1)
-                    pdf.cell(col_widths[1], 7, row[1], border=1, align='R')
+                    pdf.cell(col_widths, 7, row, border=1)
+                    pdf.cell(col_widths, 7, row, border=1, align='R')
                     pdf.ln()
                     
                     if is_total_row:
@@ -243,7 +243,6 @@ def create_pdf_report(report_text):
 
         st.info("Content added to PDF. Outputting PDF to buffer...")
         
-        # --- CRITICAL FIX FOR PDF GENERATION ---
         pdf_output = pdf.output()
 
         st.success("PDF report generated successfully.")