Update app.py

app.py

@@ -17,6 +17,7 @@ from vinorm import TTSnorm
 # Initialize Hugging Face API
 HF_TOKEN = os.environ.get("HF_TOKEN")
 api = HfApi(token=HF_TOKEN)
+PASSWORD = os.environ.get("KEY")
 
 # Download model files if not already downloaded
 print("Downloading viXTTS model files if not already present...")
@@ -51,6 +52,13 @@ MODEL.load_checkpoint(
 if torch.cuda.is_available():
     MODEL.cuda()
 
+def authenticate(password):
+    if password == PASSWORD:
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+    else:
+        return gr.update(visible=False), gr.update(visible=True, value="Invalid password"), gr.update(visible=True)
+
+
 # Supported languages
 supported_languages = config.languages
 if "vi" not in supported_languages:
@@ -194,5 +202,20 @@ with gr.Blocks(analytics_enabled=False) as demo:
         api_name="predict",
     )
 
+  login_btn.click(
+        fn=authenticate,
+        inputs=password_input,
+        outputs=[main_column, error_message, login_column]
+    )
+
+    submit_btn.click(
+        fn=pipe,
+        inputs=[text, voice, image_in],
+        outputs=[video_o],
+        concurrency_limit=3
+    )
+
+demo.queue(max_size=10).launch(show_error=True, show_api=False)
+
 demo.queue()
 demo.launch(debug=True, show_api=True)