Fixed some package errors

Author: UtkarshTheDev

locallab/cli/interactive.py

@@ -262,26 +262,24 @@ def prompt_for_config(use_ngrok: bool = None, port: int = None, ngrok_auth_token
         click.echo("\n🔑 Enter your HuggingFace token (optional)")
         click.echo("   Get your token from: https://huggingface.co/settings/tokens")
         click.echo("   Press Enter to skip or paste your token (it will be hidden): ", nl=False)
-        hf_token = click.getchar()
-        if hf_token and hf_token != '\r' and hf_token != '\n':
-            # Read the rest of the token
-            token_chars = [hf_token]
-            while True:
-                char = click.getchar()
-                if char in ('\r', '\n'):
-                    break
-                token_chars.append(char)
-                click.echo('*', nl=False)  # Show * for each character
-            hf_token = ''.join(token_chars)
-            click.echo()  # New line after token input
-            click.echo("✅ Token saved!")
-        else:
-            click.echo("\nSkipping HuggingFace token...")
-            hf_token = ""
+        
+        # Read token character by character
+        chars = []
+        while True:
+            char = click.getchar()
+            if char in ('\r', '\n'):
+                break
+            chars.append(char)
+            click.echo('*', nl=False)  # Show * for each character
+            
+        hf_token = ''.join(chars)
 
         if hf_token:
+            click.echo("\n✅ Token saved!")
             os.environ["HUGGINGFACE_TOKEN"] = hf_token
             config["huggingface_token"] = hf_token
+        else:
+            click.echo("\nSkipping HuggingFace token...")
 
     click.echo("\n✅ Configuration complete!\n")
     return config

locallab/config.py

@@ -515,6 +515,43 @@ def reset_instructions(self, model_id: Optional[str] = None):
 system_instructions = SystemInstructions()
 
 
-
-
-
+def get_hf_token(interactive: bool = False) -> Optional[str]:
+    """Get HuggingFace token from environment or config"""
+    # First check environment
+    token = os.environ.get("HUGGINGFACE_TOKEN")
+    
+    # Then check config
+    if not token:
+        try:
+            from .cli.config import get_config_value
+            token = get_config_value("huggingface_token")
+        except:
+            pass
+    
+    # If interactive and still no token, prompt user
+    if not token and interactive:
+        try:
+            click.echo("\n🔑 HuggingFace token is recommended for better model access.")
+            click.echo("Get your token from: https://huggingface.co/settings/tokens")
+            click.echo("Enter token (press Enter to skip): ", nl=False)
+            
+            # Read token character by character
+            chars = []
+            while True:
+                char = click.getchar()
+                if char in ('\r', '\n'):
+                    break
+                chars.append(char)
+                click.echo('*', nl=False)  # Show * for each character
+            
+            token = ''.join(chars)
+            if token:
+                click.echo("\n✅ Token saved!")
+                os.environ["HUGGINGFACE_TOKEN"] = token
+            else:
+                click.echo("\nSkipping token...")
+        except:
+            # If there's any error in interactive mode, just return None
+            pass
+            
+    return token

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="locallab",
-    version="0.4.24",
+    version="0.4.26",
     packages=find_packages(include=["locallab", "locallab.*"]),
     install_requires=[
         "fastapi>=0.95.0,<1.0.0",