Merge pull request #6 from kushagra21-afk/master

fixing rate limiting concerns with the api calls

Author: yashksaini-coder

app.py

@@ -18,7 +18,6 @@
     raise ValueError("Please provide a GROQ API key")
 
 app = FastAPI()
-# Web searching agent
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"], 
@@ -29,10 +28,14 @@
 
 @app.get("/top-stocks")
 async def read_top_stocks():
-    top_stocks = ['AAPL', 'GOOGL', 'MSFT', 'AMZN', 'TSLA']
-    stock_info = get_top_stocks(top_stocks)
+    top_stocks = ['AAPL', 'MSFT', 'AMZN', 'GOOGL']
+    stock = " ".join(top_stocks)
+    stock_info = get_top_stocks(stock)
     return stock_info
 
+@app.get("/")
+def read_root():
+    return {"message": "Welcome to Investo!"}
 
 @app.get("/chat")
 def chat(query: str):

topStocks.py

@@ -1,19 +1,28 @@
 import yfinance as yf
+import requests 
+import time
+session = requests.Session()
+session.headers.update({
+    "User-Agent": "Chrome/122.0.0.0"
+})
 
 def get_top_stocks(symbols):
     stock_data = []
-    for symbol in symbols:
-        try:
-            ticker = yf.Ticker(symbol)
-            info = ticker.info
+    try:
+        top_stocks = symbols.split()
+        tickers = yf.Tickers(symbols)
+        while top_stocks:
+            stock = top_stocks.pop()
+            info = tickers.tickers[stock].info
             stock_info = {
-                'symbol': symbol,
-                'name': info.get('shortName', 'N/A'),
-                'currentPrice': info.get('currentPrice', 'N/A'),
-                'previousClose': info.get('previousClose', 'N/A'),
-                'sector': info.get('sector', 'N/A')
-            }
-            stock_data.append(stock_info)
-        except Exception as e:
-            print(f"Error fetching {symbol}: {e}")
+                    'symbol': stock,
+                    'name': info.get('shortName', 'N/A'),
+                    'currentPrice': info.get('currentPrice', 'N/A'),
+                    'previousClose': info.get('previousClose', 'N/A'),
+                    'sector': info.get('sector', 'N/A')
+                    }
+            stock_data.push(stock_info)
+    except Exception as e:
+            print(f"Error fetching {symbols}: {e}")
+            time.sleep(5)
     return stock_data