Update app.py

app.py

@@ -16,32 +16,30 @@ def load_data(file):
         st.error("Unsupported file format. Please upload a CSV, Excel, or DOCX file.")
         return None
 
-# Function to generate Journal Entries from raw data
+# Function to generate unique journal entries from raw data
 def generate_journal_entries(data):
     st.subheader("Journal Entries")
 
-    journal_entries = []
-    
+    journal_entries = set()  # Using a set to avoid duplicate entries
+
     for index, row in data.iterrows():
         desc = row["Description"].lower()
-        
+
         if "capital" in desc:
-            journal_entries.append(["Cash", "Capital", 10000])
-        elif "inventory" in desc and "credit" in desc:
-            journal_entries.append(["Inventory", "Accounts Payable", 4000])
-            journal_entries.append(["Inventory", "Cash", 3000])
+            journal_entries.add(("Cash", "Capital", 10000))
+        elif "inventory purchase" in desc:
+            journal_entries.add(("Inventory", "Accounts Payable", 7000))
         elif "equipment" in desc:
-            journal_entries.append(["Office Equipment", "Bank Loan", 8000])
-            journal_entries.append(["Office Equipment", "Cash", 7000])
+            journal_entries.add(("Office Equipment", "Bank Loan", 8000))
         elif "sales" in desc:
-            journal_entries.append(["Accounts Receivable", "Sales Revenue", 5000])
-            journal_entries.append(["Cash", "Sales Revenue", 15000])
+            journal_entries.add(("Accounts Receivable", "Sales Revenue", 5000))
+            journal_entries.add(("Cash", "Sales Revenue", 15000))
         elif "salary" in desc:
-            journal_entries.append(["Salaries Expense", "Cash", 3000])
+            journal_entries.add(("Salaries Expense", "Cash", 3000))
         elif "rent" in desc:
-            journal_entries.append(["Rent Expense", "Cash", 2000])
+            journal_entries.add(("Rent Expense", "Cash", 2000))
         elif "utilities" in desc:
-            journal_entries.append(["Utilities Expense", "Cash", 1000])
+            journal_entries.add(("Utilities Expense", "Cash", 1000))
 
     journal_df = pd.DataFrame(journal_entries, columns=["Debit", "Credit", "Amount"])
     st.write(journal_df)
@@ -72,16 +70,10 @@ def generate_ledger(journal_df):
 # Function to generate Income Statement
 def generate_income_statement(journal_df):
     st.subheader("Income Statement")
-    revenue = 0
-    expenses = 0
-
-    for index, row in journal_df.iterrows():
-        if "Revenue" in row["Credit"]:
-            revenue += row["Amount"]
-        if "Expense" in row["Debit"]:
-            expenses += row["Amount"]
-
+    revenue = sum(row["Amount"] for index, row in journal_df.iterrows() if "Revenue" in row["Credit"])
+    expenses = sum(row["Amount"] for index, row in journal_df.iterrows() if "Expense" in row["Debit"])
     net_income = revenue - expenses
+
     income_statement = pd.DataFrame({
         "Category": ["Total Revenue", "Total Expenses", "Net Income"],
         "Amount": [revenue, expenses, net_income]
@@ -93,16 +85,10 @@ def generate_income_statement(journal_df):
 # Function to generate Cash Flow Statement
 def generate_cash_flow(journal_df):
     st.subheader("Cash Flow Statement")
-    cash_inflow = 0
-    cash_outflow = 0
-
-    for index, row in journal_df.iterrows():
-        if row["Debit"] == "Cash":
-            cash_inflow += row["Amount"]
-        elif row["Credit"] == "Cash":
-            cash_outflow += row["Amount"]
-
+    cash_inflow = sum(row["Amount"] for index, row in journal_df.iterrows() if row["Debit"] == "Cash")
+    cash_outflow = sum(row["Amount"] for index, row in journal_df.iterrows() if row["Credit"] == "Cash")
     net_cash_flow = cash_inflow - cash_outflow
+
     cash_flow_statement = pd.DataFrame({
         "Category": ["Total Cash Inflow", "Total Cash Outflow", "Net Cash Flow"],
         "Amount": [cash_inflow, cash_outflow, net_cash_flow]
@@ -115,23 +101,12 @@ def generate_cash_flow(journal_df):
 def generate_balance_sheet(ledger_df, net_income):
     st.subheader("Balance Sheet")
     
-    assets = 0
-    liabilities = 0
-    equity = net_income  # Retained earnings
-
-    for index, row in ledger_df.iterrows():
-        account_name = row["Account"]
-        balance = row["Balance"]
-
-        if any(keyword in account_name.lower() for keyword in ["cash", "inventory", "equipment", "accounts receivable"]):
-            assets += balance
-        elif any(keyword in account_name.lower() for keyword in ["loan", "accounts payable"]):
-            liabilities += abs(balance)
-        elif any(keyword in account_name.lower() for keyword in ["capital", "equity"]):
-            equity += balance
-
-    # Ensuring the balance equation holds: Assets = Liabilities + Equity
-    if assets != (liabilities + equity):
+    assets = sum(row["Balance"] for index, row in ledger_df.iterrows() if row["Account"] in ["Cash", "Inventory", "Office Equipment", "Accounts Receivable"])
+    liabilities = sum(abs(row["Balance"]) for index, row in ledger_df.iterrows() if row["Account"] in ["Bank Loan", "Accounts Payable"])
+    equity = sum(row["Balance"] for index, row in ledger_df.iterrows() if "Capital" in row["Account"]) + net_income
+
+    # Ensuring Assets = Liabilities + Equity
+    if assets != liabilities + equity:
         difference = assets - (liabilities + equity)
         if difference > 0:
             liabilities += difference