Update app.py

app.py

@@ -1,109 +1,94 @@
 import streamlit as st
 import pandas as pd
+import numpy as np
 from docx import Document
-from io import BytesIO
 
-def process_financial_data(file):
-    try:
-        if file.name.endswith('.csv'):
-            df = pd.read_csv(file)
-        elif file.name.endswith(('.xls', '.xlsx')):
-            df = pd.read_excel(file)
-        elif file.name.endswith('.docx'):
-            doc = Document(file)
-            data = [para.text for para in doc.paragraphs if para.text.strip()]
-            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, None
-        
-        # Creating Financial Reports
-        ledger = df[df['Type'] == 'Ledger'] if 'Type' in df.columns else None
-        journal = df[df['Type'] == 'Journal'] if 'Type' in df.columns else None
-        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, 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, None
+# Function to load data from CSV, Excel, or DOCX
+def load_data(file):
+    if file.type == "text/csv":
+        return pd.read_csv(file)
+    elif file.type == "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet":
+        return pd.read_excel(file)
+    elif file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
+        doc = Document(file)
+        data = []
+        for para in doc.paragraphs:
+            data.append(para.text)
+        return pd.DataFrame(data, columns=["Text"])
+    else:
+        st.error("Unsupported file format. Please upload a CSV, Excel, or DOCX file.")
+        return None
 
-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")
+# Function to generate general entries
+def generate_general_entries(data):
+    st.subheader("General Entries")
+    st.write(data.head())  # Display the first few rows of the data
 
-    def add_table(df, title):
-        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
+# Function to generate ledger
+def generate_ledger(data):
+    st.subheader("Ledger")
+    if "Account" in data.columns and "Amount" in data.columns:
+        ledger = data.groupby("Account")["Amount"].sum().reset_index()
+        st.write(ledger)
+    else:
+        st.error("The uploaded file must contain 'Account' and 'Amount' columns for ledger generation.")
 
-            for _, row in df.iterrows():
-                row_cells = table.add_row().cells
-                for i, val in enumerate(row):
-                    row_cells[i].text = str(val)
+# Function to generate income statement
+def generate_income_statement(data):
+    st.subheader("Income Statement")
+    if "Revenue" in data.columns and "Expenses" in data.columns:
+        total_revenue = data["Revenue"].sum()
+        total_expenses = data["Expenses"].sum()
+        net_income = total_revenue - total_expenses
+        st.write(f"Total Revenue: {total_revenue}")
+        st.write(f"Total Expenses: {total_expenses}")
+        st.write(f"Net Income: {net_income}")
+    else:
+        st.error("The uploaded file must contain 'Revenue' and 'Expenses' columns for income statement generation.")
 
-            doc.add_paragraph("\n")
-        else:
-            doc.add_heading(title, level=2)
-            doc.add_paragraph("No data available.\n")
+# Function to generate cash flow statement
+def generate_cash_flow(data):
+    st.subheader("Cash Flow Statement")
+    if "Cash Inflow" in data.columns and "Cash Outflow" in data.columns:
+        total_inflow = data["Cash Inflow"].sum()
+        total_outflow = data["Cash Outflow"].sum()
+        net_cash_flow = total_inflow - total_outflow
+        st.write(f"Total Cash Inflow: {total_inflow}")
+        st.write(f"Total Cash Outflow: {total_outflow}")
+        st.write(f"Net Cash Flow: {net_cash_flow}")
+    else:
+        st.error("The uploaded file must contain 'Cash Inflow' and 'Cash Outflow' columns for cash flow statement generation.")
 
-    # 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")
+# Function to generate balance sheet
+def generate_balance_sheet(data):
+    st.subheader("Balance Sheet")
+    if "Assets" in data.columns and "Liabilities" in data.columns and "Equity" in data.columns:
+        total_assets = data["Assets"].sum()
+        total_liabilities = data["Liabilities"].sum()
+        total_equity = data["Equity"].sum()
+        st.write(f"Total Assets: {total_assets}")
+        st.write(f"Total Liabilities: {total_liabilities}")
+        st.write(f"Total Equity: {total_equity}")
+    else:
+        st.error("The uploaded file must contain 'Assets', 'Liabilities', and 'Equity' columns for balance sheet generation.")
 
-    buffer = BytesIO()
-    doc.save(buffer)
-    buffer.seek(0)
-    return buffer
+# Streamlit App
+st.title("Financial Calculator 📊")
 
-st.title("Comprehensive Financial Accounting App")
-st.write("Upload a financial data file (CSV, Excel, Word) to generate Ledger, Journal, and Financial Statements.")
-
-uploaded_file = st.file_uploader("Upload a file", type=["csv", "xls", "xlsx", "docx"])
+# File upload
+uploaded_file = st.file_uploader("Upload a CSV, Excel, or DOCX file", type=["csv", "xlsx", "docx"])
 
 if uploaded_file is not None:
-    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 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)
-    
-    if balance_sht is not None:
-        st.subheader("Balance Sheet")
-        st.dataframe(balance_sht)
-    
-    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")
+    data = load_data(uploaded_file)
+    if data is not None:
+        st.write("Uploaded Data Preview:")
+        st.write(data.head())
 
-st.write("Developed for comprehensive financial analysis.")
+        # Generate financial statements
+        generate_general_entries(data)
+        generate_ledger(data)
+        generate_income_statement(data)
+        generate_cash_flow(data)
+        generate_balance_sheet(data)