Update app.py

app.py

@@ -248,37 +248,33 @@ def get_my_model(url_data, progress=gr.Progress(track_tqdm=True)):
         t.start()
 
 
-# ========== नया फ़ंक्शन: logs फोल्डर से मॉडल्स की सूची बनाना ==========
+# ========== नया फ़ंक्शन: logs/ फोल्डर से सभी मॉडल स्कैन करें ==========
 def scan_models():
-    """logs/ फोल्डर के अंदर मौजूद हर सबफोल्डर को एक मॉडल मानकर उसकी .pth और .index फाइल ढूंढें"""
+    """logs/ फोल्डर के अंदर हर सबफोल्डर को एक मॉडल मानें और उसकी .pth व .index फाइलें ढूंढें।"""
     logs_dir = "logs"
     if not os.path.isdir(logs_dir):
         return []
-    
+
     models = []
     for model_name in os.listdir(logs_dir):
         model_path = os.path.join(logs_dir, model_name)
         if not os.path.isdir(model_path):
             continue
-        # सबफोल्डर में .pth और .index फाइल देखें (नाम फोल्डर जैसा होना चाहिए)
-        pth_file = os.path.join(model_path, f"{model_name}.pth")
-        index_file = os.path.join(model_path, f"{model_name}.index")
-        # अगर exact नाम न मिले तो कोई भी .pth और .index उठा लें (fallback)
-        if not os.path.isfile(pth_file) or not os.path.isfile(index_file):
-            # Fallback: फोल्डर में मौजूद पहली .pth और .index फाइल लें
-            pth_candidates = [f for f in os.listdir(model_path) if f.endswith(".pth")]
-            index_candidates = [f for f in os.listdir(model_path) if f.endswith(".index")]
-            if pth_candidates and index_candidates:
-                pth_file = os.path.join(model_path, pth_candidates[0])
-                index_file = os.path.join(model_path, index_candidates[0])
-            else:
-                continue  # इस फोल्डर में जरूरी फाइलें नहीं हैं
-        models.append((model_name, pth_file, index_file))
+
+        # फोल्डर के अंदर .pth और .index फाइलें देखें
+        pth_files = [f for f in os.listdir(model_path) if f.endswith(".pth")]
+        index_files = [f for f in os.listdir(model_path) if f.endswith(".index")]
+
+        if pth_files and index_files:
+            # पहली मिलने वाली फाइल ले लें (या आप नाम से मिलान कर सकते हैं)
+            pth_path = os.path.join(model_path, pth_files[0])
+            index_path = os.path.join(model_path, index_files[0])
+            models.append((model_name, pth_path, index_path))
     return models
 
 
-def update_model_selection(model_name):
-    """ड्रॉपडाउन से चुने गए मॉडल के अनुसार pth और index का पथ लौटाएँ"""
+def update_model_paths(model_name):
+    """चुने गए मॉडल के अनुसार pth और index के पूरे पथ लौटाएँ।"""
     models = scan_models()
     for name, pth, idx in models:
         if name == model_name:
@@ -321,14 +317,14 @@ def add_audio_effects(audio_list, type_output):
             result.append(output_path)
         except Exception as e:
             traceback.print_exc()
-            print(f"Error noisereduce: {str(e)}")
+            print(f"Error audio effects: {str(e)}")
             result.append(audio_path)
 
     return result
 
 
 def apply_noisereduce(audio_list, type_output):
-    print("Noice reduce")
+    print("Noise reduce")
 
     result = []
     for audio_path in audio_list:
@@ -336,14 +332,12 @@ def apply_noisereduce(audio_list, type_output):
 
         try:
             audio = AudioSegment.from_file(audio_path)
-
             samples = np.array(audio.get_array_of_samples())
-
             reduced_noise = nr.reduce_noise(samples, sr=audio.frame_rate, prop_decrease=0.6)
 
             reduced_audio = AudioSegment(
-                reduced_noise.tobytes(), 
-                frame_rate=audio.frame_rate, 
+                reduced_noise.tobytes(),
+                frame_rate=audio.frame_rate,
                 sample_width=audio.sample_width,
                 channels=audio.channels
             )
@@ -369,16 +363,15 @@ def convert_now(audio_files, random_tag, converter, type_output, steps):
             parallel_workers=(2 if IS_COLAB else 8),
             type_output=type_output,
         )
