Update js/index.js

js/index.js

@@ -41,13 +41,17 @@ document.getElementById('repoForm').addEventListener('submit', async function (e
     document.getElementById('copyButton').style.display = 'none';
     document.getElementById('downloadButton').style.display = 'none';
 
+    console.log("--- Repo Form Submitted ---");
+    console.log("URL:", repoUrl);
 
     try {
         const repoInfo = parseRepoUrl(repoUrl);
+        console.log("Parsed Repo Info:", repoInfo);
         currentRepoInfo = { ...repoInfo, accessToken }; // Store for later use
 
         let tree;
         if (repoInfo.source === 'github') {
+            console.log("Fetching from GitHub...");
             const { owner, repo, lastString } = repoInfo;
             let refFromUrl = '';
             let pathFromUrl = '';
@@ -68,12 +72,15 @@ document.getElementById('repoForm').addEventListener('submit', async function (e
             tree = await fetchGitHubRepoTree(owner, repo, sha, accessToken);
 
         } else if (repoInfo.source === 'huggingface') {
+            console.log("Fetching from Hugging Face...");
             const { owner, repo, repo_type, lastString } = repoInfo;
             let refFromUrl = 'main'; // Default branch
             let pathFromUrl = '';
 
             if (lastString) {
+                console.log("Fetching HF references for lastString:", lastString);
                 const refs = await getHuggingFaceReferences(owner, repo, repo_type, accessToken);
+                console.log("Available HF Refs:", refs);
                 const matchingRef = refs.find(ref => lastString.startsWith(ref + '/'));
                 if (matchingRef) {
                     refFromUrl = matchingRef;
@@ -84,16 +91,19 @@ document.getElementById('repoForm').addEventListener('submit', async function (e
                 }
             }
             
+            console.log(`Determined HF Ref: '${refFromUrl}', Path: '${pathFromUrl}'`);
             currentRepoInfo.ref = refFromUrl;
             tree = await fetchHuggingFaceTree(owner, repo, repo_type, refFromUrl, pathFromUrl, accessToken);
         }
-
+        
+        console.log("Final tree object before display:", tree);
         displayDirectoryStructure(tree);
         document.getElementById('generateTextButton').style.display = 'flex';
         document.getElementById('downloadZipButton').style.display = 'flex';
         outputText.value = 'Select files and click "Generate Text File" or "Download Zip".';
 
     } catch (error) {
+        console.error("Error fetching repository contents:", error);
         outputText.value = `Error fetching repository contents: ${error.message}\n\n` +
             "Please ensure:\n" +
             "1. The repository URL is correct and accessible.\n" +
@@ -279,6 +289,7 @@ async function fetchGitHubRepoTree(owner, repo, sha, token) {
 async function fetchHuggingFaceTree(owner, repo, repo_type, ref, path, token) {
     const typePath = repo_type === 'model' ? 'models' : repo_type === 'dataset' ? 'datasets' : 'spaces';
     const url = `https://huggingface.co/api/${typePath}/${owner}/${repo}/tree/${ref}`;
+    console.log("Fetching HF tree from URL:", url);
     
     const headers = {};
     if (token) headers['Authorization'] = `Bearer ${token}`;
@@ -287,12 +298,16 @@ async function fetchHuggingFaceTree(owner, repo, repo_type, ref, path, token) {
     if (!response.ok) handleFetchError(response, 'huggingface');
     
     let tree = await response.json();
+    console.log("Raw HF API response:", JSON.parse(JSON.stringify(tree)));
+
 
     if (path) {
+        console.log("Filtering HF tree with path:", path);
         tree = tree.filter(item => item.path.startsWith(path + '/') || item.path === path);
+        console.log("Filtered HF tree:", JSON.parse(JSON.stringify(tree)));
     }
     
-    return tree.map(item => {
+    const mappedTree = tree.map(item => {
         let repoIdForUrl;
         switch (repo_type) {
             case 'dataset':
@@ -304,13 +319,18 @@ async function fetchHuggingFaceTree(owner, repo, repo_type, ref, path, token) {
             default: // model
                 repoIdForUrl = `${owner}/${repo}`;
         }
-        return {
+        const mappedItem = {
             path: item.path,
             type: (item.type === 'file' || item.type === 'lfs') ? 'blob' : 'tree',
             urlType: 'hf',
             url: `https://huggingface.co/${repoIdForUrl}/raw/${ref}/${item.path}`
         };
+        // console.log("Mapping item:", item, "to:", mappedItem);
+        return mappedItem;
     });
+
+    console.log("Mapped HF tree:", mappedTree);
+    return mappedTree;
 }
 
 // Handle fetch errors