Update tools/final_answer.py

tools/final_answer.py

@@ -1,54 +1,14 @@
-import os
-import datetime
-from typing import Optional, Union, Tuple
-from PIL import Image
-from smolagents import Tool
+from typing import Any, Optional
+from smolagents.tools import Tool
 
 class FinalAnswerTool(Tool):
     name = "final_answer"
-    description = "Provides a final answer. Can include text and/or a generated image."
+    description = "Provides a final answer to the given problem."
+    inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
+    output_type = "any"
 
-    inputs = {
-        'text': {'type': 'string', 'description': 'Final textual answer', 'nullable': True},
-        'image_path': {'type': 'string', 'description': 'Path to a generated image file', 'nullable': True}
-    }
+    def forward(self, answer: Any) -> Any:
+        return answer
 
-    output_type = "any"  # <-- can now return text, image, or both
-
-    def forward(
-        self, 
-        text: Optional[str] = None, 
-        image_path: Optional[str] = None
-    ) -> Union[str, Image.Image, Tuple[Image.Image, str]]:
-        
-        if image_path and os.path.exists(image_path):
-            # Convert .webp to .png (since some frontends don't support webp well)
-            if image_path.lower().endswith(".webp"):
-                try:
-                    img = Image.open(image_path).convert("RGB")
-                    temp_dir = os.path.dirname(image_path) or "/tmp"
-                    os.makedirs(temp_dir, exist_ok=True)
-
-                    base_name = os.path.basename(image_path)
-                    name_without_ext = os.path.splitext(base_name)[0]
-                    temp_png_path = os.path.join(
-                        temp_dir,
-                        f"{name_without_ext}_{datetime.datetime.now().strftime('%Y%m%d%H%M%S%f')}.png"
-                    )
-
-                    img.save(temp_png_path, "PNG")
-                    image_path = temp_png_path
-                except Exception as e:
-                    print(f"[ERROR] Failed to convert image {image_path}: {e}")
-
-            img = Image.open(image_path)
-
-            if text:
-                # Return tuple if both text + image
-                return (img, text)
-            else:
-                # Return only image
-                return img
-
-        # Only text case
-        return text
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False