Update app.py

app.py

@@ -2,7 +2,8 @@ import pandas as pd
 import numpy as np
 import gradio as gr
 import matplotlib.pyplot as plt
-import yfinance as yf
+import requests
+import os
 from transformers import pipeline
 
 # Initialize Summarizer
@@ -30,16 +31,17 @@ def discounted_cash_flow(fcf, growth_rate, discount_rate, terminal_growth_rate,
     return fcf_forecast, present_values, terminal_value, terminal_value_pv, total_value
 
 def get_company_fcf(symbol):
-    ticker = yf.Ticker(symbol)
+    api_key = os.getenv("POLYGON_API_KEY")
+    url = f"https://api.polygon.io/vX/reference/financials?ticker={symbol}&apiKey={api_key}"
     try:
-        fcf = ticker.free_cashflow
-        if fcf is not None:
-            return fcf.iloc[0]  # Get the most recent free cash flow
-        else:
-            cashflow = ticker.cashflow
-            operating_cash_flow = cashflow.loc["Total Cash From Operating Activities"].iloc[0]
-            capex = cashflow.loc["Capital Expenditures"].iloc[0]
-            return operating_cash_flow + capex  # Capex is negative so add
+        response = requests.get(url)
+        response.raise_for_status()
+        data = response.json()
+
+        # Navigate the JSON to find Free Cash Flow
+        fcf = data['results'][0]['financials']['cash_flow_statement']['free_cash_flow']['value']
+        return float(fcf)
+
     except Exception as e:
         print(f"DEBUG: Error fetching FCF for {symbol}: {e}")
         return None
@@ -98,8 +100,8 @@ iface = gr.Interface(
         ["AAPL", 0.06, 0.08, 0.025, 5]
     ],
     title="AI-Powered DCF Valuation Calculator",
-    description="Enter a stock symbol and your assumptions. The app pulls live data, runs a DCF, and generates a research summary."
+    description="Enter a stock symbol and your assumptions. The app pulls live data from Polygon.io, runs a DCF, and generates a research summary."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()