Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import matplotlib.pyplot as plt
 from openpyxl import Workbook
 from io import BytesIO
+import base64
 import re
 
 def parse_assumptions(assumptions_text):
@@ -151,7 +152,9 @@ def generate_financials(assumptions_text):
         cf_df.to_excel(writer, sheet_name='Sheet2', index=False)
         
     print("Excel file 'output.xlsx' created with two sheets.")
-
+    
+    # Generate charts
+    pl_chart_base64, cf_chart_base64 = generate_charts(pl_df, cf_df)
     return 'financial_projections.xlsx', f"data:image/png;base64,{pl_chart_base64}", f"data:image/png;base64,{cf_chart_base64}"
 
 # Gradio interface