Update app.py

Added image generation tool and logs of text_to_json

app.py

@@ -47,7 +47,9 @@ 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},
     ]
-    return json.loads(model(messages, response_format={"type": "json_object", "schema": json_schema}).content)
+    data= model(messages, response_format={"type": "json_object", "schema": json_schema}).content
+    print(data)
+    return json.loads(data)
 
 
 @tool
@@ -87,7 +89,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,text_to_json], ## add your tools here (don't remove final answer)
+    tools=[final_answer,text_to_json,image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,