Update app.py

app.py

@@ -19,14 +19,14 @@ def generate_financial_statements(file, file_type):
     if file_type == "csv":
         df = pd.read_csv(file)
         context = df.to_string()
-    elif file_type == "excel":
+    elif file_type == "xlsx":
         df = pd.read_excel(file)
         context = df.to_string()
-    elif file_type == "docs":
+    elif file_type == "docx":
         context = read_docs(file)
     else:
         st.error("Unsupported file type. Please upload a CSV, Excel, or DOCS file.")
-        return
+        return None
 
     # Define financial statement queries
     queries = [
@@ -60,10 +60,13 @@ st.write("Upload your financial data (CSV, Excel, or DOCS) to generate journal,
 # File upload
 uploaded_file = st.file_uploader("Upload your file", type=["csv", "xlsx", "docx"])
 if uploaded_file is not None:
-    file_type = uploaded_file.name.split(".")[-1]
+    file_type = uploaded_file.name.split(".")[-1].lower()  # Ensure lowercase file type
     financial_statements = generate_financial_statements(uploaded_file, file_type)
 
-    # Display results
-    for statement_type, statement in financial_statements.items():
-        st.subheader(statement_type)
-        st.write(statement)
+    # Display results if financial_statements is not None
+    if financial_statements is not None:
+        for statement_type, statement in financial_statements.items():
+            st.subheader(statement_type)
+            st.write(statement)
+    else:
+        st.error("Failed to generate financial statements. Please check the file type and content.")