-
     return audio_files
 
 
 def run(
     audio_files,
-    file_m,                # अब यह hidden टेक्स्टबॉक्स से आएगा (pth का पथ)
+    file_m,
     pitch_alg,
     pitch_lvl,
-    file_index,            # hidden टेक्स्टबॉक्स से index का पथ
+    file_index,
     index_inf,
     r_m_f,
     e_r,
@@ -389,7 +382,7 @@ def run(
     steps,
 ):
     if not audio_files:
-        raise ValueError("The audio pls")
+        raise ValueError("Please provide audio files")
 
     if isinstance(audio_files, str):
         audio_files = [audio_files]
@@ -400,12 +393,11 @@ def run(
     except Exception as e:
         print(e)
 
-    # अगर file_m टेक्स्ट फाइल है तो पुराने लॉजिक से हैंडल करें
     if file_m is not None and file_m.endswith(".txt"):
         file_m, file_index = find_my_model(file_m, file_index)
         print(file_m, file_index)
 
-    random_tag = "USER_"+str(random.randint(10000000, 99999999))
+    random_tag = "USER_" + str(random.randint(10000000, 99999999))
 
     converter.apply_conf(
         tag=random_tag,
@@ -432,7 +424,7 @@ def run(
     return result
 
 
-# ========== UI कॉन्फ़िगरेशन (बदलाव सहित) ==========
+# ========== UI कॉन्फ़िगरेशन ==========
 
 def audio_conf():
     return gr.File(
@@ -444,21 +436,20 @@ def audio_conf():
 
 
 def model_dropdown_conf():
-    """मॉडल चुनने के लिए ड्रॉपडाउन बनाएँ"""
     models = scan_models()
     choices = [name for name, _, _ in models]
     return gr.Dropdown(
+        label="Select Model",
         choices=choices,
         value=choices[0] if choices else None,
-        label="Select Model",
         interactive=True,
     )
 
 
-# ये दो छुपे हुए टेक्स्टबॉक्स हैं जिनमें असली फाइल पथ जाएँगे
 def hidden_model_path_conf():
     return gr.Textbox(visible=False)
 
+
 def hidden_index_path_conf():
     return gr.Textbox(visible=False)
 
@@ -551,9 +542,9 @@ def active_tts_conf():
 def tts_voice_conf():
     return gr.Dropdown(
         label="tts voice",
-        choices=voices,
+        choices=[],
         visible=False,
-        value="en-US-EmmaMultilingualNeural-Female",
+        value=None,
     )
 
 
@@ -613,6 +604,7 @@ def format_output_gui():
         value="wav",
     )
 
+
 def denoise_conf():
     return gr.Checkbox(
         False,
@@ -633,7 +625,7 @@ def effects_conf():
 
 def infer_tts_audio(tts_voice, tts_text, play_tts):
     out_dir = "output"
-    folder_tts = "USER_"+str(random.randint(10000, 99999))
+    folder_tts = "USER_" + str(random.randint(10000, 99999))
 
     os.makedirs(out_dir, exist_ok=True)
     os.makedirs(os.path.join(out_dir, folder_tts), exist_ok=True)
@@ -646,14 +638,11 @@ def infer_tts_audio(tts_voice, tts_text, play_tts):
 
 
 def show_components_tts(value_active):
-    return gr.update(
-        visible=value_active
-    ), gr.update(
-        visible=value_active
-    ), gr.update(
-        visible=value_active
-    ), gr.update(
-        visible=value_active
+    return (
+        gr.update(visible=value_active),
+        gr.update(visible=value_active),
+        gr.update(visible=value_active),
+        gr.update(visible=value_active),
     )
 
 
@@ -684,12 +673,10 @@ def down_button_conf():
 
 
 def show_components_down(value_active):
-    return gr.update(
-        visible=value_active
-    ), gr.update(
-        visible=value_active
-    ), gr.update(
-        visible=value_active
+    return (
+        gr.update(visible=value_active),
+        gr.update(visible=value_active),
+        gr.update(visible=value_active),
     )
 
 
@@ -703,12 +690,13 @@ CSS = """
 }
 """
 
+
 def get_gui(theme):
     with gr.Blocks(theme=theme, css=CSS, fill_width=True, fill_height=False, delete_cache=delete_cache_time) as app:
         gr.Markdown(title)
         gr.Markdown(description)
 
-        # ===== TTS सेक्शन (कोई बदलाव नहीं) =====
+        # ===== TTS सेक्शन =====
         active_tts = active_tts_conf()
         with gr.Row():
             with gr.Column(scale=1):
@@ -737,7 +725,28 @@ def get_gui(theme):
             outputs=[aud, tts_play],
         )
 
-        # ===== URL-to-Model सेक्शन (कोई बदलाव नहीं) =====
+        # ===== URL-to-Model सेक्शन =====
         down_active_gui = down_active_conf()
         down_info = gr.Markdown(
-            f"Provide a link to a zip file, like this one: `https://huggingface.co/MrDawg/ToothBrushing/resolve/main/ToothBrushing.zip?download=true`, or 
+            f"Provide a link to a zip file, like this one: `https://huggingface.co/MrDawg/ToothBrushing/resolve/main/ToothBrushing.zip?download=true`, or separate links with a comma for the .pth and .index files, like this: `{test_model}`",
+            visible=False
+        )
+        with gr.Row():
+            with gr.Column(scale=3):
+                down_url_gui = down_url_conf()
+            with gr.Column(scale=1):
+                down_button_gui = down_button_conf()
+
+        # ये हैं असली मॉडल पथ वाले छुप��� हुए टेक्स्टबॉक्स
+        hidden_model_path = hidden_model_path_conf()
+        hidden_index_path = hidden_index_path_conf()
+
+        down_active_gui.change(
+            show_components_down,
+            [down_active_gui],
+            [down_info, down_url_gui, down_button_gui]
+        )
+
+        # जब URL से मॉडल डाउनलोड हो, तो उसके पथ छुपे हुए टेक्स्टबॉक्स में डालें
+        def update_from_url(url_data):
+            model_path, index_path = get_my_model