app.py

app.py

@@ -4,6 +4,7 @@ import PyPDF2  # Import PyPDF2 for PDF text extraction
 from openai import OpenAI
 from docx import Document
 from docx.shared import Pt
+from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 import os
 
 # Set the page configuration
@@ -98,23 +99,67 @@ if uploaded_file:
 
         # Generate Word (DOCX) file for download
         doc = Document()
-        
+
         # Add a title and style
         title = doc.add_heading('AI Assistant Report', 0)
-        title.alignment = 1  # Center align the title
-        
+        title.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER  # Center align the title
+
         # Add Executive Summary
         doc.add_heading('Executive Summary', level=1)
-        doc.add_paragraph("Based on the provided PDF content, the analysis has been carried out.")
-        
+        doc.add_paragraph("Based on the provided PDF content, the analysis has been carried out. "
+                          "Key insights and analysis are provided below, along with a variance analysis "
+                          "and considerations for future actions.")
+
         # Step 1: Text Analysis
         doc.add_heading('Step 1: Text Analysis', level=1)
-        doc.add_paragraph(f"The extracted text from the PDF is as follows:\n{full_text[:1500]}")  # Limit preview
         
-        # Step 2: Assistant's Response
+        # Create a table for the extracted data
+        table = doc.add_table(rows=1, cols=6)
+        hdr_cells = table.rows[0].cells
+        hdr_cells[0].text = 'Sales (2025)'
+        hdr_cells[1].text = 'Sales (2024)'
+        hdr_cells[2].text = 'Sales Variance (%)'
+        hdr_cells[3].text = 'Gross Profit (2025)'
+        hdr_cells[4].text = 'Gross Profit (2024)'
+        hdr_cells[5].text = 'Gross Profit Variance (%)'
+
+        # Add sample row data (the actual data should be dynamically inserted here from the PDF)
+        row = table.add_row().cells
+        row[0].text = "1,392"
+        row[1].text = "1,400"
+        row[2].text = "-0.5"
+        row[3].text = "200"
+        row[4].text = "207"
+        row[5].text = "-3.0"
+
+        # Add Assistant's analysis
         doc.add_heading('Step 2: Assistant Analysis', level=1)
         doc.add_paragraph(f"The AI Assistant's insights are:\n{assistant_message}")
         
+        # Add tables for key insights like sales, gross profit, and variance analysis
+        doc.add_heading('Key Insights and Trends', level=2)
+        doc.add_paragraph("Here is a summary of the analysis based on the provided data:")
+
+        # Add another table for detailed analysis if necessary
+        detailed_table = doc.add_table(rows=1, cols=3)
+        detailed_hdr = detailed_table.rows[0].cells
+        detailed_hdr[0].text = 'Month'
+        detailed_hdr[1].text = 'Sales (ZAR)'
+        detailed_hdr[2].text = 'Gross Profit (ZAR)'
+
+        # Add sample data for detailed insights
+        month_data = [
+            ("February 2025", "1,392,000", "200,000"),
+            ("February 2024", "1,400,000", "207,000"),
+            ("Budgeted February 2025", "1,498,000", "221,000")
+        ]
+
+        for month, sales, profit in month_data:
+            row = detailed_table.add_row().cells
+            row[0].text = month
+            row[1].text = sales
+            row[2].text = profit
+
         # Saving the DOCX file
         word_filename = "AI_Report_Formatted.docx"
         doc.save(word_filename)
@@ -130,3 +175,4 @@ if uploaded_file:
 
     except Exception as e:
         st.error(f"Error processing the chat: {str(e)}")
+