Add optional per-image-profile Qwen prompt enhancers

profiles.py

@@ -14,6 +14,8 @@ SCHEMAS={
  'video':{'width':(256,1920,768),'height':(256,1088,512),'frames':(1,241,33),'fps':(1,60,24),'steps':(1,100,20),'seed':(-1,2147483647,-1)}
 }
 VIDEO_DEVICE_DEFAULTS={'video_device':'auto','text_encoder_device':'same'}
+PROMPT_ENHANCER_REPOS={'t2i':'Qwen/Qwen-Image-2.1-PE-T2I','i2i':'Qwen/Qwen-Image-2.1-PE-I2I'}
+PROMPT_ENHANCER_DEFAULTS={'t2i':None,'i2i':None,'device':'auto'}
 def video_parameters(params):
  params={**VIDEO_DEVICE_DEFAULTS,**params}
  for key in VIDEO_DEVICE_DEFAULTS:
@@ -82,12 +84,14 @@ class Profiles:
  def __init__(self,path,catalog):
   self.path=Path(path);self.catalog=catalog;self.lock=threading.RLock()
   self.image_runtime_ready=lambda:False
+  self.prompt_enhancer_ready=lambda task:False
   self.video_blockers=None;self.chat_blockers=None;self.tts_blockers=None;self.stt_blockers=None
   self.rows=json.loads(self.path.read_text()) if self.path.exists() else []
  def status(self):
   with self.lock:
    rows=json.loads(json.dumps(self.rows))
   for p in rows:
+   if p['kind']=='image':p['prompt_enhancer']={**PROMPT_ENHANCER_DEFAULTS,**p.get('prompt_enhancer',{})}
    if p['kind']=='video':p['parameters']={**VIDEO_DEVICE_DEFAULTS,**p['parameters']}
    if p['kind']=='chat':p['parameters']={**CHAT_GPU_DEFAULTS,**CHAT_SAMPLING,**p['parameters']}
    try:
@@ -111,6 +115,9 @@ class Profiles:
    if p['kind']=='stt' and self.stt_blockers and p.get('model'):p['blockers']=self.stt_blockers(p)
    if p['kind']=='chat' and self.chat_blockers and p.get('model'):p['blockers']=self.chat_blockers(p)
    if p['kind']=='image' and (p['parameters']['width']>1024 or p['parameters']['height']>1024):p['blockers'].append('Die Bildlaufzeit unterstützt derzeit maximal 1024 × 1024 Pixel.')
+   if p['kind']=='image':
+    for task,repo in PROMPT_ENHANCER_REPOS.items():
+     if p['prompt_enhancer'].get(task)==repo and not self.prompt_enhancer_ready(task):p['blockers'].append('Prompt-Aufwerter '+task.upper()+' fehlt. Unter Bildgenerierung → Aufwerter installieren.')
    p['runnable']=not p['blockers'];p['state']='ready' if p['runnable'] else 'configured'
   return {'profiles':[p for p in rows if p['kind']!='video'],'schemas':{k:v for k,v in SCHEMAS.items() if k!='video'}}
  def save(self,data):
@@ -140,12 +147,26 @@ class Profiles:
    if data['revision']!=(existing['revision'] if existing else 0):raise ValueError('Profil wurde zwischenzeitlich geändert. Ansicht neu laden.')
    if any(p['name']==name and p is not existing for p in self.rows):raise ValueError('API-Profilname bereits vergeben.')
    components=existing.get('components',{}) if existing and existing['model_id']==model['id'] else {}
-   row=dict(components=components,id=existing['id'] if existing else uuid.uuid4().hex,revision=data['revision']+1,name=name,kind=kind,model_id=model['id'],parameters=params,updated_at=time.time())
+   row=dict(components=components,prompt_enhancer=existing.get('prompt_enhancer',dict(PROMPT_ENHANCER_DEFAULTS)) if existing and existing['model_id']==model['id'] and kind=='image' else dict(PROMPT_ENHANCER_DEFAULTS) if kind=='image' else None,id=existing['id'] if existing else uuid.uuid4().hex,revision=data['revision']+1,name=name,kind=kind,model_id=model['id'],parameters=params,updated_at=time.time())
    rows=[row if p is existing else p for p in self.rows] if existing else self.rows+[row]
    self.path.parent.mkdir(parents=True,exist_ok=True,mode=0o700)
    temp=self.path.with_suffix('.tmp');temp.write_text(json.dumps(rows));temp.replace(self.path);self.rows=rows
    return row
 
+ def configure_prompt_enhancer(self,data):
+  if not isinstance(data,dict) or set(data)!={'id','revision','prompt_enhancer'}:raise ValueError('Ungültige Prompt-Aufwerter-Einstellung.')
+  config=data['prompt_enhancer']
+  if not isinstance(config,dict) or set(config)!=set(PROMPT_ENHANCER_DEFAULTS):raise ValueError('Unvollständige Prompt-Aufwerter-Einstellung.')
+  if any(config[task] not in (None,repo) for task,repo in PROMPT_ENHANCER_REPOS.items()):raise ValueError('Für diese Aufgabe ist der Aufwerter nicht geprüft.')
+  if config['device'] not in ('auto','5080','3060','cpu'):raise ValueError('Ungültiges Gerät für den Aufwerter.')
+  with self.lock:
+   old=next((p for p in self.rows if p['id']==data['id'] and p['kind']=='image'),None)
+   if not old or data['revision']!=old['revision']:raise ValueError('Bildprofil geändert oder nicht gefunden. Ansicht neu laden.')
+   updated=dict(old,prompt_enhancer=dict(config),revision=old['revision']+1,updated_at=time.time())
+   rows=[updated if p is old else p for p in self.rows]
+   temp=self.path.with_suffix('.tmp');temp.write_text(json.dumps(rows));temp.replace(self.path);self.rows=rows
+   return updated
+
  def delete(self,data):
   if set(data)!={'id','revision'}:raise ValueError('Ungültige Profilfelder.')
   with self.lock: