Enhance Gradio UI and add news fetching tool with DuckDuckGo integration

Gradio_UI.py

@@ -283,7 +283,7 @@ class GradioUI:
                     [upload_file, file_uploads_log],
                     [upload_status, file_uploads_log],
                 )
-            text_input = gr.Textbox(lines=1, label="Chat Message")
+            text_input = gr.Textbox(lines=1, label="Chat Message :)")
             text_input.submit(
                 self.log_user_message,
                 [text_input, file_uploads_log],

app.py

@@ -1,22 +1,61 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from duckduckgo_search import DDGS
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+import bs4
 from Gradio_UI import GradioUI
+from playwright.sync_api import sync_playwright
+
+def fetch_url(url: str):
+    with sync_playwright() as p:
+        browser = p.webkit.launch()
+        page = browser.new_page()
+        page.goto(url)
+        print(f"fetch: {page.title()} in {url}")
+        content = bs4.BeautifulSoup(page.content(), features="lxml")
+
+        result = ""
+        for chunk in content.find_all('p'):
+            result += chunk.text + "\n"
+        browser.close()
+    return result
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_cutom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def get_news_from_duckduckgo(keywords: str)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that does nothing yet 
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        keywords: A string of keywords to search for in the news.
     """
-    return "What magic will you build ?"
+    ddgs = DDGS()
+    results = ddgs.news(keywords=keywords, timelimit="w", max_results=5)
+    
+    if len(results) == 0:
+        raise Exception("No results found! Try a less restrictive/shorter query.")
+    
+    model = HfApiModel(
+        max_tokens=1048,
+        temperature=0.5,
+        model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',# it is possible that this model may be overloaded
+        custom_role_conversions=None,
+        )
+    
+    summaries = []
+    for result in results:
+        content = fetch_url(result['url'])
+        messages = [
+            {"role": "system", "content": [{"type": "text", "text": "You are a news bot. Please summarize the news articles."}]},
+            {"role": "user", "content": [{"type": "text", "text": result['title'] + "\n" + content}]} 
+        ]
+        summary = model(messages).content
+        print(summary)
+        summaries.append(summary)    
+
+    return "## Search Results\n\n" + "\n\n".join(summaries)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -36,7 +75,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 model = HfApiModel(
-max_tokens=2096,
+max_tokens=2096*4,
 temperature=0.5,
 model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',# it is possible that this model may be overloaded
 custom_role_conversions=None,
@@ -51,7 +90,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_news_from_duckduckgo], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

requirements.txt

@@ -3,3 +3,4 @@ smolagents
 requests
 duckduckgo_search
 pandas
+gradio

tmp.py

@@ -0,0 +1,41 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Requirement already satisfied: playwright in c:\\users\\janal\\appdata\\local\\programs\\python\\python312\\lib\\site-packages (1.50.0)\n",
+      "Requirement already satisfied: pyee<13,>=12 in c:\\users\\janal\\appdata\\local\\programs\\python\\python312\\lib\\site-packages (from playwright) (12.1.1)\n",
+      "Requirement already satisfied: greenlet<4.0.0,>=3.1.1 in c:\\users\\janal\\appdata\\local\\programs\\python\\python312\\lib\\site-packages (from playwright) (3.1.1)\n",
+      "Requirement already satisfied: typing-extensions in c:\\users\\janal\\appdata\\local\\programs\\python\\python312\\lib\\site-packages (from pyee<13,>=12->playwright) (4.12.2)\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "[notice] A new release of pip is available: 24.2 -> 25.0.1\n",
+      "[notice] To update, run: python.exe -m pip install --upgrade pip\n"
+     ]
+    }
+   ],
+   "source": [
+    "%%python -m pip install playwright\n",
+    "%%playwright install"
+   ]
+  }
+ ],
+ "metadata": {
+  "language_info": {
+   "name": "python"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}