Improve utility file copying for Google Colab and other environments

adapter.py

@@ -29,25 +29,43 @@ def ensure_utility_files_available():
             missing_files.append(file)
     
     if missing_files:
+        print(f"🔍 Looking for missing utility files: {missing_files}")
+        
         # Try to find the repository root (where all utility files are)
         # Look for common parent directories that might contain the files
         possible_paths = [
             os.path.join(current_dir, '..'),  # Parent directory
             os.path.join(current_dir, '..', '..'),  # Grandparent directory
+            os.path.join(current_dir, '..', '..', '..'),  # Great-grandparent directory
             os.path.join(os.path.expanduser('~'), 'ablang2'),  # Home directory
             '/data/hn533621/ablang2',  # Known repository location
+            '/content/ablang2',  # Google Colab common location
+            '/tmp/ablang2',  # Temporary directory
         ]
         
+        # Also try to find files in the Hugging Face cache structure
+        cache_dir = os.path.dirname(current_dir)
+        if 'huggingface' in cache_dir:
+            # Look in the repository root within the cache
+            repo_root = os.path.join(cache_dir, '..', '..', '..', '..')
+            possible_paths.append(repo_root)
+        
         for path in possible_paths:
             if os.path.exists(path):
+                print(f"🔍 Checking path: {path}")
                 # Check if all missing files exist in this path
                 all_found = True
                 for file in missing_files:
-                    if not os.path.exists(os.path.join(path, file)):
+                    file_path = os.path.join(path, file)
+                    if not os.path.exists(file_path):
                         all_found = False
+                        print(f"   ❌ Missing: {file}")
                         break
+                    else:
+                        print(f"   ✅ Found: {file}")
                 
                 if all_found:
+                    print(f"🎯 Found all files in: {path}")
                     # Copy all missing files
                     for file in missing_files:
                         src = os.path.join(path, file)
@@ -57,36 +75,30 @@ def ensure_utility_files_available():
                     return True
         
         # If we get here, we couldn't find the files
-        raise FileNotFoundError(
-            f"Missing utility files: {missing_files}. "
-            "These files are required for the adapter to work. "
-            "Please ensure the repository is properly set up."
-        )
+        print(f"❌ Could not find utility files in any of the searched paths:")
+        for path in possible_paths:
+            print(f"   - {path}")
+        
+        # For Colab environments, provide a helpful error message
+        if 'google.colab' in str(sys.modules):
+            raise FileNotFoundError(
+                f"Missing utility files: {missing_files}. "
+                "This appears to be a Google Colab environment. "
+                "Please ensure you have cloned the repository and the utility files are available. "
+                "Try running: !git clone https://huggingface.co/hemantn/ablang2"
+            )
+        else:
+            raise FileNotFoundError(
+                f"Missing utility files: {missing_files}. "
+                "These files are required for the adapter to work. "
+                "Please ensure the repository is properly set up."
+            )
     
     return True
 
 # Ensure utility files are available before importing
 ensure_utility_files_available()
 
-# Copy utility files to the current directory if they don't exist
-for file in UTILITY_FILES:
-    src = os.path.join(current_dir, file)
-    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):