Update app.py

app.py

@@ -13,12 +13,12 @@ enhanced_accessibility = False #@param {type:"boolean"}
 #@markdown ---
 use_gpu = False #@param {type:"boolean"}
 
-from fastapi import FastAPI, Request, Form
+from fastapi import FastAPI, Request, Form, Depends
 from fastapi.responses import HTMLResponse
 from fastapi.responses import FileResponse
 from fastapi.templating import Jinja2Templates
 from fastapi.staticfiles import StaticFiles
-
+from typing import Tuple
 # ...
 # Mount a directory to serve static files (e.g., CSS and JavaScript)
 
@@ -125,7 +125,21 @@ def detect_onnx_models(path):
         return onnx_models[0], onnx_configs[0]
     else:
         return None
-
+# Define a dependency function to get the selected_model and selected_speaker_id on startup
+def get_initial_values():
+    # You can set default values or load them from a configuration file here
+    selected_model = onnx_models[0] if onnx_models else "default_model"
+    selected_speaker_id = "default_speaker_id"  # Default value
+
+    # Check if there are onnx models and load the speaker_id_map from the first model's config
+    if onnx_models:
+        first_model_config = model_configurations.get(onnx_models[0])
+        if first_model_config:
+            speaker_id_map = first_model_config.get("speaker_id_map")
+            if speaker_id_map:
+                selected_speaker_id = next(iter(speaker_id_map))  # Get the first speaker_id
+
+    return selected_model, selected_speaker_id
 @app.get("/get_speaker_id_map")
 async def get_speaker_id_map(selected_model: str):
     config = model_configurations.get(selected_model)
@@ -177,6 +191,21 @@ def load_model_configuration(models_path, config_name):
         return None
 
 
+# Define a dependency function to get the selected_model and selected_speaker_id on startup
+def get_initial_values() -> Tuple[str, str]:
+    # You can set default values or load them from a configuration file here
+    selected_model = onnx_models[0] if onnx_models else "default_model"
+    selected_speaker_id = "default_speaker_id"  # Default value
+
+    # Check if there are onnx models and load the speaker_id_map from the first model's config
+    if onnx_models:
+        first_model_config = model_configurations.get(onnx_models[0])
+        if first_model_config:
+            speaker_id_map = first_model_config.get("speaker_id_map")
+            if speaker_id_map:
+                selected_speaker_id = next(iter(speaker_id_map))  # Get the first speaker_id
+
+    return selected_model, selected_speaker_id
 
 
         
@@ -185,12 +214,13 @@ async def main(
     request: Request,
     text_input: str = Form(default="1, 2, 3. This is a test. Enter some text to generate."),
     selected_model: str = Form(...),  # Selected model
-    speaker_select: str = Form(...),  # Selected speaker ID
+    selected_speaker_id: str = Form(...),  # Selected speaker ID
     speaker: str = Form(...),
     speed_slider: float = Form(...),
     noise_scale_slider: float = Form(...),
     noise_scale_w_slider: float = Form(...),
-    play: bool = Form(True)
+    play: bool = Form(True),
+    initial_values: Tuple[str, str] = Depends(get_initial_values)  # Use the dependency here
 ):
     # ... (previous code)
 
@@ -240,8 +270,11 @@ async def main(
         "file_url": file_url,
         "text_input": text_input,
         "data": data,
-        "model_names": onnx_models,
+     #   "model_names": onnx_models,
         "selected_model": selected_model,
+        "model_names": model_configurations.keys(),
+        "initial_selected_model": initial_selected_model,
+        "initial_selected_speaker_id": initial_selected_speaker_id,
         "speaker_id_map": speaker_id_map,  # Make sure speaker_id_map is included here
         "speaker_select": speaker_select,
         "dynamic_content": response_html