Simplify adapter to use direct imports without package structure

adapter.py

@@ -68,65 +68,39 @@ def ensure_utility_files_available():
 # Ensure utility files are available before importing
 ensure_utility_files_available()
 
-# Create the ablang2.pretrained_utils package structure
-if not os.path.exists('ablang2'):
-    os.makedirs('ablang2', exist_ok=True)
-if not os.path.exists('ablang2/pretrained_utils'):
-    os.makedirs('ablang2/pretrained_utils', exist_ok=True)
-
-# Create __init__.py files
-with open('ablang2/__init__.py', 'w') as f:
-    f.write('# Mock ablang2 package\n')
-
-with open('ablang2/pretrained_utils/__init__.py', 'w') as f:
-    f.write('# Mock pretrained_utils package\n')
-
-# Copy utility files to the package structure
+# Copy utility files to the current directory if they don't exist
 for file in UTILITY_FILES:
     src = os.path.join(current_dir, file)
-    dst = os.path.join(current_dir, 'ablang2', 'pretrained_utils', file)
-    if os.path.exists(src) and not os.path.exists(dst):
-        # Ensure the destination directory exists
-        os.makedirs(os.path.dirname(dst), exist_ok=True)
-        shutil.copy2(src, dst)
-
-# Also copy ablang_encodings.py as encodings.py (original name for compatibility)
-# The file should be in the current directory after the first copy operation
-encodings_src = os.path.join(current_dir, 'ablang_encodings.py')
-print(f"🔍 Checking if ablang_encodings.py exists: {os.path.exists(encodings_src)}")
-if os.path.exists(encodings_src):
-    # Ensure the destination directory exists
-    os.makedirs('ablang2/pretrained_utils', exist_ok=True)
-    dst = 'ablang2/pretrained_utils/encodings.py'
-    print(f"📋 Copying {encodings_src} to {dst}")
-    try:
-        shutil.copy2(encodings_src, dst)
-        print("✅ Copied ablang_encodings.py as encodings.py for compatibility")
-        # Verify the copy worked
-        if os.path.exists(dst):
-            print(f"✅ Verified {dst} exists after copy")
-        else:
-            print(f"❌ {dst} does not exist after copy!")
-    except Exception as e:
-        print(f"❌ Error copying file: {e}")
-else:
-    print("❌ ablang_encodings.py not found!")
-
-# Debug: Check what files are actually in the pretrained_utils directory
-pretrained_utils_dir = os.path.join(current_dir, 'ablang2', 'pretrained_utils')
-if os.path.exists(pretrained_utils_dir):
-    print(f"📁 Files in {pretrained_utils_dir}:")
-    for f in os.listdir(pretrained_utils_dir):
+    if not os.path.exists(src):
+        # Try to find the file in possible source locations
+        possible_paths = [
+            os.path.join(current_dir, '..'),  # Parent directory
+            os.path.join(current_dir, '..', '..'),  # Grandparent directory
+            os.path.join(os.path.expanduser('~'), 'ablang2'),  # Home directory
+            '/data/hn533621/ablang2',  # Known repository location
+        ]
+        
+        for path in possible_paths:
+            potential_src = os.path.join(path, file)
+            if os.path.exists(potential_src):
+                shutil.copy2(potential_src, src)
+                print(f"✅ Copied {file} to current directory")
+                break
+
+# Debug: Check what files are in the current directory
+print(f"📁 Files in current directory ({current_dir}):")
+for f in os.listdir(current_dir):
+    if f.endswith('.py'):
         print(f"   {f}")
 
-# Now import using the original structure
-from ablang2.pretrained_utils.restoration import AbRestore
-from ablang2.pretrained_utils.encodings import AbEncoding
-from ablang2.pretrained_utils.alignment import AbAlignment
-from ablang2.pretrained_utils.scores import AbScores
+# Import utility modules directly (no package structure needed)
+from restoration import AbRestore
+from ablang_encodings import AbEncoding
+from alignment import AbAlignment
+from scores import AbScores
 import torch
 import numpy as np
-from ablang2.pretrained_utils.extra_utils import res_to_seq, res_to_list
+from extra_utils import res_to_seq, res_to_list
 
 class HuggingFaceTokenizerAdapter:
     def __init__(self, tokenizer, device):