Update app.py

app.py

@@ -6,80 +6,30 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
-from transformers import pipeline
 
-summarizer = pipeline("summarization")
-
-@tool
-def summarize_article(article_text: str) -> str:
-    """
-    A tool that summarizes the content of a news article.
-    
-    Args:
-        article_text: The text content of the article to summarize.
-    
-    Returns:
-        A summarized version of the article.
-    """
-    try:
-        # Summarize the article text
-        summary = summarizer(article_text, max_length=50, min_length=25, do_sample=False)
-        return summary[0]['summary_text']
-    except Exception as e:
-        return f"Error summarizing article: {str(e)}"
+import os
 
 @tool
 def news_aggregator(keyword: str, max_results: int = 5) -> str:
-    """
-    A tool that fetches the latest news articles and summarizes them.
-    
-    Args:
-        keyword: The topic or keyword to search for (e.g., "AI", "technology").
-        max_results: The maximum number of articles to return (default is 5).
-    
-    Returns:
-        A string containing the titles, summaries, and URLs of the latest news articles.
-    """
     try:
-        
-        api_key = "d43565ac0e674145a22c5f0de4dcd69c"  # Replace with your actual NewsAPI key
-        
-        # Construct the API URL
+        api_key = os.getenv("NEWSAPI_KEY")  # Fetch the API key from environment variables
+        if not api_key:
+            return "NewsAPI key is missing. Please set the NEWSAPI_KEY environment variable."
         url = f"https://newsapi.org/v2/everything?q={keyword}&pageSize={max_results}&apiKey={api_key}"
-        
-        # Make the API request
         response = requests.get(url)
-        response.raise_for_status()  # Raise an error for bad status codes
-        
-        # Parse the JSON response
+        response.raise_for_status()
         data = response.json()
-        
-        # Check if the response contains articles
         articles = data.get("articles", [])
         if not articles:
-            return f"No news articles found for the keyword '{keyword}'."
-        
-        # Process each article
-        result = f"Here are the latest news articles about '{keyword}':\n"
+            return "No news articles found for the given keyword."
+        result = "Here are the latest news articles:\n"
         for i, article in enumerate(articles, 1):
             title = article.get("title", "No title")
             url = article.get("url", "#")
-            content = article.get("content", "")  # Get the article content
-            
-            # Summarize the article
-            summary = summarize_article(content)
-            
-            # Format the results
-            result += f"{i}. {title}\n"
-            result += f"   Summary: {summary}\n"
-            result += f"   URL: {url}\n"
-        
+            result += f"{i}. {title}\n   {url}\n"
         return result
-    
     except requests.exceptions.RequestException as e:
         return f"Error fetching news articles: {str(e)}"
-    except Exception as e:
-        return f"An unexpected error occurred: {str(e)}"
 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:
@@ -100,7 +50,7 @@ with open("prompts.yaml", 'r') as stream:
 # Agent and tools
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, news_aggregator, summarize_article], ## add your tools here (don't remove final answer)
+    tools=[final_answer, news_aggregator], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,