Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,7 +6,7 @@ import streamlit as st
 
 st.set_page_config(layout="wide")
 
-# you need to save your FMP API Key into the respective environment variable to let it work
+# get your API key from env
 apikey = os.environ["FMP_API_KEY"]
 
 def parse_json(url):
@@ -294,6 +294,145 @@ def draw_income_sankey(income_statement, symbol, height, font_size):
     )
     return fig
 
+def draw_cashflow_sankey(cash_flow, symbol, height, font_size):
+    flows = [
+        # Operating Activities (Inflow)
+        ("Net Income",                             "Operating Activities",                 cash_flow["netIncome"]),
+        ("Depreciation & Amortization",            "Operating Activities",                 cash_flow["depreciationAndAmortization"]),
+        ("Deferred Income Tax",                    "Operating Activities",                 cash_flow["deferredIncomeTax"]),
+        ("Stock-Based Compensation",               "Operating Activities",                 cash_flow["stockBasedCompensation"]),
+        ("Change in Working Capital",              "Operating Activities",                 cash_flow["changeInWorkingCapital"]),
+        ("Accounts Receivables Δ",                 "Operating Activities",                 cash_flow["accountsReceivables"]),
+        ("Inventory Δ",                            "Operating Activities",                 cash_flow["inventory"]),
+        ("Accounts Payable Δ",                     "Operating Activities",                 cash_flow["accountsPayables"]),
+        #("Other Working Capital Δ",                "Operating Activities",                 cash_flow["otherWorkingCapital"]), # overlap with some lines
+        ("Other Non-Cash Items",                   "Operating Activities",                 cash_flow["otherNonCashItems"]),
+        ("Operating Activities",                   "Net Cash Inflow (Operating)",       cash_flow["netCashProvidedByOperatingActivities"]),
+    
+        # Investing Activities (Outflow)
+        ("Investments in PP&E",                    "Investing Activities",                 cash_flow["investmentsInPropertyPlantAndEquipment"]),
+        ("Acquisitions, Net",                      "Investing Activities",                 cash_flow["acquisitionsNet"]),
+        ("Purchases of Investments",               "Investing Activities",                 cash_flow["purchasesOfInvestments"]),
+        ("Sales/Maturities of Investments",        "Investing Activities",                 cash_flow["salesMaturitiesOfInvestments"]),
+        ("Other Investing Activities",             "Investing Activities",                 cash_flow["otherInvestingActivities"]),
+        ("Investing Activities",                   "Net Cash Outflow (Investing)",       cash_flow["netCashProvidedByInvestingActivities"]),
+    
+        # Financing Activities (Outflow)
+        ("Net Debt Issuance",                      "Financing Activities",                 cash_flow["netDebtIssuance"]),
+        #("Long-Term Net Debt Issuance",            "Financing Activities",                 cash_flow["longTermNetDebtIssuance"]), # already under net debt
+        #("Short-Term Net Debt Issuance",           "Financing Activities",                 cash_flow["shortTermNetDebtIssuance"]), # already under net debt
+        #("Net Stock Issuance",                     "Financing Activities",                 cash_flow["netStockIssuance"]),
+        ("Net Common Stock Issuance",              "Financing Activities",                 cash_flow["netCommonStockIssuance"]),
+        #("Common Stock Issuance",                  "Financing Activities",                 cash_flow["commonStockIssuance"]), # already under net common stock issuance
+        #("Common Stock Repurchased",               "Financing Activities",                 cash_flow["commonStockRepurchased"]), # already under net common stock issuance
+        ("Net Preferred Stock Issuance",           "Financing Activities",                 cash_flow["netPreferredStockIssuance"]),
+        ("Net Dividends Paid",                     "Financing Activities",                 cash_flow["netDividendsPaid"]),
+        #("Common Dividends Paid",                  "Financing Activities",                 cash_flow["commonDividendsPaid"]),  # already under net dividends paid
+        #("Preferred Dividends Paid",               "Financing Activities",                 cash_flow["preferredDividendsPaid"]), # already under net dividends paid
+        ("Other Financing Activities",             "Financing Activities",                 cash_flow["otherFinancingActivities"]),
+        ("Financing Activities",                   "Net Cash Outflow (Financing)",       cash_flow["netCashProvidedByFinancingActivities"]),
+    
+        # Combine In and Out Flows
+        ("Net Cash Inflow (Operating)",         "Net Change in Cash",                   cash_flow["netCashProvidedByOperatingActivities"]),
+        ("Net Cash Outflow (Investing)",         "Net Change in Cash",                   cash_flow["netCashProvidedByInvestingActivities"]),
+        ("Net Cash Outflow (Financing)",         "Net Change in Cash",                   cash_flow["netCashProvidedByFinancingActivities"]),
+        ("Effect of Forex on Cash",                "Net Change in Cash",                   cash_flow["effectOfForexChangesOnCash"]),
+    
+        # Beginning & Change → Ending Balance
+        ("Cash at Beginning of Period",            "Cash at End of Period",                cash_flow["cashAtBeginningOfPeriod"]),
+        ("Net Change in Cash",                     "Cash at End of Period",                cash_flow["netChangeInCash"]),
+    ]
+    
+    adjusted_flows = []
+    for src, tgt, val in flows:
+        if val >= 0:
+            adjusted_flows.append((src, tgt, val, 'rgba(50,200,50,0.6)'))
+        else:
+            # reverse direction for readability
+            adjusted_flows.append((tgt, src, -val,    'rgba(200,50,50,0.6)'))
+    
+    
+    labels = []
+    for s, t, _, _ in adjusted_flows:
+        if s not in labels: labels.append(s)
+        if t not in labels: labels.append(t)
+    
+    node_values = { lbl: cash_flow[
+        {
+          "Net Income": "netIncome",
+          "Depreciation & Amortization": "depreciationAndAmortization",
+          "Deferred Income Tax": "deferredIncomeTax",
+          "Stock-Based Compensation": "stockBasedCompensation",
+          "Change in Working Capital": "changeInWorkingCapital",
+          "Accounts Receivables Δ": "accountsReceivables",
+          "Inventory Δ": "inventory",
+          "Accounts Payable Δ": "accountsPayables",
+          "Other Working Capital Δ": "otherWorkingCapital",
+          "Other Non-Cash Items": "otherNonCashItems",
+          "Operating Activities": "netCashProvidedByOperatingActivities",
+          "Investments in PP&E": "investmentsInPropertyPlantAndEquipment",
+          "Acquisitions, Net": "acquisitionsNet",
+          "Purchases of Investments": "purchasesOfInvestments",
+          "Sales/Maturities of Investments": "salesMaturitiesOfInvestments",
+          "Other Investing Activities": "otherInvestingActivities",
+          "Investing Activities": "netCashProvidedByInvestingActivities",
+          "Net Debt Issuance": "netDebtIssuance",
+          "Long-Term Net Debt Issuance": "longTermNetDebtIssuance",
+          "Short-Term Net Debt Issuance": "shortTermNetDebtIssuance",
+          "Net Stock Issuance": "netStockIssuance",
+          "Net Common Stock Issuance": "netCommonStockIssuance",
+          "Common Stock Issuance": "commonStockIssuance",
+          "Common Stock Repurchased": "commonStockRepurchased",
+          "Net Preferred Stock Issuance": "netPreferredStockIssuance",
+          "Net Dividends Paid": "netDividendsPaid",
+          "Common Dividends Paid": "commonDividendsPaid",
+          "Preferred Dividends Paid": "preferredDividendsPaid",
+          "Other Financing Activities": "otherFinancingActivities",
+          "Financing Activities": "netCashProvidedByFinancingActivities",
+          "Net Cash Inflow (Operating)": "netCashProvidedByOperatingActivities",
+          "Net Cash Outflow (Investing)": "netCashProvidedByInvestingActivities",
+          "Net Cash Outflow (Financing)": "netCashProvidedByFinancingActivities",
+          "Effect of Forex on Cash": "effectOfForexChangesOnCash",
+          "Net Change in Cash": "netChangeInCash",
+          "Cash at Beginning of Period": "cashAtBeginningOfPeriod",
+          "Cash at End of Period": "cashAtEndOfPeriod",
+        }[lbl]
+    ] for lbl in labels }
+    
+    
+    def fmt(val):
+        if abs(val) >= 1e9: return f"${val/1e9:.1f}B"
+        if abs(val) >= 1e6: return f"${val/1e6:.1f}M"
+        if abs(val) >= 1e3: return f"${val/1e3:.0f}K"
+        return f"${val:.0f}"
+    
+    
+    idx     = { lbl:i for i,lbl in enumerate(labels) }
+    source  = [ idx[s] for s, t, _, _ in adjusted_flows ]
+    target  = [ idx[t] for s, t, _, _ in adjusted_flows ]
+    value   = [     v   for _, _, v, _ in adjusted_flows ]
+    colors  = [     c   for _, _, _, c in adjusted_flows ]
+    
+    label_with_values = []
+    for lbl in labels:
+        val  = node_values[lbl]
+        base = lbl
+        if val < 0: base += " [NEGATIVE]"
+        label_with_values.append(f"{base} ({fmt(val)})")
+
+    fig = go.Figure(go.Sankey(
+        arrangement="snap",
+        node = dict(label=label_with_values, pad=15, thickness=20),
+        link = dict(source=source, target=target, value=value, color=colors)
+    ))
+    fig.update_layout(
+        title_text=f"Income Statement Sankey — {symbol}",
+        height=height,
+        font_size=font_size
+    )
+    return fig
+
+
 st.title("Financial Sankeys")
 
 symbol = st.sidebar.text_input("Ticker symbol", "AMZN").upper()
