Update app.py

app.py

@@ -5,15 +5,14 @@ import requests
 import yaml
 import os
 from typing import Dict, List, Optional
-
 @tool
 def fetch_news(topic: str, num_results: int = 5) -> List[Dict]:
     """Fetches recent news articles about any topic using Serper.dev.
-    
+
     Args:
         topic: The topic to search for news about
         num_results: Number of news articles to retrieve (default: 5)
-    
+
     Returns:
         List of dictionaries containing article information
     """
@@ -21,7 +20,7 @@ def fetch_news(topic: str, num_results: int = 5) -> List[Dict]:
         api_key = os.environ.get("SERPER_API_KEY")
         if not api_key:
             return "Error: SERPER_API_KEY not found in environment variables"
-        
+
         url = f"https://google.serper.dev/news"
         headers = {
             "X-API-KEY": api_key
@@ -31,15 +30,15 @@ def fetch_news(topic: str, num_results: int = 5) -> List[Dict]:
             "gl": "us",
             "hl": "en"
         }
-        
+
         response = requests.get(url, headers=headers, params=params)
         response.raise_for_status()
-        
+
         results = response.json()
-        
+
         if "news" not in results:
             return []
-            
+
         articles = []
         for article in results["news"][:num_results]:
             articles.append({
@@ -49,33 +48,32 @@ def fetch_news(topic: str, num_results: int = 5) -> List[Dict]:
                 'link': article.get('link', 'No link'),
                 'snippet': article.get('snippet', 'No preview available')
             })
-        
+
         return articles
-        
+
     except Exception as e:
         return f"Error: {str(e)}"
-
 @tool
 def summarize_news(articles: List[Dict]) -> str:
     """Creates a summary of the news articles followed by a list of sources.
-    
+
     Args:
         articles: List of article dictionaries containing title, source, date, link, and snippet
-    
+
     Returns:
         A string containing a summary followed by article references
     """
     if not articles or not isinstance(articles, list):
         return "No articles to summarize"
-    
+
     # Collect all snippets for the overall summary
     all_snippets = [article['snippet'] for article in articles if article.get('snippet')]
-    
+
     # Create a high-level summary from snippets
     summary = "📰 Summary:\n"
     summary += "Latest news covers " + ", ".join(set(article['source'] for article in articles)) + ". "
     summary += "Key points: " + ". ".join(all_snippets[:2]) + "\n\n"
-    
+
     # List individual articles
     summary += "🔍 Articles:\n"
     for idx, article in enumerate(articles, 1):
@@ -83,17 +81,15 @@ def summarize_news(articles: List[Dict]) -> str:
         link = article['link']
         date = article['date']
         snippet = article['snippet'][:100] + "..." if len(article['snippet']) > 100 else article['snippet']
-        
-        summary += f"{idx}. **{title}**\n"
+
+        summary += f"{idx}. {title}\n"
         summary += f"   {snippet}\n"
         summary += f"   [Read more]({link}) ({date})\n\n"
-    
-    return summary
 
+    return summary
 # Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-
 # Initialize the model
 model = HfApiModel(
     max_tokens=2096,
@@ -101,9 +97,7 @@ model = HfApiModel(
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
 )
-
 final_answer = FinalAnswerTool()
-
 # Create the agent with all tools
 agent = CodeAgent(
     model=model,
@@ -116,13 +110,6 @@ agent = CodeAgent(
     description="An agent that fetches and summarizes news about any topic",
     prompt_templates=prompt_templates
 )
-
-class ResetGradioUI(GradioUI):
-    def interact_with_agent(self, prompt):
-        # Override the interact_with_agent method to always reset
-        return super().interact_with_agent(prompt, reset_agent_memory=True)
-
-# Launch the modified Gradio interface
-if __name__ == "__main__":
-    ui = ResetGradioUI(agent)
-    ui.launch(debug=True, share=True)  # Remove reset_agent_memory from launch parameters
+# Launch the Gradio interface
+if name == "main":
+    GradioUI(agent).launch()