Update app.py

app.py

@@ -7,8 +7,7 @@ from huggingface_hub import list_repo_files
 import uuid
 import re 
 import gradio as gr
-
-
+key = os.getenv("SECRET_KEY", None)
 #translate langauge 
 from deep_translator import GoogleTranslator
 def bulk_translate(text, target_language, chunk_size=500):
@@ -463,26 +462,29 @@ def save_current_data():
     os.makedirs("./last",exist_ok=True)
     
      
-def KOKORO_TTS_API(text, Language="American English",voice="af_bella", speed=1,translate_text=False,remove_silence=False,keep_silence_up_to=0.05):
-    if translate_text:    
-        text=bulk_translate(text, Language, chunk_size=500)
-    save_path,timestamps=generate_and_save_audio(text=text, Language=Language,voice=voice, speed=speed,remove_silence=remove_silence,keep_silence_up_to=keep_silence_up_to)
-    if remove_silence==False:
-        if Language in ["American English", "British English"]:
-            word_level_timestamps=adjust_timestamps(timestamps)
-            word_level_srt = modify_filename(save_path.replace(".wav", ".srt"), prefix="word_level_")
-            normal_srt = modify_filename(save_path.replace(".wav", ".srt"), prefix="sentence_")
-            json_file = modify_filename(save_path.replace(".wav", ".json"), prefix="duration_")
-            write_word_srt(word_level_timestamps, output_file=word_level_srt, skip_punctuation=True)
-            write_sentence_srt(word_level_timestamps, output_file=normal_srt, min_pause=0.01)
-            make_json(word_level_timestamps, json_file)
-            save_current_data()
-            shutil.copy(save_path, "./last/")
-            shutil.copy(word_level_srt, "./last/")
-            shutil.copy(normal_srt, "./last/")
-            shutil.copy(json_file, "./last/")
-            return save_path,save_path,word_level_srt,normal_srt,json_file
-    return save_path,save_path,None,None,None    
+def KOKORO_TTS_API(text, Language="American English",voice="af_bella", speed=1,translate_text=False,remove_silence=False,keep_silence_up_to=0.05, input_key=None):
+    if input_key == key:
+        if translate_text:    
+            text=bulk_translate(text, Language, chunk_size=500)
+        save_path,timestamps=generate_and_save_audio(text=text, Language=Language,voice=voice, speed=speed,remove_silence=remove_silence,keep_silence_up_to=keep_silence_up_to)
+        if remove_silence==False:
+            if Language in ["American English", "British English"]:
+                word_level_timestamps=adjust_timestamps(timestamps)
+                word_level_srt = modify_filename(save_path.replace(".wav", ".srt"), prefix="word_level_")
+                normal_srt = modify_filename(save_path.replace(".wav", ".srt"), prefix="sentence_")
+                json_file = modify_filename(save_path.replace(".wav", ".json"), prefix="duration_")
+                write_word_srt(word_level_timestamps, output_file=word_level_srt, skip_punctuation=True)
+                write_sentence_srt(word_level_timestamps, output_file=normal_srt, min_pause=0.01)
+                make_json(word_level_timestamps, json_file)
+                save_current_data()
+                shutil.copy(save_path, "./last/")
+                shutil.copy(word_level_srt, "./last/")
+                shutil.copy(normal_srt, "./last/")
+                shutil.copy(json_file, "./last/")
+                return save_path,save_path,word_level_srt,normal_srt,json_file
+        return save_path,save_path,None,None,None    
+    else:
+        return None,None,None,None,None
     
     
 
@@ -514,6 +516,7 @@ def ui():
         with gr.Row():
             with gr.Column():
                 text = gr.Textbox(label='📝 Enter Text', lines=3)
+                input_key = gr.Textbox(label='Input Key', lines=1)
                 
                 with gr.Row():
                     language_name = gr.Dropdown(lang_list, label="🌍 Select Language", value=lang_list[0])
@@ -542,8 +545,8 @@ def ui():
                     srt_file = gr.File(label='📜 Download Sentence-Level SRT')
                     sentence_duration_file = gr.File(label='⏳ Download Sentence Timestamp JSON')
 
-        text.submit(KOKORO_TTS_API, inputs=[text, language_name, voice_name, speed,translate_text, remove_silence], outputs=[audio, audio_file,word_level_srt_file,srt_file,sentence_duration_file])
-        generate_btn.click(KOKORO_TTS_API, inputs=[text, language_name, voice_name, speed,translate_text, remove_silence], outputs=[audio, audio_file,word_level_srt_file,srt_file,sentence_duration_file])
+        text.submit(KOKORO_TTS_API, inputs=[text, language_name, voice_name, speed,translate_text, remove_silence,input_key], outputs=[audio, audio_file,word_level_srt_file,srt_file,sentence_duration_file])
+        generate_btn.click(KOKORO_TTS_API, inputs=[text, language_name, voice_name, speed,translate_text, remove_silence,input_key], outputs=[audio, audio_file,word_level_srt_file,srt_file,sentence_duration_file])
 
         # Add examples to the interface
         gr.Examples(examples=dummy_examples, inputs=[text, language_name, voice_name])