Refactor cache_manager CLI and simplify cache clearing

Replaces the clear_mllmcelltype_cache function with inline logic in the CLI, streamlining interactive and non-interactive cache management. Updates usage instructions and removes redundant imports and comments for clarity.

Author: cafferychen777

python/mllmcelltype/cache_manager.py

@@ -1,38 +1,12 @@
-"""Cache management module for mLLMCelltype.
+"""Cache management CLI for mLLMCelltype.
 
-This module provides utilities for managing the mLLMCelltype cache system,
-including cache inspection, clearing, and validation functions.
-
-Functions:
-    clear_mllmcelltype_cache(): Interactive cache clearing
-    clear_cache_cli(): Command-line interface for cache management
+Usage:
+    python -m mllmcelltype.cache_manager          # Interactive mode
+    python -m mllmcelltype.cache_manager --clear  # Clear cache without confirmation
+    python -m mllmcelltype.cache_manager --info   # Show cache information
 """
 
-import os
-import shutil
-
-from .utils import get_cache_stats
-
-
-def clear_mllmcelltype_cache():
-    """Clear the mLLMCelltype cache directory."""
-    info = get_cache_stats(detailed=False)
-    cache_dir = info["path"]
-
-    if info["exists"]:
-        print(f"Found cache directory: {cache_dir}")
-        print(f"Found {info['count']} cache files")
-
-        # Ask for confirmation
-        response = input("Do you want to clear all cache files? (yes/no): ")
-        if response.lower() == "yes":
-            shutil.rmtree(cache_dir)
-            os.makedirs(cache_dir, exist_ok=True)
-            print("Cache cleared successfully!")
-        else:
-            print("Cache clearing cancelled.")
-    else:
-        print("No cache directory found.")
+from .utils import clear_cache, get_cache_stats
 
 
 def clear_cache_cli():
@@ -42,26 +16,34 @@ def clear_cache_cli():
     print("mLLMCelltype Cache Manager")
     print("-" * 30)
 
-    # Check for command line arguments
     if len(sys.argv) > 1 and sys.argv[1] == "--clear":
-        # Non-interactive mode
-        from .utils import clear_cache
-
         removed = clear_cache()
         print(f"\nCleared {removed} cache files.")
+
     elif len(sys.argv) > 1 and sys.argv[1] == "--info":
-        # Show cache info
         info = get_cache_stats(detailed=False)
         print(f"\nCache directory: {info['path']}")
         print(f"Number of cache files: {info['count']}")
         print(f"Total cache size: {info['size_mb']:.2f} MB")
+
     else:
         # Interactive mode
-        clear_mllmcelltype_cache()
+        info = get_cache_stats(detailed=False)
+        if info["exists"]:
+            print(f"Found cache directory: {info['path']}")
+            print(f"Found {info['count']} cache files")
+            response = input("\nDo you want to clear all cache files? (yes/no): ")
+            if response.lower() == "yes":
+                removed = clear_cache()
+                print(f"Cleared {removed} cache files.")
+            else:
+                print("Cache clearing cancelled.")
+        else:
+            print("No cache directory found.")
 
     print("\nUsage:")
     print("  python -m mllmcelltype.cache_manager          # Interactive mode")
-    print("  python -m mllmcelltype.cache_manager --clear  # Clear cache without confirmation")
+    print("  python -m mllmcelltype.cache_manager --clear  # Clear without confirmation")
     print("  python -m mllmcelltype.cache_manager --info   # Show cache information")