0x2620/pandora_transcribe
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix task

Browse source
This commit is contained in:
j 2025-01-24 20:35:50 +05:30
commit 41e256be5d
2 changed files with 8 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tasks.py
View file

@ -6,10 +6,11 @@ from app.celery import app
from django.conf import settings from django.conf import settings
from item import utils from item import utils
from user.models import User
from item.models import Item 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__) logger = logging.getLogger('pandora.' + __name__)
@ -17,24 +18,13 @@ logger = logging.getLogger('pandora.' + __name__)
def transcribe(**kwargs): def transcribe(**kwargs):
user = User.objects.get(username=kwargs["user"]) user = User.objects.get(username=kwargs["user"])
item = Item.objects.get(public_id=kwargs["item"]) 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) gpu = getattr(settings, "TRANSCRIBE_GPU", False)
join_sentences = getattr(settings, "TRANSCRIBE_JOIN", False) join_sentences = getattr(settings, "TRANSCRIBE_JOIN", False)
model = getattr(settings, "TRANSCRIBE_MODEL", "small") model = getattr(settings, "TRANSCRIBE_MODEL", "small")
translate = getattr(settings, "TRANSCRIBE_TRANSLATE", "") translate = getattr(settings, "TRANSCRIBE_TRANSLATE", "")
logger.error( extract_subtitles(
"transcribe %s %s %s %s %s %s %s",
item,
user,
subtitles,
translate,
gpu,
join_sentences,
model
)
transcribe.extract_subtitles(
item, item,
user, user,
subtitles, subtitles,

6
views.py
View file

@ -21,11 +21,11 @@ def transcribeAudio(request, data):
item: id // item id item: id // item id
} }
''' '''
response = {} response = json_response()
item = get_object_or_404_json(Item, public_id=data["item"]) 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) t = tasks.transcribe.delay(item=data["item"], user=request.user.username)
response["taskId"] = t.task_id response["data"]["taskId"] = t.task_id
else: else:
response = json_response(status=403, text='permission denied') response = json_response(status=403, text='permission denied')
return render_to_json_response(response) return render_to_json_response(response)