@@ -303,6 +442,8 @@ bs_height = st.sidebar.slider("Balance Sheet height", 500, 1500, 800)
 bs_font   = st.sidebar.slider("Balance Sheet font size", 5, 15, 10)
 is_height = st.sidebar.slider("Income Statement height", 500, 1500, 600)
 is_font   = st.sidebar.slider("Income Statement font size", 5, 15, 10)
+cf_height = st.sidebar.slider("Cash Flow Statement height", 500, 1500, 800)
+cf_font   = st.sidebar.slider("Cash Flow Statement font size", 5, 15, 10)
 
 # where the data came from
 st.sidebar.markdown("## [Financial Modeling Prep API](https://site.financialmodelingprep.com/?utm_source=medium&utm_medium=medium&utm_campaign=damian8)\
@@ -319,7 +460,7 @@ if symbol:
         fig_bs = draw_balance_sankey(balance_sheet, symbol.upper(), bs_height, bs_font)
         st.plotly_chart(fig_bs, use_container_width=True)
     except Exception as e:
-        st.error(f"Failed to fetch balance sheet: {e}")
+        st.error(f"{e}")
 
     # Income Statement
     st.header(f"Income Statement — {symbol}")
@@ -329,4 +470,15 @@ if symbol:
         fig_is = draw_income_sankey(income_statement, symbol.upper(), is_height, is_font)
         st.plotly_chart(fig_is, use_container_width=True)
     except Exception as e:
-        st.error(f"Failed to fetch income statement: {e}")
+        st.error(f"{e}")
+
+    # Cash Flow Statement
+    st.header(f"Cash Flow Statement — {symbol}")
+    try:
+        df_cf = parse_json(f"https://financialmodelingprep.com/stable/cash-flow-statement?symbol={symbol}&apikey={apikey}")
+        cash_flow = df_cf.iloc[0]
+        fig_cf = draw_cashflow_sankey(cash_flow, symbol.upper(), cf_height, cf_font)
+        st.plotly_chart(fig_cf, use_container_width=True)
+    except Exception as e:
+        st.error(f"{e}")
+