Update app.py

app.py

@@ -9,14 +9,27 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def get_news_headlines()-> 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
+    """A dummy tool that imitates fetching of the top 3 headlines from the news aggregator omni.se 
+    """
+
+    dummy = """
+    Trump hävdar felaktigt att Ukraina startade kriget
+USA:s president Donald Trump påstår felaktigt att det var Ukraina som startade kriget med Ryssland. Angående att Kyiv inte haft en plats vid förhandlingsbordet i Saudiarabien säger han att Ukraina ”borde avslutat” det för tre år sedan.
+– Ni borde aldrig ha startat det. Ni kunde ha slutit ett avtal, säger Trump vid en pressträff på Mar-a-Lago.
+
+Tidöpartierna vill inte ta bort ordet ”kvinna” ur abortlag
+Tidöpartierna är kritiska till förslaget att ta bort ordet ”kvinna” ur den svenska abortlagstiftningen och i stället ersätta det med ”den som är gravid”.
+– Det är kvinnor som blir gravid och det är kvinnor som föder barn. Det är den biologiska grunden, så enkelt är det, säger Linda Lindberg, gruppledare i SD, till Ekot.
+Förslaget kommer eftersom även den som har juridiskt kön ”man” kan bli gravid, och då måste ha möjlighet att göra abort.
+
+Asap Rocky frias i rätten: ”Räddade mitt liv”
+Rapartisten Asap Rocky frias av juryn. Rakim Mayers, som han egentligen heter, stod åtalad för att ha skjutit mot sin barndomskompis och riskerade upp till 24 års fängelse vid en fällande dom.
+– Tack för att ni räddade mitt liv, sa han till juryn.
     """
-    return "What magic will you build ?"
+    
+    return dummy
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +68,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=[get_news_headlines, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,