Update app.py

app.py

@@ -1,27 +1,42 @@
+import yfinance as yf
 import gradio as gr
-from yahoo_fin import stock_info
-
-def get_stock_info(symbol):
-    try:
-        # Fetch real-time stock price
-        price = stock_info.get_live_price(symbol)
-        # Fetch company information
-        company_info = stock_info.get_quote_table(symbol)
-        
-        # Handling 'None' values gracefully if they exist
-        company_name = company_info.get('Name', 'N/A')
-        sector = company_info.get('Sector', 'N/A')
-        market_cap = company_info.get('Market Cap', 'N/A')
-        
-        return f"Company: {company_name}\nSector: {sector}\nMarket Cap: {market_cap}\nStock Price: ${price:.2f}"
-    except Exception as e:
-        return f"Error retrieving data: {e}"
-
-# Gradio Interface
-iface = gr.Interface(fn=get_stock_info,
-                     inputs=gr.Textbox(label="Enter Stock Symbol (e.g., AAPL)"),
-                     outputs="text",
-                     title="Stock Market Agent",
-                     description="Get real-time stock prices and company information from Yahoo Finance.")
-
-iface.launch()
+
+# Function to fetch and return data for a given ticker symbol
+def fetch_stock_data(ticker_symbol):
+    # Create a Ticker object
+    ticker = yf.Ticker(ticker_symbol)
+
+    # Fetch historical market data
+    historical_data = ticker.history(period="1y")  # data for the last year
+
+    # Fetch basic financials
+    financials = ticker.financials
+
+    # Fetch stock actions like dividends and splits
+    actions = ticker.actions
+
+    return historical_data, financials, actions
+
+# Gradio interface
+def create_gradio_interface():
+    # Define the input and output for Gradio
+    ticker_input = gr.inputs.Textbox(label="Enter Ticker Symbol")
+    historical_output = gr.outputs.Dataframe(label="Historical Data (1 Year)")
+    financials_output = gr.outputs.Dataframe(label="Financials")
+    actions_output = gr.outputs.Dataframe(label="Stock Actions")
+
+    # Define the Gradio interface
+    interface = gr.Interface(
+        fn=fetch_stock_data,
+        inputs=ticker_input,
+        outputs=[historical_output, financials_output, actions_output],
+        live=True,
+        title="Stock Data Fetcher",
+        description="Enter a ticker symbol to fetch historical data, financials, and stock actions."
+    )
+    
+    # Launch the interface
+    interface.launch()
+
+# Run the Gradio interface
+create_gradio_interface()