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

better imports

Browse source
This commit is contained in:
j 2026-01-15 17:02:34 +00:00
commit 1be6e977ac
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
generate.py
View file

@ -13,6 +13,8 @@ from byteplussdkarkruntime import Ark
from django.conf import settings from django.conf import settings
from item.models import Item from item.models import Item
from document.models import Document
from archive.models import File, Stream
os.environ["FAL_KEY"] = settings.FAL_KEY os.environ["FAL_KEY"] = settings.FAL_KEY
@ -479,7 +481,8 @@ def reshoot_item(item, extra_prompt=None, first_frame=None, keep=False):
ai.data["seed"] = status["seed"] ai.data["seed"] = status["seed"]
if first_frame: if first_frame:
ai.data["firstframe"] = first_frame.split('?')[0] ai.data["firstframe"] = first_frame.split('?')[0]
if isinstance(first_frame, Document):
first_frame.add(ai)
ai.save() ai.save()
if not keep: if not keep:
shutil.rmtree(os.path.dirname(output)) shutil.rmtree(os.path.dirname(output))
@ -874,7 +877,6 @@ def luma_modify_item(item, prompt="", image_prompt=None, first_frame=None, keep=
def add_ai_variant(item, video_path, type): def add_ai_variant(item, video_path, type):
if isinstance(item, str): if isinstance(item, str):
item = Item.objects.get(public_id=item) item = Item.objects.get(public_id=item)
from archive.models import File, Stream
ai = Item() ai = Item()
ai.user = item.user ai.user = item.user
@ -900,7 +902,6 @@ def add_ai_variant(item, video_path, type):
return ai return ai
def add_ai_image(item, position, url, extension=None): def add_ai_image(item, position, url, extension=None):
from document.models import Document
if extension is None: if extension is None:
extension = url.split('.')[-1].split('?')[0] extension = url.split('.')[-1].split('?')[0]
if extension == 'jpeg': extension = 'jpg' if extension == 'jpeg': extension = 'jpg'