Hugging Face's logo

Spaces:
DroolingPanda
/
kokoro-tts-server
Runtime error

App Files Community
kokoro-tts-server
File size: 2,936 Bytes 
05b45a5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
 
export class VoiceService {
    constructor() {
        this.availableVoices = [];
        this.selectedVoices = new Map(); // Changed to Map to store voice:weight pairs
    }

    async loadVoices() {
        try {
            const response = await fetch('/v1/audio/voices');
            if (!response.ok) {
                const error = await response.json();
                throw new Error(error.detail?.message || 'Failed to load voices');
            }
            
            const data = await response.json();
            if (!data.voices?.length) {
                throw new Error('No voices available');
            }

            this.availableVoices = data.voices;
            
            // Select first voice if none selected
            if (this.selectedVoices.size === 0) {
                const firstVoice = this.availableVoices.find(voice => voice && voice.trim());
                if (firstVoice) {
                    this.addVoice(firstVoice);
                }
            }

            return this.availableVoices;
        } catch (error) {
            console.error('Failed to load voices:', error);
            throw error;
        }
    }

    getAvailableVoices() {
        return this.availableVoices;
    }

    getSelectedVoices() {
        return Array.from(this.selectedVoices.keys());
    }

    getSelectedVoiceWeights() {
        return Array.from(this.selectedVoices.entries()).map(([voice, weight]) => ({
            voice,
            weight
        }));
    }

    getSelectedVoiceString() {
        const entries = Array.from(this.selectedVoices.entries());
        
        // If only one voice with weight 1, return just the voice name
        if (entries.length === 1 && entries[0][1] === 1) {
            return entries[0][0];
        }
        
        // Otherwise return voice(weight) format
        return entries
            .map(([voice, weight]) => `${voice}(${weight})`)
            .join('+');
    }

    addVoice(voice, weight = 1) {
        if (this.availableVoices.includes(voice)) {
            this.selectedVoices.set(voice, parseFloat(weight) || 1);
            return true;
        }
        return false;
    }

    updateWeight(voice, weight) {
        if (this.selectedVoices.has(voice)) {
            this.selectedVoices.set(voice, parseFloat(weight) || 1);
            return true;
        }
        return false;
    }

    removeVoice(voice) {
        return this.selectedVoices.delete(voice);
    }

    clearSelectedVoices() {
        this.selectedVoices.clear();
    }

    filterVoices(searchTerm) {
        if (!searchTerm) {
            return this.availableVoices;
        }
        
        const term = searchTerm.toLowerCase();
        return this.availableVoices.filter(voice => 
            voice.toLowerCase().includes(term)
        );
    }

    hasSelectedVoices() {
        return this.selectedVoices.size > 0;
    }
}

export default VoiceService;