exceptions, logging, refactoring, thumbnail, markdown

app.py

@@ -1,6 +1,23 @@
+import os
+
 import gradio
 import sign_language_translator as slt
 
+description = """Enter your text and select languages from the dropdowns, then click Submit to generate a video. [`Repository`](https://github.com/sign-language-translator/sign-language-translator)
+
+The text is preprocessed, tokenized and rearranged and then each token is mapped to a prerecorded video which are concatenated and returned. [`model code`](https://github.com/sign-language-translator/sign-language-translator/blob/main/sign_language_translator/models/text_to_sign/concatenative_synthesis.py)
+
+> NOTE: This model only supports a fixed vocabulary. See the [`*-dictionary-mapping.json`](https://github.com/sign-language-translator/sign-language-datasets/tree/main/parallel_texts) files for supported words."""
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+hf_writer = (
+    gradio.HuggingFaceDatasetSaver(
+        HF_TOKEN, "crowdsourced-text-to-sign-language-rule-based-translation-corpus"
+    )
+    if HF_TOKEN
+    else None
+)
+
 model = slt.models.ConcatenativeSynthesis("ur", "pk-sl", "video")
 
 
@@ -20,13 +37,13 @@ def text_to_video(
     # ToDo: video.watermark("Sign Language Translator\nAI Generated Video")
 
 
-def predict(text: str, text_language: str, sign_language: str):
+def predict(text: str, text_lang: str, sign_lang: str):
     try:
         path = "output.mp4"
-        text_to_video(text, text_language, sign_language, output_path=path, codec="mp4v")
+        text_to_video(text, text_lang, sign_lang, output_path=path, codec="mp4v")
         return path
     except Exception as exc:
-        return gradio.Error(f"Error during translation: {exc}")
+        raise gradio.Error(f"Error during translation: {exc}")
 
 
 gradio_app = gradio.Interface(
@@ -56,16 +73,17 @@ gradio_app = gradio.Interface(
         include_audio=False,
     ),
     title="Concatenative Synthesis: Rule Based Text to Sign Language Translator",
-    description="Enter your text and select options from the dropdowns, then click Submit to generate a video.\n\n"
-    "The text is preprocessed, tokenized and rearranged and then each token is mapped to a prerecorded video which are concatenated and returned.\n\n"
-    "> NOTE: This model only supports a fixed vocabulary. Check the `*-dictionary-mapping.json` on https://github.com/sign-language-translator/sign-language-datasets/tree/main/parallel_texts",
+    description=description,
     examples=[
         ["یہ بہت اچھا ہے۔", "ur", "pakistan-sign-language"],
         ["یہ کام بہت آسان ہے۔", "ur", "pakistan-sign-language"],
         ["पाँच घंटे।", "hi", "pakistan-sign-language"],
-        # ["आप कैसे हैं", "hi", "pakistan-sign-language"],
+        # ["आप कैसे हैं?", "hi", "pakistan-sign-language"],
     ],
     allow_flagging="auto",
+    flagging_callback=hf_writer,
+    thumbnail="https://cdn-uploads.huggingface.co/production/uploads/6368b375fbfe97c16a401079/1hUEuDUvqCZM0fLVhIAT1.png",
+    # cache_examples="lazy",
 )
 
 if __name__ == "__main__":