Update app.py

Adding a new tool to check / verify news

app.py

@@ -33,6 +33,24 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def classify_news_article(article_text: str) -> str:
+    """
+    Classifies a news article as 'Real' or 'Fake' based on its content.
+    
+    Args:
+        article_text (str): The content of the news article to be classified.
+    
+    Returns:
+        str: 'Real' if the article is classified as real news, 'Fake' otherwise.
+    """
+    # Perform classification
+    model_name = "Pulk17/Fake-News-Detection"
+    classifier = pipeline("text-classification", model=model_name)
+    result = classifier(article_text[:500])[0]  # Truncate to 500 words as per model's requirement
+    label = result['label']  # 'LABEL_0' for fake news, 'LABEL_1' for real news
+    return "Real" if label == 'LABEL_1' else "Fake"   
+
 
 final_answer = FinalAnswerTool()
 
@@ -55,7 +73,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[image_generation_tool,classify_news_article], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,