Add domain parameter to GettextWrapper and pass languages to translat…

…ion method explicitely

streamlit_gettext/streamlit_gettext.py

@@ -3,9 +3,13 @@
 import gettext as _gettext_module
 import re
 from pathlib import Path
+from typing import TYPE_CHECKING
 
 import streamlit as st
 
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+
 
 def get_preferred_languages() -> list[str]:
     accept_language = st.context.headers.get("Accept-Language") or ""
@@ -16,24 +20,25 @@ class GettextWrapper:
     """
     A wrapper for the gettext module
     """
-
     locale_path: Path
+    domain: str
 
-    def __init__(self, locale_path: str | Path) -> None:
+    def __init__(self, locale_path: str | Path, domain: str = "messages") -> None:
         self.locale_path = Path(locale_path)
+        self.domain = domain
 
-    def _translation(self) -> _gettext_module.NullTranslations:
+    def translation(self, languages: Iterable[str] | None = None) -> _gettext_module.NullTranslations:
         return _gettext_module.translation(
-            "messages",
+            domain=self.domain,
             localedir=self.locale_path,
-            languages=get_preferred_languages(),
+            languages=languages,
             fallback=True,
         )
 
     def gettext(self, message: str) -> str:
-        translation = self._translation()
+        translation = self.translation(get_preferred_languages())
         return translation.gettext(message)
 
     def ngettext(self, singular: str, plural: str, n: int) -> str:
-        translation = self._translation()
+        translation = self.translation(get_preferred_languages())
         return translation.ngettext(singular, plural, n)