test log in new

app.py

@@ -54,6 +54,10 @@ except (ValueError, TypeError, IndexError):
     DoFormant, Quefrency, Timbre = False, 1.0, 1.0
     CSVutil('csvdb/formanting.csv', 'w+', 'formanting', DoFormant, Quefrency, Timbre)
 
+def update_message(request: gr.Request):
+    change_choices(request.username)
+    return f"Welcome, {request.username}"
+
 def download_models():
     # Download hubert base model if not present
     if not os.path.isfile('./hubert_base.pt'):
@@ -237,6 +241,7 @@ hubert_model = None
 voicecraft_model = None
 voicecraft_config = None
 phn2num = None
+associated_links = {}
 
 def load_hubert():
     global hubert_model
@@ -518,16 +523,25 @@ def get_vc(sid):
     return {"visible": False, "maximum": n_spk, "__type__": "update"}
 
 
-def change_choices():
+def change_choices(username=None):
     names = []
+    print(associated_links)
     for name in os.listdir(weight_root):
         if name.endswith(".pth"):
-            names.append(name)
+            if username is None:
+                names.append(name)
+            else:
+                if associated_links.get(name) == username:
+                    names.append(name)
     index_paths = []
     for root, dirs, files in os.walk(index_root, topdown=False):
         for name in files:
             if name.endswith(".index") and "trained" not in name:
-                index_paths.append("%s/%s" % (root, name))
+                if username is None:
+                    index_paths.append("%s/%s" % (root, name))
+                else:
+                    if associated_links.get(name) == username:
+                        index_paths.append("%s/%s" % (root, name))
     return {"choices": sorted(names), "__type__": "update"}, {
         "choices": sorted(index_paths),
         "__type__": "update",
@@ -1366,7 +1380,7 @@ def check_for_name():
     else:
         return ''
             
-def download_from_url(url, model):
+def download_from_url(url, model, associated_user=None):
     if url == '':
         return "URL cannot be left empty."
     if model =='':
@@ -1400,8 +1414,11 @@ def download_from_url(url, model):
                 if file.endswith(".index"):
                     os.mkdir(f'./logs/{model}')
                     shutil.copy2(file_path,f'./logs/{model}')
+                    associated_links[file] = associated_user
                 elif "G_" not in file and "D_" not in file and file.endswith(".pth"):
                     shutil.copy(file_path,f'./weights/{model}.pth')
+                    if associated_user is not None:
+                        associated_links[f'{model}.pth'] = associated_user
         shutil.rmtree("zips")
         shutil.rmtree("unzips")
         change_choices()
@@ -1569,7 +1586,7 @@ def run_joint(seed, stop_repetition, sample_batch_size, left_margin, right_margi
     os.environ["USER"] = "USER"
     # take a look at demo/temp/mfa_alignment, decide which part of the audio to use as prompt
     cut_off_sec = cutoff_value  # NOTE: according to forced-alignment file, the word "common" stop as 3.01 sec, this should be different for different audio
-    target_transcript = transcribed_text + target_transcript
+    target_transcript = transcribed_text + ' ' + target_transcript
     print(target_transcript)
     info = torchaudio.info(audio_fn)
     audio_dur = info.num_frames / info.sample_rate
@@ -1705,6 +1722,7 @@ def zip_downloader(model):
 download_from_url('https://drive.google.com/uc?id=1O98vvnle_nZP8ZdpnZFLZ5TU1UZe7x0p&confirm=t', 'JVKE-main')
 download_from_url('https://drive.google.com/uc?id=1Wag0vPlp42kRDffccXljjjlK7QsHf2xe&confirm=t', 'JVKE-main-v2')
 download_from_url('https://drive.google.com/uc?id=1h810cil3YRlN4pu4oO43zKq9z3cYjItp&confirm=t', 'jvke-nighttime-v4')
+download_from_url('https://drive.google.com/uc?id=1fa6FSLwqSQMI49NvSXOpI4pUVuKsrop5&confirm=t', 'Andoni')
 weight_root = "weights"
 index_root = "logs"
 names = []
@@ -1721,17 +1739,7 @@ for root, dirs, files in os.walk(index_root, topdown=False):
 with gr.Blocks(theme=gr.themes.Default(primary_hue="pink", secondary_hue="rose"), title="HITGEN AI") as app:
     with gr.Tabs():
         with gr.TabItem("Inference"):
-            # gr.HTML("<h1>  Ilaria RVC 💖   </h1>")     
-            # gr.HTML("<h10>   You can find voice models on AI Hub: https://discord.gg/aihub   </h10>")   
-            # gr.HTML("<h4>  Huggingface port by Ilaria of the Rejekt Easy GUI </h4>")
-
-            # Inference Preset Row
-            # with gr.Row():
-            #     mangio_preset = gr.Dropdown(label="Inference Preset", choices=sorted(get_presets()))
-            #     mangio_preset_name_save = gr.Textbox(
-            #         label="Your preset name"
-            #     )
-            #     mangio_preset_save_btn = gr.Button('Save Preset', variant="primary")
+            app.load(update_message)
 
             # Other RVC stuff
             with gr.Row():
@@ -2399,5 +2407,5 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="pink", secondary_hue="rose")
                 "-------------------------------\n"
             )
                 
-    app.queue(concurrency_count=511, max_size=1022).launch(share=False, quiet=False, auth=('jvke', 'thisfeelslikeai'))
+    app.queue(concurrency_count=511, max_size=1022).launch(share=False, quiet=False, auth=[('jvke', 'thisfeelslikeai'), ('cmss60', 'yourseedislate')])
 #endregion