Update app.py

app.py

@@ -3,6 +3,7 @@
 import streamlit as st
 import pandas as pd
 from docx import Document
+from io import BytesIO
 
 def process_financial_data(file):
     try:
@@ -16,23 +17,53 @@ def process_financial_data(file):
             df = pd.DataFrame({'Content': data})
         else:
             st.error("Unsupported file format. Please upload a CSV, Excel, or Word document.")
-            return None, None, None, None, None
+            return None, None, None, None, None, None
         
         # Creating Ledger
         ledger = df[df['Type'] == 'Ledger'] if 'Type' in df.columns else None
         
-        # Creating Journal
+        # 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
         cash_flow = df[df['Type'] == 'Cash Flow Statement'] if 'Type' in df.columns else None
         
-        return ledger, journal, income_statement, balance_sheet, cash_flow
+        return ledger, journal, trial_balance, income_statement, balance_sheet, cash_flow
     except Exception as e:
         st.error(f"Error processing file: {e}")
-        return None, None, None, None, None
+        return None, None, None, None, None, None
+
+def generate_word_report(ledger, journal, trial_balance, income_stmt, balance_sht, cash_flw):
+    doc = Document()
+    doc.add_heading("Financial Report", level=1)
+    
+    def add_table(df, title):
+        if df is not None:
+            doc.add_heading(title, level=2)
+            table = doc.add_table(rows=1, cols=len(df.columns))
+            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")
+    
+    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)
+    return buffer
 
 st.title("Comprehensive Financial Accounting App")
 st.write("Upload a financial data file (CSV, Excel, Word) to generate Ledger, Journal, and Financial Statements.")
@@ -40,16 +71,20 @@ st.write("Upload a financial data file (CSV, Excel, Word) to generate Ledger, Jo
 uploaded_file = st.file_uploader("Upload a file", type=["csv", "xls", "xlsx", "docx"])
 
 if uploaded_file is not None:
-    ledger, journal, income_stmt, balance_sht, cash_flw = process_financial_data(uploaded_file)
+    ledger, journal, trial_balance, income_stmt, balance_sht, cash_flw = process_financial_data(uploaded_file)
     
     if ledger is not None:
         st.subheader("Ledger")
         st.dataframe(ledger)
     
     if journal is not None:
-        st.subheader("Journal")
+        st.subheader("Journal Entries")
         st.dataframe(journal)
     
+    if trial_balance is not None:
+        st.subheader("Trial Balance")
+        st.dataframe(trial_balance)
+    
     if income_stmt is not None:
         st.subheader("Income Statement")
         st.dataframe(income_stmt)
@@ -61,6 +96,10 @@ if uploaded_file is not None:
     if cash_flw is not None:
         st.subheader("Cash Flow Statement")
         st.dataframe(cash_flw)
+    
+    if st.button("Download Financial Report as Word Document"):
+        doc_file = generate_word_report(ledger, journal, trial_balance, income_stmt, balance_sht, cash_flw)
+        st.download_button(label="Download Report", data=doc_file, file_name="Financial_Report.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
 
 st.write("Developed for comprehensive financial analysis.")