Add selectable native LTX Original runtime and DeskWEB API integration

video_original.py

@@ -0,0 +1,74 @@
+"""Selectable native LTX Original and ComfyUI, sharing Deck's GPU admission gate."""
+import json
+import os
+from pathlib import Path, PurePosixPath
+import shutil
+import socket
+import subprocess
+import time
+import urllib.request
+from video_comfy import VideoComfy
+
+class VideoRuntimes(VideoComfy):
+ def __init__(self,*args,original_runtime):
+  self.original_runtime=original_runtime
+  super().__init__(*args)
+ def runtime_kind(self):return self.selection.get('runtime','comfy')
+ def models(self):
+  rows=super().models()
+  if self.runtime_kind()=='original':
+   installed=self.original_runtime.status()['installed']
+   for row in rows:
+    row['blockers']=[x for x in row['blockers'] if not x.startswith('Passende ComfyUI')]
+    if not installed:row['blockers'].append('LTX Original fehlt. Unter Einstellungen → Laufzeiten → LTX Original installieren.')
+    row['runnable']=not row['blockers'];row['state']='ready' if row['runnable'] else 'configured';row['label']='LTX Original bereit · lädt bei Anfrage' if row['runnable'] else 'Voraussetzungen fehlen'
+  return rows
+ def select_runtime(self,kind):
+  if kind not in ('comfy','original'):raise ValueError('Laufzeit muss comfy oder original sein.')
+  with self.lock:
+   if self.scheduler.gpu_mode!='llm' or self.state=='switching':raise ValueError('Zuerst auf LLM wechseln; laufende Video-API wird nicht umgebogen.')
+   self.selection['runtime']=kind;self._save_selection()
+  return self.status()
+ def _save_selection(self):
+  self.root.mkdir(parents=True,exist_ok=True);temp=self.path.with_suffix('.tmp');temp.write_text(json.dumps(self.selection));temp.replace(self.path)
+ def select(self,ident):
+  kind=self.runtime_kind();super().select(ident)
+  self.selection['runtime']=kind;self._save_selection();return self.status()
+ def status(self):
+  result=super().status();result['runtime_kind']=self.runtime_kind();result['runtimes']={'comfy':self.runtime.status(),'original':self.original_runtime.status()}
+  if self.runtime_kind()=='original':
+   if result.get('error') and result['error'].startswith('ComfyUI wurde'):result['error']='LTX Original wurde unerwartet beendet. Auf LLM wechseln und Backend erneut starten.'
+   result['runtime']=self.original_runtime.status();result['memory_policy']='LTX Original: API startet im Videomodus; Gewichte laden bei Generierung. LLM-Wechsel beendet den gesamten Backend-Prozess.'
+   if result['service']:
+    result['service'].update(api_url='/api/… (Originale LTX-Desktop-API auf dem gemeinsamen API-Port)',message='5080: Transformer/VAE · 3060: Textencoder · BF16-Gewichtsstreaming; Bedienung in LTX DeskWEB.',ui_url='http://127.0.0.1:8118')
+  return result
+ def _start(self):
+  if self.runtime_kind()=='comfy':return super()._start()
+  raw=subprocess.check_output(['nvidia-smi','--query-gpu=uuid,name,memory.used','--format=csv,noheader,nounits'],text=True)
+  gpus=[line.split(',') for line in raw.strip().splitlines()];ordered=[next((g for g in gpus if name in g[1]),None) for name in ('5080','3060')]
+  if any(g is None for g in ordered):raise ValueError('RTX 5080 und RTX 3060 benötigt.')
+  if any(float(g[2])>256 for g in ordered):raise ValueError('GPUs nicht frei; fremde Prozesse werden nicht beendet.')
+  row=next(x for x in self.models() if x['model']['id']==self.selection['model_id'])
+  work=self.root/'original-work';models=work/'models/ltx-2.5';models.mkdir(parents=True,exist_ok=True)
+  for ident in [row['model']['id'],*row['components'].values()]:
+   entry=self.catalog.entry(ident);source=self.catalog.root/ident/('model'+PurePosixPath(entry['file']).suffix);link=models/PurePosixPath(entry['file']).name
+   if link.exists() and not link.is_symlink():raise ValueError('LTX-Modellansicht enthält eine fremde Datei.')
+   if link.is_symlink():link.unlink()
+   link.symlink_to(source.resolve())
+  settings_path=work/'settings.json';settings=json.loads(settings_path.read_text()) if settings_path.is_file() else {}
+  settings.update(modelsDir=str(work/'models'),activeLtxModelId='ltx-2.5-22b-distilled',useLocalTextEncoder=True,useConvVae=False)
+  if not settings_path.is_file():settings.update(promptEnhancerEnabledT2V=False,promptEnhancerEnabledI2V=False)
+  settings_path.write_text(json.dumps(settings))
+  for name in ('outputs','remote-inputs/deskweb'):(work/name).mkdir(parents=True,exist_ok=True)
+  with socket.socket() as sock:sock.bind(('127.0.0.1',0));self.port=sock.getsockname()[1]
+  python,backend=self.original_runtime.paths();env=dict(os.environ,CUDA_VISIBLE_DEVICES=','.join(g[0].strip() for g in ordered),OMP_NUM_THREADS='2',HF_HUB_DISABLE_TELEMETRY='1')
+  self.switch_phase='Originale LTX-API starten · Gewichte laden erst bei Anfrage'
+  self.process=subprocess.Popen([str(python),str(Path(__file__).with_name('ltx_original_entry.py')),'--backend',str(backend),'--data',str(work),'--port',str(self.port)],env=env,cwd=backend,stdout=subprocess.DEVNULL,stderr=subprocess.DEVNULL,start_new_session=True)
+  deadline=time.monotonic()+180
+  while time.monotonic()<deadline:
+   if self.process.poll() is not None:raise ValueError('LTX-Backend-Start fehlgeschlagen; Installation und Komponenten prüfen.')
+   try:
+    with urllib.request.urlopen(f'http://127.0.0.1:{self.port}/health',timeout=2) as response:
+     if response.status==200:return
+   except (OSError,TimeoutError):time.sleep(.5)
+  raise ValueError('LTX-API ist noch nicht bereit.')