Update app.py

app.py

@@ -120,14 +120,18 @@ def infer(
         x_tst_lengths = torch.LongTensor([phones.size(0)]).to(device)
         emo = emo.to(device).unsqueeze(0)
         del phones
-        # ดึง spk2id object (HParams)
-        spk2id_raw = hps.data["spk2id"]
-        
-        # แปลงเป็น dict จริง ๆ
-        spk2id_dict = {k: v for k, v in spk2id_raw.items()}
+        spk2id_dict = {k: v for k, v in hps.data["spk2id"].items()}
+
+        # ถ้า sid เป็น index (เช่น 0) → แปลงเป็นชื่อ
+        if isinstance(sid, int) or sid.isdigit():
+            sid_int = int(sid)
+            name_map = {v: k for k, v in spk2id_dict.items()}
+            if sid_int not in name_map:
+                raise ValueError(f"Speaker index {sid_int} not found.")
+            sid = name_map[sid_int]
+        else:
+            sid = str(sid).upper()
         
-        # จัดการ speaker ID
-        sid = str(sid).upper()
         if sid not in spk2id_dict:
             raise ValueError(f"Speaker ID '{sid}' not found. Available: {list(spk2id_dict.keys())}")