Update app.py

app.py

@@ -1,87 +1,96 @@
 import streamlit as st
 import pandas as pd
-import numpy as np
 from docx import Document
 
 # 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.")
+    try:
+        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 = [para.text for para in doc.paragraphs if para.text.strip()]  # Remove empty lines
+            return pd.DataFrame(data, columns=["Text"])
+        else:
+            st.error("Unsupported file format. Please upload a CSV, Excel, or DOCX file.")
+            return None
+    except Exception as e:
+        st.error(f"Error loading file: {e}")
         return None
 
-# Function to generate general entries
+# Function to generate general journal entries
 def generate_general_entries(data):
-    st.subheader("General Entries")
-    st.write(data.head())  # Display the first few rows of the data
+    st.subheader("General Journal Entries")
+    if data.empty:
+        st.error("No data available for journal entries.")
+        return
+    st.write(data)
 
 # Function to generate ledger
 def generate_ledger(data):
     st.subheader("Ledger")
-    if "Account" in data.columns and "Amount" in data.columns:
+    required_columns = {"Account", "Amount"}
+    if required_columns.issubset(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.")
+        st.error(f"Missing columns for ledger: {required_columns - set(data.columns)}")
 
 # Function to generate income statement
 def generate_income_statement(data):
     st.subheader("Income Statement")
-    if "Revenue" in data.columns and "Expenses" in data.columns:
+    required_columns = {"Revenue", "Expenses"}
+    if required_columns.issubset(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}")
+        st.write(f"**Total Revenue:** ${total_revenue:,.2f}")
+        st.write(f"**Total Expenses:** ${total_expenses:,.2f}")
+        st.write(f"**Net Income:** ${net_income:,.2f}")
     else:
-        st.error("The uploaded file must contain 'Revenue' and 'Expenses' columns for income statement generation.")
+        st.error(f"Missing columns for income statement: {required_columns - set(data.columns)}")
 
 # 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:
+    required_columns = {"Cash Inflow", "Cash Outflow"}
+    if required_columns.issubset(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}")
+        st.write(f"**Total Cash Inflow:** ${total_inflow:,.2f}")
+        st.write(f"**Total Cash Outflow:** ${total_outflow:,.2f}")
+        st.write(f"**Net Cash Flow:** ${net_cash_flow:,.2f}")
     else:
-        st.error("The uploaded file must contain 'Cash Inflow' and 'Cash Outflow' columns for cash flow statement generation.")
+        st.error(f"Missing columns for cash flow statement: {required_columns - set(data.columns)}")
 
 # 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:
+    required_columns = {"Assets", "Liabilities", "Equity"}
+    if required_columns.issubset(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}")
+        st.write(f"**Total Assets:** ${total_assets:,.2f}")
+        st.write(f"**Total Liabilities:** ${total_liabilities:,.2f}")
+        st.write(f"**Total Equity:** ${total_equity:,.2f}")
     else:
-        st.error("The uploaded file must contain 'Assets', 'Liabilities', and 'Equity' columns for balance sheet generation.")
+        st.error(f"Missing columns for balance sheet: {required_columns - set(data.columns)}")
 
 # Streamlit App
-st.title("Financial Calculator 📊")
+st.title("📊 Financial Statement Generator")
 
 # File upload
 uploaded_file = st.file_uploader("Upload a CSV, Excel, or DOCX file", type=["csv", "xlsx", "docx"])
 
 if uploaded_file is not None:
     data = load_data(uploaded_file)
-    if data is not None:
-        st.write("Uploaded Data Preview:")
+    
+    if data is not None and not data.empty:
+        st.write("### Uploaded Data Preview:")
         st.write(data.head())
 
         # Generate financial statements
@@ -90,5 +99,8 @@ if uploaded_file is not None:
         generate_income_statement(data)
         generate_cash_flow(data)
         generate_balance_sheet(data)
+    else:
+        st.error("The uploaded file contains no usable data.")
+