Update app.py

Added text_to_json tool

app.py

@@ -3,20 +3,53 @@ import datetime
 import requests
 import pytz
 import yaml
+import json
 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 !
+import json
+
 @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 text_to_json(text: str, json_schema: dict) -> dict:
+    """
+    Uses an LLM to transform the information of plain text into structured data using a JSON schema.
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        text (str): The desired text to be transformed into JSON.
+        json_schema (dict): The JSON schema that defines the structure of the valid JSON.
+    
+    Returns:
+        dict: A valid JSON object as a Python dictionary.
+    
+    Example:
+        >>> text = "John Doe, 30 years old, lives in New York and works as a software engineer."
+        >>> json_schema = {
+        ...     "type": "object",
+        ...     "properties": {
+        ...         "name": {"type": "string"},
+        ...         "age": {"type": "integer"},
+        ...         "city": {"type": "string"},
+        ...         "occupation": {"type": "string"}
+        ...     },
+        ...     "required": ["name", "age", "city", "occupation"]
+        ... }
+        >>> result = text_to_json(text, json_schema)
+        >>> print(result)
+        {'name': 'John Doe', 'age': 30, 'city': 'New York', 'occupation': 'software engineer'}
     """
-    return "What magic will you build ?"
+    model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+        custom_role_conversions=None,
+    )
+    messages = [
+        {"role": "system", "content": "You are an assistant that converts text into JSON. Convert the provided text into a JSON object based on the provided schema. JSON_SCHEMA:\n```json " + json.dumps(json_schema) + "```"},
+        {"role": "user", "content": text},
+    ]
+    return json.loads(model(messages, response_format={"type": "json_object", "schema": json_schema}).content)
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: