Update app.py

app.py

@@ -59,10 +59,7 @@ class CodeGenerationTool(Tool):
         # This is a simplified example, real implementation would use a code generation model
         language = arguments.get("language", "python")
         code = arguments.get("code", "print('Hello, World!')")
-        return {"output": f"
-
-
-{language}\n{code}\n
+        return {"output": f"""{language}{code}"""
 
 class DataRetrievalTool(Tool):
     def __init__(self):