Update app.py

app.py

@@ -23,17 +23,27 @@ sector_averages = {
 }
 
 # Helper Functions
+def safe_request(url):
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+        return response
+    except requests.exceptions.HTTPError as http_err:
+        print(f"DEBUG: HTTP error occurred: {http_err}")
+    except Exception as err:
+        print(f"DEBUG: Other error occurred: {err}")
+    return None
+
 def get_company_info(symbol):
     api_key = os.getenv("POLYGON_API_KEY")
-    print(f"DEBUG: Using API Key: {api_key}")
+    if not api_key:
+        print("DEBUG: API Key is missing!")
+        return None
     url = f"https://api.polygon.io/v3/reference/tickers/{symbol}?apiKey={api_key}"
     print(f"DEBUG: Fetching company info from URL: {url}")
-    try:
-        response = requests.get(url)
-        print(f"DEBUG: Company Info Status Code: {response.status_code}")
-        print(f"DEBUG: Company Info Response: {response.text}")
-        response.raise_for_status()
-        data = response.json()['results']
+    response = safe_request(url)
+    if response:
+        data = response.json().get('results', {})
         return {
             'Name': data.get('name', 'N/A'),
             'Industry': data.get('sic_description', 'N/A'),
@@ -41,54 +51,42 @@ def get_company_info(symbol):
             'Market Cap': data.get('market_cap', 0),
             'Total Revenue': data.get('total_employees', 0) * 100000
         }
-    except Exception as e:
-        print(f"DEBUG: Error fetching company info: {e}")
-        return None
+    return None
 
 def get_current_price(symbol):
-    url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/prev?adjusted=true&apiKey={POLYGON_API_KEY}"
+    api_key = os.getenv("POLYGON_API_KEY")
+    url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/prev?adjusted=true&apiKey={api_key}"
     print(f"DEBUG: Fetching current price from URL: {url}")
-    try:
-        response = requests.get(url)
-        print(f"DEBUG: Current Price Status Code: {response.status_code}")
-        print(f"DEBUG: Current Price Response: {response.text}")
-        response.raise_for_status()
+    response = safe_request(url)
+    if response:
         data = response.json()['results'][0]
         return float(data['c'])
-    except Exception as e:
-        print(f"DEBUG: Error fetching current price: {e}")
-        return None
+    return None
 
 def get_dividends(symbol):
-    url = f"https://api.polygon.io/v3/reference/dividends?ticker={symbol}&apiKey={POLYGON_API_KEY}"
-    try:
-        response = requests.get(url)
-        response.raise_for_status()
+    api_key = os.getenv("POLYGON_API_KEY")
+    url = f"https://api.polygon.io/v3/reference/dividends?ticker={symbol}&apiKey={api_key}"
+    response = safe_request(url)
+    if response:
         data = response.json()['results'][0]
         return {
             'Dividend Amount': data.get('cash_amount', 0),
             'Ex-Dividend Date': data.get('ex_dividend_date', 'N/A')
         }
-    except Exception as e:
-        print(f"DEBUG: Error fetching dividends: {e}")
-        return {'Dividend Amount': 0, 'Ex-Dividend Date': 'N/A'}
+    return {'Dividend Amount': 0, 'Ex-Dividend Date': 'N/A'}
 
 def get_historical_prices(symbol):
+    api_key = os.getenv("POLYGON_API_KEY")
     end = datetime.date.today()
     start = end - datetime.timedelta(days=365)
-    url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/range/1/day/{start}/{end}?adjusted=true&sort=asc&apiKey={POLYGON_API_KEY}"
-    try:
-        response = requests.get(url)
-        print(f"DEBUG: Historical Prices Status Code: {response.status_code}")
-        print(f"DEBUG: Historical Prices Response: {response.text}")
-        response.raise_for_status()
+    url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/range/1/day/{start}/{end}?adjusted=true&sort=asc&apiKey={api_key}"
+    response = safe_request(url)
+    if response:
         results = response.json()['results']
         dates = [datetime.datetime.fromtimestamp(r['t']/1000) for r in results]
         prices = [r['c'] for r in results]
         return dates, prices
-    except Exception as e:
-        print(f"DEBUG: Error fetching historical prices: {e}")
-        return [], []
+    return [], []
 
 def generate_summary(info, ratios):
     recommendation = "Hold"
@@ -97,7 +95,6 @@ def generate_summary(info, ratios):
     elif ratios['P/E Ratio'] > 30 and ratios['P/B Ratio'] > 5 and ratios['PEG Ratio'] > 2.0:
         recommendation = "Sell"
 
-    # Much better structured prompt
     report = (
         f"Company Overview:\n"
         f"Name: {info['Name']}\n"
@@ -113,11 +110,9 @@ def generate_summary(info, ratios):
         f"Recommended Investment Action: {recommendation}.\n\n"
         f"Please provide a detailed financial analysis based on the information above."
     )
-
     summary = summarizer(report, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
     return summary
 
-
 # (Rest of the code remains the same)
 def calculate_ratios(market_cap, total_revenue, price, dividend_amount, assumed_eps=5.0, growth_rate=0.1, book_value=500000000):
     pe_ratio = price / assumed_eps if assumed_eps else 0