Update utils.py

utils.py

@@ -42,7 +42,7 @@ class StockDataFetcher:
         }
         try:
             print("Downloading data of", symbol.upper())
-            if symbol[-2:].upper() == "PE" or symbol[-2:].upper() == "CE" or symbol[-3:].upper() == "FUT":
+            if symbol[-2:].upper() == "PE" or symbol[-2:].upper() == "CE" or symbol[-3:].upper() == "FUT" or symbol.upper() == "NIFTY" or symbol.upper() == "BANKNIFTY":
               response = requests.get(self.base_fno_url + symbol.upper(), params=params, headers=self.headers)
             else:
               response = requests.get(self.base_url + symbol.upper(), params=params, headers=self.headers)
@@ -61,7 +61,7 @@ class StockDataFetcher:
 
     def fetch_latest_price(self, symbol):
       try:
-        if symbol[-2:].upper() == "PE" or symbol[-2:].upper() == "CE" or symbol[-3:].upper() == "FUT":
+        if symbol[-2:].upper() == "PE" or symbol[-2:].upper() == "CE" or symbol[-3:].upper() == "FUT" or symbol.upper() == "NIFTY" or symbol.upper() == "BANKNIFTY":
           response = requests.get(self.latest_option_price + symbol.upper() + "/latest", headers=self.headers)
         else:
           response = requests.get(self.latest_stock_price + symbol.upper() + "/latest", headers=self.headers)