fix tag again

ui/src/app/api/hf-jobs/route.ts

@@ -606,60 +606,40 @@ def generate_model_card_readme(repo_id: str, config: dict, model_name: str, cura
         else:
             license_info = {"license": "creativeml-openrail-m"}
         
-        # Generate tags based on model architecture
+        # Generate tags based on model architecture group
         tags = []
         lower_arch = (arch or "").lower()
         lower_model_name = (model_config.get("name_or_path", "") or "").lower()
         base_model_lower = (base_model or "").lower()
 
-        instruction_arches = {'flux_kontext', 'hidream_e1', 'qwen_image_edit', 'qwen_image_edit_plus'}
-        is_instruction = lower_arch in instruction_arches or 'kontext' in lower_model_name
-        is_edit_model = (
-            'edit' in lower_arch
-            or 'image_edit' in lower_arch
-            or 'edit' in lower_model_name
-            or 'image-edit' in lower_model_name
-            or 'edit' in base_model_lower
-            or 'image-edit' in base_model_lower
-        )
+        # Define model groups based on the frontend options.ts structure
+        # Group: 'image' -> text-to-image
+        # Group: 'instruction' -> image-to-image
+        # Group: 'video' -> check for i2v in arch name for image-to-video vs text-to-video
 
-        datasets_config = config.get('config', {}).get('process', [{}])[0].get('datasets', [])
-        sample_config = process_config.get('sample', {}) if isinstance(process_config, dict) else {}
-        sample_num_frames = sample_config.get('num_frames') if isinstance(sample_config, dict) else None
-
-        video_indicators = [
-            'video' in lower_arch,
-            'video' in lower_model_name,
-            'video' in base_model_lower,
-            't2v' in lower_arch,
-            't2v' in lower_model_name,
-            't2v' in base_model_lower,
-            'i2v' in lower_arch,
-            'i2v' in lower_model_name,
-            'i2v' in base_model_lower,
-            process_config.get('type') == 'video' if isinstance(process_config, dict) else False,
-            any(isinstance(dataset, dict) and dataset.get('do_i2v') for dataset in datasets_config),
-            isinstance(sample_num_frames, (int, float)) and sample_num_frames and sample_num_frames > 1,
-            lower_arch.startswith('wan'),
-            'wan' in lower_model_name,
-            'wan' in base_model_lower,
-        ]
-        is_video = any(video_indicators)
-        is_i2v = any(
-            indicator
-            for indicator in [
-                'i2v' in lower_arch,
-                'i2v' in lower_model_name,
-                'i2v' in base_model_lower,
-                any(isinstance(dataset, dict) and dataset.get('do_i2v') for dataset in datasets_config),
-            ]
-        )
+        image_arches = {
+            'flux', 'flex1', 'flex2', 'chroma', 'lumina2',
+            'qwen_image', 'hidream', 'sdxl', 'sd15', 'omnigen2'
+        }
+        instruction_arches = {
+            'flux_kontext', 'qwen_image_edit', 'qwen_image_edit_plus', 'hidream_e1'
+        }
+        video_arches = {
+            'wan21:1b', 'wan21_i2v:14b480p', 'wan21_i2v:14b', 'wan21:14b',
+            'wan22_14b:t2v', 'wan22_14b_i2v', 'wan22_5b'
+        }
 
-        if is_instruction or is_edit_model:
+        # Determine the task type based on architecture group
+        if lower_arch in instruction_arches:
             tags.append("image-to-image")
-        elif is_video:
+        elif lower_arch in video_arches:
+            # Video models: check if i2v is in the architecture name
+            is_i2v = 'i2v' in lower_arch
             tags.append("image-to-video" if is_i2v else "text-to-video")
+        elif lower_arch in image_arches:
+            tags.append("text-to-image")
         else:
+            # Fallback to text-to-image for unknown architectures
             tags.append("text-to-image")
 
         if "xl" in lower_arch: