Update app.py

app.py

@@ -69,9 +69,8 @@ def n_weeks_before(date_string, n):
 
 def get_stock_data(stock_symbol, steps):
 
-    try:
-        stock_data = yf.download(stock_symbol, steps[0], steps[-1])
-    except Exception:
+    stock_data = yf.download(stock_symbol, steps[0], steps[-1])
+    if len(stock_data) == 0:
         raise gr.Error(f"Failed to download stock price data for symbol {stock_symbol} from yfinance!")
     
 #     print(stock_data)
@@ -104,10 +103,9 @@ def get_news(symbol, data):
         end_date = row['End Date'].strftime('%Y-%m-%d')
 #         print(symbol, ': ', start_date, ' - ', end_date)
         time.sleep(1) # control qpm
-        try:
-            weekly_news = finnhub_client.company_news(symbol, _from=start_date, to=end_date)
-        except Exception:
-            raise gr.Error(f"Failed to download company news for symbol {stock_symbol} from finnhub!")
+        weekly_news = finnhub_client.company_news(symbol, _from=start_date, to=end_date)
+        if len(weekly_news) == 0:
+            raise gr.Error(f"No company news found for symbol {stock_symbol} from finnhub!")
         weekly_news = [
             {
                 "date": datetime.fromtimestamp(n['datetime']).strftime('%Y%m%d%H%M%S'),
@@ -125,10 +123,9 @@ def get_news(symbol, data):
 
 def get_company_prompt(symbol):
 
-    try:
-        profile = finnhub_client.company_profile2(symbol=symbol)
-    except Exception:
-        raise gr.Error(f"Failed to download company profile for symbol {stock_symbol} from finnhub!")
+    profile = finnhub_client.company_profile2(symbol=symbol)
+    if not profile:
+        raise gr.Error(f"Failed to find company profile for symbol {stock_symbol} from finnhub!")
         
     company_template = "[Company Introduction]:\n\n{name} is a leading entity in the {finnhubIndustry} sector. Incorporated and publicly traded since {ipo}, the company has established its reputation as one of the key players in the market. As of today, {name} has a market capitalization of {marketCapitalization:.2f} in {currency}, with {shareOutstanding:.2f} shares outstanding." \
         "\n\n{name} operates primarily in the {country}, trading under the ticker {ticker} on the {exchange}. As a dominant force in the {finnhubIndustry} space, the company continues to innovate and drive progress within the industry."
@@ -168,10 +165,9 @@ def sample_news(news, k=5):
 
 def get_current_basics(symbol, curday):
 
-    try:
-        basic_financials = finnhub_client.company_basic_financials(symbol, 'all')
-    except Exception:
-        raise gr.Error(f"Failed to download basic financials for symbol {stock_symbol} from finnhub!")
+    basic_financials = finnhub_client.company_basic_financials(symbol, 'all')
+    if not basic_financials['series']:
+        raise gr.Error(f"Failed to find basic financials for symbol {stock_symbol} from finnhub!")
         
     final_basics, basic_list, basic_dict = [], [], defaultdict(dict)