Upload folder using huggingface_hub

server/stock_price.py

@@ -3,27 +3,33 @@
 import yfinance as yf
 
 
+    
+
 def get_stock_price(symbol, exchange, days=None):
     try:
-        exchange_normalized = str(exchange).lower()
-        if exchange_normalized == "nse":
+        symbol = symbol.upper()
+        exchange = exchange.lower()
+
+        if exchange == "nse" or exchange == "NSE":
             ticker = symbol + ".NS"
-        elif exchange_normalized == "bse":
+        elif exchange == "bse" or exchange == "BSE":
             ticker = symbol + ".BO"
         else:
             return []
 
         stock = yf.Ticker(ticker)
-        data = stock.history(period="1d") if days is None else stock.history(period=f"{days}d")
+
+        period = f"{days}d" if days else "5d"
+
+        data = stock.history(period=period)
 
         if data.empty:
+            print("No data for:", ticker)
             return []
 
-        prices = data['Close'].tolist()
-        
-        return prices
+        return data['Close'].tolist()
 
-    except Exception:
+    except Exception as e:
+        print("ERROR:", e)
         return []
-    
 # print(get_stock_price("RELIANCE","NSE",days=5))