Update app.py

app.py

@@ -105,12 +105,15 @@ class Translator:
 lan = Translator()
 def detect_onnx_models(path):
     onnx_models = glob.glob(path + '/*.onnx')
+    onnx_configs = glob.glob(path + '/*.json')
     if len(onnx_models) > 1:
-        return onnx_models
+        return onnx_models, onnx_configs
     elif len(onnx_models) == 1:
-        return onnx_models[0]
+        return onnx_models[0], onnx_configs[0]
     else:
         return None
+
+
 renamed_audio_file = None
 
 model_configurations = {}
@@ -130,10 +133,7 @@ async def load_model_data():
 
     # Collect data for all models in the directory and populate model_configurations
     onnx_models = detect_onnx_models(models_path)
-    for model_name in onnx_models:
-        # Load the configuration data for each model (including speaker_id_map)
-        config = load_model_configuration(model_name)
-        model_configurations[model_name] = config
+
         
 #@app.post("/synthesize")
 #@app.post("/", response_class=FileResponse)
@@ -171,7 +171,7 @@ async def main(
 #        layout={'visibility': 'hidden'}
 #    )
         # Handle loading the selected model and speaker ID here
-    if selected_model in model_configurations:
+    if selected_model in onnx_models:
         config = model_configurations[selected_model]
         onnx_model = config["onnx_model"]  # Replace with the actual key for your ONNX model file
         model, config = load_onnx(onnx_model, sess_options, providers)
@@ -184,7 +184,7 @@ async def main(
         document.getElementById("synthesize_button").disabled = true;
         </script>
         """
-        onnx_model = onnx_models
+        onnx_model = selected_model
         model, config = load_onnx(onnx_model, sess_options, providers)
 
 
@@ -302,7 +302,12 @@ async def main(
     
 
         # Save the audio as a temporary WAV file
-    return templates.TemplateResponse("interface.html", {"request": request, "file_url": file_url, "text_input": text_input, "data": data, "voices": voices, "dynamic_content": response_html})
+    return templates.TemplateResponse("interface.html", {"request": request, "file_url": file_url, "text_input": text_input, "data": data, "model_names": model_configurations.keys(),  # Pass the list of available model names
+        "selected_model": selected_model,  # Pass the selected model
+        "speaker_id_map": speaker_id_map,  # Pass the speaker ID map for the selected model
+        "selected_speaker_id": selected_speaker_id,  # Pass the selected speaker ID
+    #    "model": model,  # Pass the loaded model (if any) 
+        "voices": voices, "dynamic_content": response_html})
    
     # Serve the audio file with the correct media type
  #   return FileResponse(renamed_audio_file)