Update app.py

app.py

@@ -2,6 +2,11 @@ import pandas as pd
 import numpy as np
 import gradio as gr
 import matplotlib.pyplot as plt
+import yfinance as yf
+from transformers import pipeline
+
+# Initialize Summarizer
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 # Simple DCF Calculator Setup
 def discounted_cash_flow(fcf, growth_rate, discount_rate, terminal_growth_rate, forecast_years=5):
@@ -24,7 +29,25 @@ 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 dcf_interface(fcf, growth_rate, discount_rate, terminal_growth_rate, forecast_years):
+def get_company_fcf(symbol):
+    ticker = yf.Ticker(symbol)
+    try:
+        cashflow = ticker.cashflow
+        fcf = cashflow.loc['Total Cash From Operating Activities'][0] - cashflow.loc['Capital Expenditures'][0]
+        return fcf
+    except Exception:
+        return None
+
+def generate_summary(symbol, intrinsic_value):
+    text = f"Based on a DCF analysis, the intrinsic value of {symbol} is estimated to be ${intrinsic_value:,.2f}."
+    summary = summarizer(text, max_length=100, min_length=30, do_sample=False)[0]['summary_text']
+    return summary
+
+def dcf_interface(symbol, growth_rate, discount_rate, terminal_growth_rate, forecast_years):
+    fcf = get_company_fcf(symbol)
+    if fcf is None:
+        return "Error fetching financial data. Please check the symbol or try another company.", None, None
+
     fcf_forecast, present_values, terminal_value, terminal_value_pv, total_intrinsic_value = discounted_cash_flow(
         fcf, growth_rate, discount_rate, terminal_growth_rate, forecast_years
     )
@@ -42,17 +65,13 @@ def dcf_interface(fcf, growth_rate, discount_rate, terminal_growth_rate, forecas
     ax.set_ylabel('Free Cash Flow ($)')
     ax.grid(True)
 
-    summary = f"""
-🏦 Total Intrinsic Value Estimate: ${total_intrinsic_value:,.2f}
-Terminal Value (undiscounted): ${terminal_value:,.2f}
-Present Value of Terminal Value: ${terminal_value_pv:,.2f}
-"""
+    summary = generate_summary(symbol, total_intrinsic_value)
     return df, fig, summary
 
 iface = gr.Interface(
     fn=dcf_interface,
     inputs=[
-        gr.Number(label="Initial Free Cash Flow ($)"),
+        gr.Textbox(label="Stock Symbol (e.g., AAPL)"),
         gr.Number(label="Annual Growth Rate (e.g., 0.06 for 6%)"),
         gr.Number(label="Discount Rate (e.g., 0.08 for 8%)"),
         gr.Number(label="Terminal Growth Rate (e.g., 0.025 for 2.5%)"),
@@ -61,14 +80,14 @@ iface = gr.Interface(
     outputs=[
         gr.Dataframe(label="DCF Forecast Table"),
         gr.Plot(label="Free Cash Flow Forecast Chart"),
-        gr.Textbox(label="Summary")
+        gr.Textbox(label="AI Summary")
     ],
     examples=[
-        [100_560_000_000, 0.06, 0.08, 0.025, 5]  # Example: Apple
+        ["AAPL", 0.06, 0.08, 0.025, 5]
     ],
-    title="DCF Valuation Calculator",
-    description="Estimate a company's intrinsic value using Discounted Cash Flow (DCF) analysis. Adjust inputs or run a preset example."
+    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."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()