Update app.py

app.py

@@ -1,5 +1,4 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-import os 
 import datetime
 import requests
 import pytz
@@ -7,38 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
-
 from transformers import pipeline
-from diffusers import StableDiffusionPipeline
-import torch
-
-
-pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")  # Use GPU if available
-
-@tool
-def generate_image(prompt: str) -> str:
-    """
-    A tool that generates an image based on a text prompt.
-    
-    Args:
-        prompt: The text prompt to generate an image from.
-    
-    Returns:
-        The file path or URL of the generated image.
-    """
-    try:
-        # Generate the image
-        image = pipe(prompt).images[0]
-        
-        # Save the image to a file
-        image_path = f"generated_images/{prompt[:20]}.png"  # Save with a unique name
-        image.save(image_path)
-        
-        return image_path
-    except Exception as e:
-        return f"Error generating image: {str(e)}"
-
 
 summarizer = pipeline("summarization")
 
@@ -63,18 +31,18 @@ def summarize_article(article_text: str) -> str:
 @tool
 def news_aggregator(keyword: str, max_results: int = 5) -> str:
     """
-    A tool that fetches the latest news articles, summarizes them and generates images.
+    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 image paths of the latest news articles.
+        A string containing the titles, summaries, and URLs of the latest news articles.
     """
     try:
         
-        api_key = 'd43565ac0e674145a22c5f0de4dcd69c'
+        api_key = "d43565ac0e674145a22c5f0de4dcd69c"  # Replace with your actual NewsAPI key
         
         # Construct the API URL
         url = f"https://newsapi.org/v2/everything?q={keyword}&pageSize={max_results}&apiKey={api_key}"
@@ -101,13 +69,9 @@ def news_aggregator(keyword: str, max_results: int = 5) -> str:
             # Summarize the article
             summary = summarize_article(content)
             
-            # Generate an image based on the title
-            image_path = generate_image(title)
-            
             # Format the results
             result += f"{i}. {title}\n"
             result += f"   Summary: {summary}\n"
-            result += f"   Image: {image_path}\n"
             result += f"   URL: {url}\n"
         
         return result
@@ -136,7 +100,7 @@ with open("prompts.yaml", 'r') as stream:
 # Agent and tools
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, news_aggregator, summarize_article, generate_image], ## add your tools here (don't remove final answer)
+    tools=[final_answer, news_aggregator, summarize_article], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,