Removed old code, added support for any* document, disabled "upload" button when we are processing the document

gradio_app.py

@@ -1,30 +1,29 @@
 import gradio as gr
 from src.translate_any_doc import translate_document
 from src.salamandraTA7b_translator import SalamandraTA7bTranslator
-from src.mtuoc_aina_translator import MTUOCAinaTranslator
 from src.aligner import Aligner
-from nltk.tokenize.treebank import TreebankWordDetokenizer
 
 config_folder = 'fast_align_config'
 temp_folder = 'tmp'
 hf_token = ""
 
-detokenizer = TreebankWordDetokenizer()
 translator = SalamandraTA7bTranslator(hf_token)
 
 
-# ip = ''
-# port = ''
-# translator = MTUOCAinaTranslator(ip, port)
-
-
 def upload_file(filepath, source_lang, target_lang):
     aligner = Aligner(config_folder, source_lang, target_lang, temp_folder)
-    translated_file_name = translate_document(filepath, source_lang, target_lang, translator, aligner, detokenizer)
+    translated_file_name = translate_document(filepath, source_lang, target_lang, translator, aligner)
     return [gr.UploadButton(visible=False),
             gr.DownloadButton(label=f"Download {translated_file_name}", value=translated_file_name, visible=True)]
 
 
+def before_processing():
+    return [
+        gr.UploadButton("Processing...", interactive=False),
+        gr.DownloadButton(visible=False)  # Keep download hidden until processing finishes
+    ]
+
+
 def download_file():
     return [gr.UploadButton(visible=True), gr.DownloadButton(visible=False)]
 
@@ -32,7 +31,7 @@ def download_file():
 with gr.Blocks() as demo:
     with gr.Tab("Text"):
         gr.Interface(fn=translator.translate, inputs=["text", "text", "text"], outputs="text")
-    with gr.Tab("Docx documents"):
+    with gr.Tab("Documents"):
         with gr.Row():
             dropdown1 = gr.Dropdown(label="Source language", choices=["en", "ca"], value=None,
                                     interactive=True)
@@ -42,7 +41,7 @@ with gr.Blocks() as demo:
             u = gr.UploadButton("Upload a file", file_count="single")
             d = gr.DownloadButton("Download the file", visible=False)
 
-        u.upload(upload_file, [u, dropdown1, dropdown2], [u, d])
+        u.upload(fn=before_processing, inputs=None, outputs=[u, d]).then(upload_file, [u, dropdown1, dropdown2], [u, d])
         d.click(download_file, None, [u, d])
 if __name__ == "__main__":
     demo.launch()