Update index.html

index.html

@@ -32,8 +32,8 @@
 async function generateCommit() {
     const MODEL_ID = "nivethitha1703/developer-git-commit-agent";
     
-    // FIX: Safely reads the hidden token from Hugging Face infrastructure without revealing it in your code
-    const HF_TOKEN = window.huggingface?.variables?.HF_TOKEN; 
+    // Safely reads the public variable without hardcoding the token string in text
+    const HF_TOKEN = window.huggingface?.variables?.HF_TOKEN || ""; 
     
     const diffText = document.getElementById("gitDiff").value;
     const outputBox = document.getElementById("outputBox");
@@ -44,12 +44,6 @@ async function generateCommit() {
         return;
     }
     
-    if (!HF_TOKEN) {
-        outputBox.innerText = "Error: HF_TOKEN secret not found in Space Settings. Please add it first.";
-        outputBox.style.display = "block";
-        return;
-    }
-    
     outputBox.style.display = "none";
     loadingStatus.style.display = "block";
     
@@ -59,7 +53,7 @@ async function generateCommit() {
         const response = await fetch(`https://huggingface.co{MODEL_ID}`, {
             method: "POST",
             headers: {
-                "Authorization": `Bearer ${HF_TOKEN}`,
+                "Authorization": HF_TOKEN ? `Bearer ${HF_TOKEN}` : "", 
                 "Content-Type": "application/json"
             },
             body: JSON.stringify({
@@ -71,24 +65,26 @@ async function generateCommit() {
         const data = await response.json();
         loadingStatus.style.display = "none";
         
+        if (data.error && data.error.includes("loading")) {
+            outputBox.innerText = "⏳ Model is currently loading on Hugging Face servers. Retrying automatically in 10 seconds...";
+            outputBox.style.display = "block";
+            setTimeout(generateCommit, 10000);
+            return;
+        }
+        
         if (data.error) {
             outputBox.innerText = "API Error: " + data.error;
-        } else if (Array.isArray(data) && data[0]?.generated_text) {
-            let result = data[0].generated_text;
-            result = result.replace(prompt, "").trim();
-            outputBox.innerText = result;
-        } else if (data.generated_text) {
-            let result = data.generated_text;
+        } else {
+            // Support both array and object response structures from the inference API
+            let result = Array.isArray(data) ? data[0].generated_text : data.generated_text;
             result = result.replace(prompt, "").trim();
             outputBox.innerText = result;
-        } else {
-            outputBox.innerText = JSON.stringify(data);
         }
         outputBox.style.display = "block";
         
     } catch (error) {
         loadingStatus.style.display = "none";
-        outputBox.innerText = "Error calling Inference API: " + error.message;
+        outputBox.innerText = "Fetch Error: " + error.message;
         outputBox.style.display = "block";
     }
 }