Update app.py

app.py

@@ -10,7 +10,7 @@ from Gradio_UI import GradioUI
 
 import json
 @tool
-def text_to_json(text: str, json_schema: dict) -> dict:
+def text_to_json(text: str, json_schema: dict) -> str:
     """
     Uses an LLM to transform the information of plain text into structured data using a JSON schema.
     
@@ -19,7 +19,7 @@ def text_to_json(text: str, json_schema: dict) -> dict:
         json_schema (dict): The JSON schema that defines the structure of the valid JSON.
     
     Returns:
-        dict: A valid JSON object as a Python dictionary.
+        str: A valid JSON string.
     
     Example:
         >>> text = "John Doe, 30 years old, lives in New York and works as a software engineer."
@@ -47,9 +47,8 @@ def text_to_json(text: str, json_schema: dict) -> dict:
         {"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},
     ]
-    data= model(messages, response_format={"type": "json_object", "schema": json_schema}).content
-    print(data)
-    return json.loads(data)
+    data = model(messages, response_format={"type": "json_object", "schema": json_schema}).content
+    return data
 
 
 @tool