This commit is contained in:
j 2025-01-24 20:35:50 +05:30
parent ade9a049c2
commit 41e256be5d
2 changed files with 8 additions and 18 deletions

View file

@ -6,10 +6,11 @@ from app.celery import app
from django.conf import settings
from item import utils
from user.models import User
from item.models import Item
from taskqueue.models import Task
from user.models import User
from . import transcribe
from .transcribe import extract_subtitles
logger = logging.getLogger('pandora.' + __name__)
@ -17,24 +18,13 @@ logger = logging.getLogger('pandora.' + __name__)
def transcribe(**kwargs):
user = User.objects.get(username=kwargs["user"])
item = Item.objects.get(public_id=kwargs["item"])
subtitles = utils.get_by_key(layers, 'isSubtitles', True)
subtitles = utils.get_by_key(settings.CONFIG['layers'], 'isSubtitles', True)['id']
gpu = getattr(settings, "TRANSCRIBE_GPU", False)
join_sentences = getattr(settings, "TRANSCRIBE_JOIN", False)
model = getattr(settings, "TRANSCRIBE_MODEL", "small")
translate = getattr(settings, "TRANSCRIBE_TRANSLATE", "")
logger.error(
"transcribe %s %s %s %s %s %s %s",
item,
user,
subtitles,
translate,
gpu,
join_sentences,
model
)
transcribe.extract_subtitles(
extract_subtitles(
item,
user,
subtitles,

View file

@ -21,11 +21,11 @@ def transcribeAudio(request, data):
item: id // item id
}
'''
response = {}
response = json_response()
item = get_object_or_404_json(Item, public_id=data["item"])
if has_capability(request.user, 'canTranscribeAudio') and item.editalbe(request.user):
if has_capability(request.user, 'canTranscribeAudio') and item.editable(request.user):
t = tasks.transcribe.delay(item=data["item"], user=request.user.username)
response["taskId"] = t.task_id
response["data"]["taskId"] = t.task_id
else:
response = json_response(status=403, text='permission denied')
return render_to_json_response(response)