Creates more instructional README with examples

.gitignore

@@ -42,3 +42,20 @@ loras_i2v/
 .env
 
 _test*
+
+*.db
+
+# Test outputs and generated test files
+test_outputs/
+test_results_comparison/
+tests/
+
+# Task database files
+tasks.db-shm
+tasks.db-wal
+
+# venv
+venv/*
+venv312/*
+venv/
+venv312/

add_task.py

@@ -0,0 +1,73 @@
+import argparse
+import json
+import sys
+from pathlib import Path
+
+# Ensure project source directory is on the import path
+proj_root = Path(__file__).resolve().parent
+if str(proj_root) not in sys.path:
+    sys.path.insert(0, str(proj_root))
+
+from source import db_operations as db_ops
+from source.common_utils import generate_unique_task_id as sm_generate_unique_task_id
+
+
+def _load_params(param_arg: str) -> dict:
+    """Load params from a JSON string or from a @file reference."""
+    if param_arg.startswith("@"):
+        file_path = Path(param_arg[1:]).expanduser()
+        if not file_path.exists():
+            raise FileNotFoundError(f"Params file not found: {file_path}")
+        with open(file_path, "r", encoding="utf-8") as fp:
+            return json.load(fp)
+    # Otherwise, treat as literal JSON string
+    return json.loads(param_arg)
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(
+        "add_task – enqueue a task for the Wan2GP headless server"
+    )
+    parser.add_argument(
+        "--type",
+        required=True,
+        help="Task type string (e.g. travel_orchestrator, travel_segment, generate_openpose, …)",
+    )
+    parser.add_argument(
+        "--params",
+        required=True,
+        help="JSON string with task payload OR @<path-to-json-file>",
+    )
+    parser.add_argument(
+        "--dependant-on",
+        dest="dependant_on",
+        default=None,
+        help="Optional task_id that this new task depends on.",
+    )
+    args = parser.parse_args()
+
+    try:
+        payload_dict = _load_params(args.params)
+    except Exception as e:
+        print(f"[ERROR] Could not parse --params: {e}")
+        sys.exit(1)
+
+    # Auto-generate task_id if needed
+    if "task_id" not in payload_dict or not payload_dict["task_id"]:
+        payload_dict["task_id"] = sm_generate_unique_task_id(f"{args.type[:8]}_")
+        print(f"[INFO] task_id not supplied – generated: {payload_dict['task_id']}")
+
+    try:
+        db_ops.add_task_to_db(
+            task_payload=payload_dict,
+            task_type_str=args.type,
+            dependant_on=args.dependant_on,
+        )
+        print(f"[SUCCESS] Task {payload_dict['task_id']} ({args.type}) enqueued.")
+    except Exception as e_db:
+        print(f"[ERROR] Failed to enqueue task: {e_db}")
+        sys.exit(2)
+
+
+if __name__ == "__main__":
+    main()

check_loras.py

@@ -0,0 +1,101 @@
+#!/usr/bin/env python3
+"""
+LoRA File Integrity Checker
+
+This script checks LoRA files in common directories for corruption or size issues.
+It can also clean up corrupted files if requested.
+
+Usage:
+    python check_loras.py                    # Check all LoRA directories
+    python check_loras.py --fix              # Check and remove corrupted files
+    python check_loras.py --dir /path/to/loras  # Check specific directory
+"""
+
+import argparse
+import sys
+from pathlib import Path
+
+# Add source directory to path to import common_utils
+sys.path.append(str(Path(__file__).parent / "source"))
+from common_utils import check_loras_in_directory
+
+def main():
+    parser = argparse.ArgumentParser(description="Check LoRA file integrity")
+    parser.add_argument("--dir", type=str, help="Specific directory to check")
+    parser.add_argument("--fix", action="store_true", help="Remove corrupted files")
+    parser.add_argument("--verbose", "-v", action="store_true", help="Verbose output")
+
+    args = parser.parse_args()
+
+    # Default LoRA directories to check
+    default_dirs = [
+        "Wan2GP/loras",
+        "Wan2GP/loras_hunyuan", 
+        "Wan2GP/loras_hunyuan_i2v",
+        "Wan2GP/loras_i2v",
+        "Wan2GP/loras_ltxv"
+    ]
+
+    if args.dir:
+        dirs_to_check = [args.dir]
+    else:
+        dirs_to_check = default_dirs
+
+    total_checked = 0
+    total_valid = 0
+    total_invalid = 0
+
+    print("🔍 LoRA File Integrity Check")
+    print("=" * 50)
+
+    for lora_dir in dirs_to_check:
+        dir_path = Path(lora_dir)
+        print(f"\n📁 Checking directory: {dir_path}")
+
+        if not dir_path.exists():
+            print(f"   ⚠️  Directory not found: {dir_path}")
+            continue
+
+        results = check_loras_in_directory(dir_path, fix_issues=args.fix)
+
+        if "error" in results:
+            print(f"   ❌ Error: {results['error']}")
+            continue
+
+        total_checked += results["total_files"]
+        total_valid += results["valid_files"]
+        total_invalid += results["invalid_files"]
+
+        print(f"   📊 Files found: {results['total_files']}")
+        print(f"   ✅ Valid: {results['valid_files']}")
+        print(f"   ❌ Invalid: {results['invalid_files']}")
+
+        if results["invalid_files"] > 0:
+            print(f"   🚨 Issues found:")
+            for issue in results["issues"]:
+                print(f"      {issue}")
+
+        if args.verbose:
+            print(f"   📋 Detailed results:")
+            for summary_line in results["summary"]:
+                print(f"      {summary_line}")
+
+    print("\n" + "=" * 50)
+    print(f"📈 Summary:")
+    print(f"   Total files checked: {total_checked}")
+    print(f"   Valid files: {total_valid}")
+    print(f"   Invalid files: {total_invalid}")
+
+    if total_invalid > 0:
+        print(f"\n💡 Found {total_invalid} corrupted files.")
+        if not args.fix:
+            print("   Run with --fix to automatically remove corrupted files.")
+        else:
+            print("   Corrupted files have been removed.")
+        sys.exit(1)
+    else:
+        print("\n🎉 All LoRA files are valid!")
+        sys.exit(0)
+
+if __name__ == "__main__":
+    main()