Spaces:
Sleeping
Sleeping
| from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool | |
| import datetime | |
| import requests | |
| import xml.etree.ElementTree as ET | |
| import pytz | |
| import yaml | |
| from tools.final_answer import FinalAnswerTool | |
| from bs4 import BeautifulSoup | |
| from typing import List, Dict | |
| import urllib.request | |
| from Gradio_UI import GradioUI | |
| # Below is an example of a tool that does nothing. Amaze us with your creativity ! | |
| search_tool = DuckDuckGoSearchTool() | |
| system_message="""You have access to a variety of tools that you can use to fetch news sources. Your goal is to provide a summary of all news in a maximum of five sentences.""" | |
| def get_latest_news() -> Dict[str, List[Dict]]: | |
| """ | |
| Tool returns latest news from major news outlets using reliable RSS feeds. | |
| Returns: | |
| A summary of all titles compiled in a single concise text of maximum five sentences.Please do not provide the titles, just a five sentence summary of your making encompassing all titles. | |
| # Dict[str, List[Dict]]: A dictionary where keys are news sources and values are lists of news items. | |
| """ | |
| rss_feeds = { | |
| "NPR": { | |
| "News": "https://feeds.npr.org/1001/rss.xml", | |
| "World": "https://feeds.npr.org/1004/rss.xml", | |
| "Politics": "https://feeds.npr.org/1014/rss.xml", | |
| "Business": "https://feeds.npr.org/1006/rss.xml" | |
| }, | |
| "BBC": { | |
| "Top Stories": "http://feeds.bbci.co.uk/news/rss.xml", | |
| "World": "http://feeds.bbci.co.uk/news/world/rss.xml", | |
| "Politics": "http://feeds.bbci.co.uk/news/politics/rss.xml", | |
| "Business": "http://feeds.bbci.co.uk/news/business/rss.xml" | |
| } | |
| } | |
| news_items = {} | |
| for source, feeds in rss_feeds.items(): | |
| news_items[source] = [] | |
| print(f"\nTrying source: {source}") | |
| for feed_name, feed_url in feeds.items(): | |
| try: | |
| headers = { | |
| 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36', | |
| 'Accept': 'application/rss+xml,application/xml;q=0.9,*/*;q=0.8' | |
| } | |
| print(f" Fetching {feed_name}...", end=" ") | |
| req = urllib.request.Request(feed_url, headers=headers) | |
| response = urllib.request.urlopen(req, timeout=10) | |
| xml_data = response.read().decode('utf-8') | |
| root = ET.fromstring(xml_data) | |
| # Try different possible XML paths for items | |
| items = root.findall('.//item') | |
| if not items: | |
| items = root.findall('./channel/item') | |
| successful_items = 0 | |
| for item in items[:5]: | |
| title = item.find('title') | |
| link = item.find('link') | |
| pub_date = item.find('pubDate') | |
| description = item.find('description') | |
| # Only add items that have at least a title | |
| if title is not None and title.text: | |
| news_item = { | |
| 'category': feed_name, | |
| 'title': title.text.strip(), | |
| 'link': link.text.strip() if link is not None and link.text else '', | |
| 'published': pub_date.text if pub_date is not None else '', | |
| 'summary': description.text[:200] + '...' if description is not None and description.text else '' | |
| } | |
| news_items[source].append(news_item) | |
| successful_items += 1 | |
| print(f"Success! Found {successful_items} articles") | |
| except Exception as e: | |
| print(f"Failed: {str(e)}") | |
| news_items[source].append({ | |
| 'category': feed_name, | |
| 'title': f"Error fetching {feed_name} feed: {str(e)}", | |
| 'link': '', | |
| 'published': '', | |
| 'summary': '' | |
| }) | |
| return news_items | |
| final_answer = FinalAnswerTool() | |
| # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder: | |
| model = HfApiModel( | |
| max_tokens=2096, | |
| temperature=0.5, | |
| #model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded | |
| model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud', | |
| custom_role_conversions=None, | |
| ) | |
| with open("prompts.yaml", 'r') as stream: | |
| prompt_templates = yaml.safe_load(stream) | |
| agent = CodeAgent( | |
| model=model, | |
| tools=[final_answer, search_tool, get_latest_news], ## add your tools here (don't remove final answer) | |
| max_steps=6, | |
| verbosity_level=1, | |
| grammar=None, | |
| planning_interval=None, | |
| name=None, | |
| description=None, | |
| prompt_templates=prompt_templates | |
| ) | |
| GradioUI(agent).launch() |