Remove Kaggle auth credentials from functions to use hf secrets instead

app.py

@@ -13,15 +13,9 @@ from Gradio_UI import GradioUI
 os.environ['KAGGLE_USERNAME'] = ''
 os.environ['KAGGLE_KEY'] = ''
 
-@tool
-def auth_kaggle(username:str, key:str) -> KaggleApi:
-    """Authenticate with Kaggle using the provided username and key.
-    Args:
-        username: Username for Kaggle.
-        key: API key for Kaggle.
+def auth_kaggle() -> KaggleApi:
+    """Authenticate Kaggle and return the API object.
     """
-    os.environ['KAGGLE_USERNAME'] = username
-    os.environ['KAGGLE_KEY'] = key
     api = KaggleApi()
     try:
         api.authenticate()
@@ -30,24 +24,16 @@ def auth_kaggle(username:str, key:str) -> KaggleApi:
     return api
 
 @tool
-def clean_kaggle_auth() -> str:
-    """Remove Kaggle credentials from the environment variables for security purposes.
-    """
-    os.environ['KAGGLE_USERNAME'] = ''
-    os.environ['KAGGLE_KEY'] = ''
-    return 'Kaggle credentials removed'
-
-@tool
-def search_kaggle_datasets(kaggle_api:KaggleApi, 
-                           search_term:str, 
+def search_kaggle_datasets(search_term:str, 
                            max_results:int = 10
                            ) -> list[dict[str]]:
     """Search for datasets on Kaggle based on a search term and return list of datasets metadata.
     Args:
-        kaggle_api: Kaggle API object with authentication.
         search_term: The term to search for.
         max_results: Maximum number of results to return.
     """
+
+    kaggle_api = auth_kaggle()
     
     # Search for datasets
     datasets = kaggle_api.dataset_list(search=search_term)
@@ -70,14 +56,12 @@ def search_kaggle_datasets(kaggle_api:KaggleApi,
 
 @tool
 def download_kaggle_dataset(
-    kaggle_api: KaggleApi,
     dataset_ref: str,
     download_path: str,
     unzip: bool = True
 ) -> str:
     """Download a dataset from Kaggle.
     Args:
-        kaggle_api: Kaggle API object with authentication.
         dataset_ref: The reference of the dataset (e.g., "username/dataset-name").
         download_path: The directory where the dataset will be downloaded.
         unzip: Whether to unzip the dataset after downloading. Default is True.
@@ -85,6 +69,8 @@ def download_kaggle_dataset(
     # Ensure the download path exists
     os.makedirs(download_path, exist_ok=True)
 
+    kaggle_api = auth_kaggle()
+
     # Download the dataset
     kaggle_api.dataset_download_files(dataset_ref, path=download_path, unzip=unzip)
 
@@ -111,8 +97,6 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     tools=[final_answer,
-           auth_kaggle,
-           clean_kaggle_auth, 
            search_kaggle_datasets, 
            download_kaggle_dataset,
            image_generation_tool],