Merge pull request #2 from yashksaini-coder/kush

Top Stocks API endpoint

Author: yashksaini-coder

main.py

@@ -2,6 +2,8 @@
 import groq
 import os
 from dotenv import load_dotenv
+from fastapi.middleware.cors import CORSMiddleware
+from topStocks import get_top_stocks
 # Load API key from .env file
 
 load_dotenv(dotenv_path=".env")
@@ -15,6 +17,14 @@
 
 app = FastAPI()
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"], 
+    allow_credentials=True,
+    allow_methods=["*"],  
+    allow_headers=["*"],  
+)
+
 @app.get("/ask")
 def ask(query: str):
     """
@@ -34,4 +44,10 @@ def ask(query: str):
         return {"question": query, "answer": answer}
 
     except Exception as e:
-        return {"error": str(e)}
+        return {"error": str(e)}
+
+@app.get("/top-stocks")
+async def read_top_stocks():
+    top_stocks = ['AAPL', 'GOOGL', 'MSFT', 'AMZN', 'TSLA']
+    stock_info = get_top_stocks(top_stocks)
+    return stock_info

topStocks.py

@@ -0,0 +1,19 @@
+import yfinance as yf
+
+def get_top_stocks(symbols):
+    stock_data = []
+    for symbol in symbols:
+        try:
+            ticker = yf.Ticker(symbol)
+            info = ticker.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}")
+    return stock_data