Update app.py

app.py

@@ -10,48 +10,43 @@ from Gradio_UI import GradioUI
 #S&P500 Sentiment StoryTeller
 @tool
 def market_sentiment_story(index: str) -> str:
-    """Fetches market index data from Financial Modeling Prep and returns a story prompt.
-    
+    """Fetches market data for a stock index and returns a prompt for generating a story.
     Args:
         index: One of 'sp500', 'dowjones', or 'nasdaq'.
     """
-    index = index.lower()
-
-    # Mapping user input to FMP-compatible symbols
+    # 1. Map user input to Yahoo Finance symbols
     symbols = {
         "sp500": "^GSPC",
         "dowjones": "^DJI",
         "nasdaq": "^IXIC"
     }
 
-    if index not in symbols:
+    # 2. Get the correct symbol for the requested index
+    symbol = symbols.get(index.lower())
+    if not symbol:
         return "Please choose one of: 'sp500', 'dowjones', or 'nasdaq'."
 
-    symbol = symbols[index]
-
     try:
-        # Use demo API key from FMP
-        url = f"https://financialmodelingprep.com/api/v3/quote/{symbol}?apikey=demo"
-        response = requests.get(url)
-        data = response.json()
-
-        if not data or not isinstance(data, list):
-            return "Could not retrieve market data at the moment."
-
-        price_info = data[0]
-        name = price_info.get("name", index.upper())
-        current_price = price_info["price"]
-        change_percent = price_info["changesPercentage"]
-
-        # Return a prompt for the LLM to expand into a story
+        # 3. Fetch data from Yahoo Finance
+        url = f"https://query1.finance.yahoo.com/v8/finance/chart/{symbol}"
+        data = requests.get(url).json()
+
+        # 4. Extract current and previous prices
+        result = data["chart"]["result"][0]["meta"]
+        current = result["regularMarketPrice"]
+        previous = result["chartPreviousClose"]
+        change = current - previous
+        percent = (change / previous) * 100
+
+        # 5. Return a prompt for the LLM to generate a story
         return (
-            f"{name} is currently at {current_price:.2f}, "
-            f"which is a change of {change_percent:.2f}% today. "
-            f"Write a short, realistic story explaining this market movement."
+            f"The {index.upper()} is currently at {current:.2f}, "
+            f"which is a change of {percent:.2f}% from the previous close. "
+            f"Write a short, realistic story that explains why it moved in this direction."
         )
 
     except Exception as e:
-        return f"Error fetching data: {str(e)}"
+        return f"Error: {str(e)}"
 
 
 @tool