| from dataclasses import dataclass |
| from typing import Optional, Dict |
| import librosa |
| import numpy as np |
| import parameters |
| from datetime import datetime |
| from S3_bucket import AWS |
|
|
| aws = AWS() |
|
|
| def upload_voice_clone_audio(audio_file,clone_id): |
| if clone_id is None: |
| clone_id="failed" |
|
|
| s3_folder = parameters.voice_clone_data_key |
| s3_key = f"{s3_folder}/{datetime.now().strftime('%Y_%b_%d_%H_%M_%S')}_{clone_id}.wav" |
| try: |
| |
| |
| aws.s3_upload_wav(obj=audio_file,s3_key=s3_key) |
| print(f"Uploaded to s3:{s3_key}") |
| return s3_key |
| except Exception as e: |
| print(f"Error uploading voice clone audio: {e}") |
| return None |
|
|
|
|
| @dataclass |
| class AudioInfo: |
| raw_audio: np.ndarray |
| sr: int |
| duration: float |
| channels: int |
| |
|
|
|
|
| class AudioStateManager: |
| def __init__(self): |
| self.current_recording_info: Optional[AudioInfo] = None |
| self.save_uploads: bool = False |
|
|
| def update_current_recording(self, filepath: str, save: bool = None): |
| try: |
| print( |
| "filepath in update_current_recording:---", |
| filepath, |
| "::and this is type of it:-", |
| type(filepath), |
| ) |
| audio, sr = librosa.load(path=filepath, sr=22000) |
| audio = audio.astype(np.float32) |
| if np.abs(audio).max() > 1: |
| audio = audio / np.abs(audio).max() |
|
|
| duration = len(audio) / sr |
|
|
| self.current_recording_info = AudioInfo( |
| raw_audio=audio, |
| sr=sr, |
| duration=duration, |
| channels=1, |
| ) |
|
|
| return duration >= 5 and duration <= 30 |
| except Exception as e: |
| print(f"Error processing audio: {e}") |
| self.current_recording_info = None |
| return False |
|
|
| def get_current_audio_info(self) -> Optional[Dict]: |
| """Get current audio information for TTS inference""" |
| if self.current_recording_info: |
| return { |
| "raw_audio": self.current_recording_info.raw_audio, |
| "sr": self.current_recording_info.sr, |
| "channels": self.current_recording_info.channels, |
| "duration": self.current_recording_info.duration, |
| } |
| return None |
|
|
|
|
