Refactor app.py and settings.py: Enhance Hugging Face token handling with improved validation and logging

app.py

@@ -160,6 +160,15 @@ logger = logging.getLogger(__name__)
 if not HF_TOKEN:
     raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
 
+# Test API connection
+api = HfApi(token=HF_TOKEN)
+try:
+    api.whoami()
+    logger.info("Successfully authenticated with Hugging Face")
+except Exception as e:
+    logger.error(f"Authentication failed: {e}")
+    raise
+
 # Global variables
 client = None
 context_store = {}

config/settings.py

@@ -9,36 +9,18 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Load environment variables
-env_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), '.env')
-load_dotenv(env_path)
+load_dotenv()
 
-# Try different possible token environment variables
-possible_token_vars = [
-    "pro_api_token_2",
-    "HUGGINGFACE_TOKEN",
-    "HF_TOKEN",
-    "HF_API_TOKEN"
-]
+# Get token with fallback
+HF_TOKEN = os.getenv('HUGGINGFACE_TOKEN')
+if not HF_TOKEN:
+    raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
 
-HF_TOKEN = None
-for token_var in possible_token_vars:
-    token = os.getenv(token_var)
-    if token:
-        HF_TOKEN = token
-        logger.info(f"Using token from {token_var}")
-        break
+# Validate token format
+if not HF_TOKEN.startswith('hf_'):
+    raise ValueError("Invalid Hugging Face token format")
 
-if not HF_TOKEN:
-    error_msg = (
-        "No valid API token found. Please set one of the following environment variables:\n"
-        "- pro_api_token_2\n"
-        "- HUGGINGFACE_TOKEN\n"
-        "- HF_TOKEN\n"
-        "- HF_API_TOKEN\n"
-        "You can set it in your .env file or as a system environment variable."
-    )
-    logger.error(error_msg)
-    raise ValueError(error_msg)
+print(f"Token loaded successfully: {HF_TOKEN[:5]}...")
 
 # API Configuration
 API_CONFIG = {