| #import gspread | |
| def match_companies_to_articles(articles_metadata, ai_results): | |
| # A. Create a lookup dictionary: URL -> Title | |
| # This allows instant access to titles without looping every time | |
| url_to_title_map = {item['link']: item['title'] for item in articles_metadata} | |
| final_list = [] | |
| for result in ai_results: | |
| article_url = result.get('url') | |
| # Look up the title, default to "Unknown" if the URL isn't in metadata | |
| article_title = url_to_title_map.get(article_url, "Unknown Title") | |
| # Iterate through the companies found in this specific article | |
| if 'companies' in result: | |
| for company in result['companies']: | |
| record = { | |
| "company_name": company['name'], | |
| "company_url": company.get('url', ''), # Handle missing URLs gracefully | |
| "article_title": article_title, | |
| "article_url": article_url | |
| } | |
| final_list.append(record) | |
| results = sorted(final_list, key=lambda x: x['company_name']) | |
| return results | |
| # | |
| # def connect_to_sheet(json_keyfile, sheet_name): | |
| # """Authenticates and returns the worksheet object.""" | |
| # try: | |
| # gc = gspread.service_account(filename=json_keyfile) | |
| # sh = gc.open(sheet_name) | |
| # return sh.sheet1 | |
| # except Exception as e: | |
| # print(f"❌ Error connecting to Google Sheets: {e}") | |
| # return None | |
| # | |
| # | |
| # def get_cached_websites(worksheet): | |
| # """ | |
| # Returns a dictionary of existing companies: {'Tesla': 'tesla.com', ...} | |
| # """ | |
| # if not worksheet: return {} | |
| # | |
| # print("📂 Reading cache from Google Sheets...") | |
| # try: | |
| # records = worksheet.get_all_records() | |
| # # Convert list of dicts to a lookup map | |
| # return { | |
| # row['company_name']: row['company_website'] | |
| # for row in records | |
| # if row.get('company_name') | |
| # } | |
| # except Exception: | |
| # return {} | |
| # | |
| # | |
| # def save_new_websites(worksheet, new_data): | |
| # """ | |
| # Appends new data to the sheet. | |
| # Expects a list of dicts: [{'company_name': 'X', 'company_website': 'Y'}] | |
| # """ | |
| # if not worksheet or not new_data: return | |
| # | |
| # print(f"💾 Saving {len(new_data)} new entries to Google Sheets...") | |
| # | |
| # # Prepare rows as list of lists: [['Name', 'URL'], ['Name', 'URL']] | |
| # rows = [[item['company_name'], item['company_website']] for item in new_data] | |
| # | |
| # # Add headers if sheet is empty | |
| # if not worksheet.get_all_values(): | |
| # worksheet.append_row(["company_name", "company_website"]) | |
| # | |
| # worksheet.append_rows(rows) | |