Update tools/final_answer.py

tools/final_answer.py

@@ -8,7 +8,16 @@ class FinalAnswerTool(Tool):
     output_type = "any"
 
     def forward(self, answer: Any) -> Any:
-        return answer
+        from PIL import Image
+        import io
+
+        # If it's a PIL Image, convert to bytes so Gradio can render it
+        if isinstance(answer, Image.Image):
+            buf = io.BytesIO()
+            answer.save(buf, format="PNG")
+            buf.seek(0)
+            return buf.getvalue()
 
+        return answer
     def __init__(self, *args, **kwargs):
         self.is_initialized = False