Update app.py

app.py

@@ -43,7 +43,45 @@ def parse_assumptions(assumptions_text):
             params[key] = value
     
     return params
-
+    
+def generate_charts(pl_df, cf_df):
+    # Create Profit Loss chart
+    plt.figure(figsize=(10, 6))
+    plt.plot(pl_df['Year'], pl_df['Revenue'], label='Revenue', marker='o')
+    plt.plot(pl_df['Year'], pl_df['Net Income'], label='Net Income', marker='s')
+    plt.title('Profit and Loss Trend')
+    plt.xlabel('Year')
+    plt.ylabel('Amount (Rp)')
+    plt.grid(True)
+    plt.legend()
+    plt.tight_layout()
+    
+    # Save PL chart to BytesIO
+    pl_chart_io = BytesIO()
+    plt.savefig(pl_chart_io, format='png')
+    plt.close()
+    pl_chart_io.seek(0)
+    pl_chart_base64 = base64.b64encode(pl_chart_io.read()).decode('utf-8')
+    
+    # Create Cashflow chart
+    plt.figure(figsize=(10, 6))
+    plt.plot(cf_df['Year'], cf_df['Net Cashflow'], label='Net Cashflow', marker='o', color='green')
+    plt.title('Cashflow Trend')
+    plt.xlabel('Year')
+    plt.ylabel('Amount (Rp)')
+    plt.grid(True)
+    plt.legend()
+    plt.tight_layout()
+    
+    # Save CF chart to BytesIO
+    cf_chart_io = BytesIO()
+    plt.savefig(cf_chart_io, format='png')
+    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):
     params = parse_assumptions(assumptions_text)
     print("halo mulai", params)
@@ -113,7 +151,7 @@ def generate_financials(assumptions_text):
         
     print("Excel file 'output.xlsx' created with two sheets.")
 
-    return 'financial_projections.xlsx'
+    return 'financial_projections.xlsx', f"data:image/png;base64,{pl_chart_base64}", f"data:image/png;base64,{cf_chart_base64}"
 
 # Gradio interface
 with gr.Blocks() as demo: