Update app.py

app.py

@@ -2,36 +2,53 @@
 
 import streamlit as st
 import pandas as pd
+import docx
 
 def process_financial_data(file):
     try:
-        df = pd.read_csv(file)
+        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 = docx.Document(file)
+            data = []
+            for para in doc.paragraphs:
+                data.append(para.text)
+            df = pd.DataFrame({'Content': data})
+        else:
+            st.error("Unsupported file format. Please upload a CSV, Excel, or Word document.")
+            return None, None, None
         
-        income_statement = df[df['Type'] == 'Income Statement']
-        balance_sheet = df[df['Type'] == 'Balance Sheet']
-        cash_flow = df[df['Type'] == 'Cash Flow Statement']
-        
-        return income_statement, balance_sheet, cash_flow
+        if 'Type' in df.columns:
+            income_statement = df[df['Type'] == 'Income Statement']
+            balance_sheet = df[df['Type'] == 'Balance Sheet']
+            cash_flow = df[df['Type'] == 'Cash Flow Statement']
+            return income_statement, balance_sheet, cash_flow
+        else:
+            return df, None, None
     except Exception as e:
         st.error(f"Error processing file: {e}")
         return None, None, None
 
 st.title("Financial Accounting App")
-st.write("Upload a financial data file (CSV) to generate statements.")
+st.write("Upload a financial data file (CSV, Excel, Word) to generate statements.")
 
-uploaded_file = st.file_uploader("Upload CSV file", type=["csv"])
+uploaded_file = st.file_uploader("Upload a file", type=["csv", "xls", "xlsx", "docx"])
 
 if uploaded_file is not None:
     income_stmt, balance_sht, cash_flw = process_financial_data(uploaded_file)
     
     if income_stmt is not None:
-        st.subheader("Income Statement")
+        st.subheader("Processed Data")
         st.dataframe(income_stmt)
 
-        st.subheader("Balance Sheet")
-        st.dataframe(balance_sht)
+        if balance_sht is not None:
+            st.subheader("Balance Sheet")
+            st.dataframe(balance_sht)
 
-        st.subheader("Cash Flow Statement")
-        st.dataframe(cash_flw)
+        if cash_flw is not None:
+            st.subheader("Cash Flow Statement")
+            st.dataframe(cash_flw)
 
 st.write("Developed for financial statement analysis.")