@@ -39,7 +39,7 @@ from ml_model.models import NeuronModel from ml_model.validators import ModelInputValidator from payments.exceptions.insufficient_balance import InsufficientBalance -from .models import Audio, Image, Video, VoiceClone, Voice, Preset +from .models import Audio, Image, Video, VoiceClone logger = logging.getLogger(__name__) @@ -266,40 +266,6 @@ class ModelVideosAPIView(MediaAPIView): class ModelAudiosAPIVIew(MediaAPIView): manager = Audio - @extend_schema( - parameters=[ - OpenApiParameter('model', str, 'path', required=True), - ], - request=MessageSerializer, - responses={ - 201: MessageSerializer(many=True), - }, - ) - def post(self, request: Request, model: str, *args, **kwargs) -> Response: - data = get_request_data(request) - if not (request.FILES.get('file') or data.get('file')): - voice_id = data.pop('voice_id', None) - preset_id = data.pop('preset_id', None) - - try: - if voice_id: - voice = Voice.objects.get(pk=voice_id, user=request.user) - transcription = voice.transcription - elif preset_id: - voice = Preset.objects.get(uid=preset_id) - transcription = voice.metadata.get('transcription', '') - else: - return super().post(request, model, *args, **kwargs) - except (Voice.DoesNotExist, Preset.DoesNotExist): - return Response( - {'detail': _('Voice not found.')}, - status=HTTP_400_BAD_REQUEST, - ) - - info = data['info'] - data.update({'file': voice.file, 'info': {'transcription': transcription, **info}}) - return super().post(request, model, data=data, *args, **kwargs) - class ModelVoiceCloneAPIView(MediaAPIView): manager = VoiceClone @@ -14,7 +14,6 @@ from ml_model.models import NeuronModel from ml_model.selectors.ml_models_selector import NeuronModelSelector from ml_model.selectors.param_selector import ParamSelector from ml_model.serializers import ModelParameterSerializer -from tools.media.models import Preset, Voice from tools.public_api.models import APIStore from tools.public_api.selectors.api_key import APIKeySelector from tools.public_api.views.base import BaseGenerationView @@ -36,34 +35,6 @@ class AudioView(BaseGenerationView): output_content_type = ContentTypes.AUDIO description = 'Get Audio Generation from model in URL slug. Only POST Requests.' - def post(self, request, model_slug, *args, **kwargs): - if not (request.FILES.get('file') or request.data.get('file')): - api_key_value = request.headers.get('Authorization') - if (split_api_key := api_key_value.split())[0] == 'Bearer': - api_key_value = split_api_key[-1] - user = APIKeySelector.get_user_by_key(key_value=api_key_value) - voice_id = str(request.data.pop('voice_id', '')) - try: - if not voice_id: - return super().post(request, model_slug, *args, **kwargs) - elif voice_id.isdigit(): - voice = Voice.objects.get(pk=voice_id, user=user) - # transcription = voice.transcription - else: - voice = Preset.objects.get(uid=voice_id) - # transcription = voice.metadata.get('transcription', '') - except (Voice.DoesNotExist, Preset.DoesNotExist): - return Response( - {'detail': _('Voice not found.')}, - status=HTTP_400_BAD_REQUEST, - ) - request.data.update( - { - 'file': voice.file, # 'info': {'transcription': transcription, **request.data['info']} - } - ) - return super().post(request, model_slug, *args, **kwargs) - class VideoView(BaseGenerationView): output_content_type = ContentTypes.VIDEO