fix: Add missing imports to resolve flake8 errors

text_visualization.py

@@ -1,3 +1,4 @@
+from typing import Dict, Any, Iterable
 from sklearn import TfIdfVectorizer
 import wordcloud
 from pydantic import BaseModel