Update tools/extract_market_data.py

tools/extract_market_data.py

@@ -23,15 +23,43 @@ class ExtractMarketDataTool(Tool):
         try:
             data = yf.Ticker(f"{stock}")
             info_data = data.info
+            # INFO
+            address = info_data['address1']
+            city_state_zip_country = f"{info_data['city']}, {info_data['state']}, {info_data['zip']}, {info_data['country']}"
+            phone = f"{info_data['phone']}"
+            website = f"{info_data['website']}"
+            industry = f"{info_data['industry']}"
+            sector = f"{info_data['sector']}"
+
             calendar_data = data.calendar
-            analyst_price_targets_data = data.analyst_price_targets
-            quarterly_income_stmt_data = data.quarterly_income_stmt
+            # CALENDAR
+            dividend_date = calendar_data['Dividend Date']
+            ex_dividend_date = calendar_data['Ex-Dividend Date']
+            earnings_date = calendar_data['Earnings Date']
+            earnings_high = calendar_data['Earnings High']
+            earnings_low = calendar_data['Earnings Low']
+            earnings_avg = calendar_data['Earnings Average']
+            rev_high = calendar_data['Revenue High']
+            rev_low = calendar_data['Revenue Low']
+            rev_avg = calendar_data['Revenue Average']
+            
 
-            output = (f"{stock} QUICK INFO:\n"
-                      f"INFO: {info_data}\n"
-                      f"CALENDAR: {calendar_data}\n"
-                      f"ANALYST PRICE TARGETS: {analyst_price_targets_data}\n"
-                      f"QUARTERLY INCOME STMT: {quarterly_income_stmt_data}")
+            output = (f"Here's information about {stock}:\n"
+                      f"address: {address}\n"
+                      f"city, state, zip, country: {city_state_zip_country}\n"
+                      f"phone: {phone}\n"
+                      f"website: {website}\n"
+                      f"industry: {industry}\n"
+                      f"sector: {sector}"\n
+                      f"dividend date: {dividend_date}\n"
+                      f"ex-dividend date: {ex_dividend_date}\n"
+                      f"earnings date: {earnings_date}\n"
+                      f"earnings high: {earnings_high}\n"
+                      f"earnings low: {earnings_low}\n"
+                      f"earnings average: {earnings_avg}\n"
+                      f"revenue high: {rev_high}\n"
+                      f"revenue low: {rev_low}\n"
+                      f"revenue average: {rev_avg}\n")
             return output
         except Exception as e:
             return f"An unexpected error occurred: {str(e)}"