Update app.py

app.py

@@ -1,5 +1,3 @@
-# app.py
-
 import streamlit as st
 import pandas as pd
 from docx import Document
@@ -19,13 +17,9 @@ def process_financial_data(file):
             st.error("Unsupported file format. Please upload a CSV, Excel, or Word document.")
             return None, None, None, None, None, None
         
-        # Creating Ledger
+        # Creating Financial Reports
         ledger = df[df['Type'] == 'Ledger'] if 'Type' in df.columns else None
-        
-        # Creating Journal Entries
         journal = df[df['Type'] == 'Journal'] if 'Type' in df.columns else None
-        
-        # Creating Financial Statements
         trial_balance = df[df['Type'] == 'Trial Balance'] if 'Type' in df.columns else None
         income_statement = df[df['Type'] == 'Income Statement'] if 'Type' in df.columns else None
         balance_sheet = df[df['Type'] == 'Balance Sheet'] if 'Type' in df.columns else None
@@ -39,27 +33,36 @@ def process_financial_data(file):
 def generate_word_report(ledger, journal, trial_balance, income_stmt, balance_sht, cash_flw):
     doc = Document()
     doc.add_heading("Financial Report", level=1)
-    
+    doc.add_paragraph("\nGenerated Financial Statements\n")
+
     def add_table(df, title):
-        if df is not None:
+        if df is not None and not df.empty:
             doc.add_heading(title, level=2)
             table = doc.add_table(rows=1, cols=len(df.columns))
+            table.style = 'Table Grid'
+            
             hdr_cells = table.rows[0].cells
             for i, col_name in enumerate(df.columns):
                 hdr_cells[i].text = col_name
+
             for _, row in df.iterrows():
                 row_cells = table.add_row().cells
                 for i, val in enumerate(row):
                     row_cells[i].text = str(val)
+
             doc.add_paragraph("\n")
-    
+        else:
+            doc.add_heading(title, level=2)
+            doc.add_paragraph("No data available.\n")
+
+    # Add each financial statement to the document
     add_table(ledger, "Ledger")
     add_table(journal, "Journal Entries")
     add_table(trial_balance, "Trial Balance")
     add_table(income_stmt, "Income Statement")
     add_table(balance_sht, "Balance Sheet")
     add_table(cash_flw, "Cash Flow Statement")
-    
+
     buffer = BytesIO()
     doc.save(buffer)
     buffer.seek(0)
@@ -103,3 +106,4 @@ if uploaded_file is not None:
 
 st.write("Developed for comprehensive financial analysis.")
 
+