Update app.py

app.py

@@ -61,6 +61,7 @@ def generate_charts(pl_df, cf_df):
     # Save PL chart to BytesIO
     pl_chart_io = BytesIO()
     plt.savefig(pl_chart_io, format='png')
+    plt.savefig('pl_chart_base64.png') # Save the image
     plt.close()
     pl_chart_io.seek(0)
     pl_chart_base64 = base64.b64encode(pl_chart_io.read()).decode('utf-8')
@@ -78,10 +79,12 @@ def generate_charts(pl_df, cf_df):
     # Save CF chart to BytesIO
     cf_chart_io = BytesIO()
     plt.savefig(cf_chart_io, format='png')
+    plt.savefig('cf_chart_base64.png') # Save the image
     plt.close()
     cf_chart_io.seek(0)
     cf_chart_base64 = base64.b64encode(cf_chart_io.read()).decode('utf-8')
     
+    
     return pl_chart_base64, cf_chart_base64
     
 def generate_financials(assumptions_text):
@@ -155,8 +158,8 @@ def generate_financials(assumptions_text):
     
     # 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}"
-
+    #return 'financial_projections.xlsx', f"data:image/png;base64,{pl_chart_base64}", f"data:image/png;base64,{cf_chart_base64}"
+    return 'financial_projections.xlsx','pl_chart_base64.png','cf_chart_base64.png'
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Financial Projections for Tea Production Machine Replacement")