Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 import pytz
 import yaml
 import os
+import torch
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -12,11 +13,22 @@ from Gradio_UI import GradioUI
 rapid_api_key=os.environ["RAPID_KEY"]
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
+#@tool
+#def reflecting_on_results(arg1: str) -> str:
+#    """A tool that receives the football teams comparison and then it reflects on the results with an insightful comment, it could be funny or witty
+#    Args:
+#        arg1: The first argument receives a string with the comparison of two teams as queried by user
+#    """
+#    insights
+#    return insights
+
+
+
 @tool
 def get_football_results(arg1: str, arg2: str) -> 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 connect to an API that provides live football data and compares the stats of two teams in the English Premier League, then it returns a text response
-    where you should look for those football teams given, compare stats and provide a reflection on how both stack up.
+    where you should look for those football teams given, returns a comparison of stats including a funny reflection on the comparison.
     Args:
         arg1: The first argument provides one of the football teams to check
         arg2: The second football team to check
@@ -37,7 +49,7 @@ def get_football_results(arg1: str, arg2: str) -> str:  # it's import to specify
     messages= [
         {
             "role": "system",
-            "content": f"You a friendly AI system embedded in a wider AI agent. You are going to faciliate the work of other LLMs. You have received from user an API response in JSON format. In the response item 28 there is a long text starting with a text like 'status:success,response...You are going to extract the team names from the response and compare the stats of the two teams. First team check is {arg1} and second team is {arg2}. Provide a funny reflection on your analysis of how these two teams stack up against each other. You should provide a text response"
+            "content": f"You are a friendly AI system embedded in a wider AI agent. You are going to faciliate the work of other LLMs. You have received from user an API response in JSON format. In the response item 28 there is a long text starting with a text like 'status:success,response...You are going to extract the team names from the response and compare the stats of the two teams. First team check is {arg1} and second team is {arg2}. Provide a funny reflection on your analysis of how these two teams stack up against each other. You should provide a text response"
         },
         {
             "role": "user",
@@ -45,9 +57,9 @@ def get_football_results(arg1: str, arg2: str) -> str:  # it's import to specify
         }
     ]
     prompt = tiny_llama.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-    insights = tiny_llama(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
+    comparison = tiny_llama(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
     
-    return insights[0]["generated_text"]
+    return comparison[0]["generated_text"]
       
 
 @tool
@@ -88,7 +100,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_football_results], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, get_football_results, reflecting_on_results] ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,