Update app.py

app.py

@@ -69,7 +69,10 @@ def n_weeks_before(date_string, n):
 
 def get_stock_data(stock_symbol, steps):
 
-    stock_data = yf.download(stock_symbol, steps[0], steps[-1])
+    try:
+        stock_data = yf.download(stock_symbol, steps[0], steps[-1])
+    except Exception:
+        raise gr.Error(f"Failed to download stock price data for symbol {stock_symbol} from yfinance!")
     
 #     print(stock_data)
     
@@ -101,7 +104,10 @@ 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
-        weekly_news = finnhub_client.company_news(symbol, _from=start_date, to=end_date)
+        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 = [
             {
                 "date": datetime.fromtimestamp(n['datetime']).strftime('%Y%m%d%H%M%S'),
@@ -118,9 +124,12 @@ def get_news(symbol, data):
 
 
 def get_company_prompt(symbol):
-    
-    profile = finnhub_client.company_profile2(symbol=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!")
+        
     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."
 
@@ -159,8 +168,11 @@ def sample_news(news, k=5):
 
 def get_current_basics(symbol, curday):
 
-    basic_financials = finnhub_client.company_basic_financials(symbol, 'all')
-    
+    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!")
+        
     final_basics, basic_list, basic_dict = [], [], defaultdict(dict)
     
     for metric, value_list in basic_financials['series']['quarterly'].items():
@@ -220,7 +232,11 @@ def get_all_prompts_online(symbol, data, curday, with_basics=True):
 
 def construct_prompt(ticker, curday, n_weeks, use_basics):
 
-    steps = [n_weeks_before(curday, n) for n in range(n_weeks + 1)][::-1]
+    try:
+        steps = [n_weeks_before(curday, n) for n in range(n_weeks + 1)][::-1]
+    except Exception:
+        raise gr.Error(f"Invalid date {curday}!")
+        
     data = get_stock_data(ticker, steps)
     data = get_news(ticker, data)
     data['Basics'] = [json.dumps({})] * len(data)