Update app.py

app.py

@@ -7,16 +7,34 @@ from tools.final_answer import FinalAnswerTool
 
 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
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def generate_key_results(objective: str) -> list[str]:
+    """
+    Queries the Qwen2.5-Coder-32B-Instruct model to generate key results for a given objective.
+
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        objective (str): The objective for which key results should be generated.
+
+    Returns:
+        list[str]: A list of key results generated by the model.
     """
-    return "What magic will you build ?"
+    url = "https://api-inference.huggingface.co/models/Qwen/Qwen2.5-Coder-32B-Instruct"
+    headers = {"Authorization": f"Bearer {HUGGINGFACE_API_KEY}"}
+    payload = {"inputs": f"Objective: {objective}\nGenerate three key results for this objective."}
+
+    response = requests.post(url, headers=headers, json=payload)
+
+    if response.status_code != 200:
+        raise Exception(f"API request failed: {response.json()}")
+
+    output_text = response.json()[0]["generated_text"]
+
+    key_results = output_text.split("\n")
+    return [kr.strip("- ") for kr in key_results if kr]
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: