Update server.js

server.js

@@ -12,6 +12,85 @@ import fetch from 'node-fetch';
 
 dotenv.config();
 
+// --- Hugging Face Space Auto-Restarter ---
+const HF_TOKEN = process.env.reset;
+const SPACE_ID = process.env.HF_SPACE_ID || 'shashwatIDR/vision';
+const HF_API_BASE = 'https://huggingface.co/api';
+
+async function restartSpace() {
+  if (!HF_TOKEN) {
+    console.error('[❌ No HF_TOKEN] Hugging Face token not set. Skipping restart.');
+    return;
+  }
+  console.log(`[🔄 Attempting restart] ${SPACE_ID}`);
+  try {
+    const response = await axios.post(`${HF_API_BASE}/spaces/${SPACE_ID}/restart`, {}, {
+      headers: {
+        Authorization: `Bearer ${HF_TOKEN}`,
+        'Content-Type': 'application/json'
+      },
+      timeout: 30000
+    });
+    console.log(`[✅ Restarted] ${SPACE_ID} at ${new Date().toLocaleString()}`);
+    if (response.data) {
+      console.log('[📊 Response]', response.data);
+    }
+  } catch (err) {
+    if (err.code === 'ENOTFOUND') {
+      console.error('[❌ DNS Error] Cannot resolve huggingface.co - check your internet connection');
+    } else if (err.code === 'ETIMEDOUT') {
+      console.error('[❌ Timeout] Request timed out - try again later');
+    } else if (err.response) {
+      console.error(`[❌ HTTP ${err.response.status}]`, err.response.data || err.response.statusText);
+      if (err.response.status === 401) {
+        console.error('[🔑 Auth Error] Check your Hugging Face token');
+      } else if (err.response.status === 404) {
+        console.error('[🔍 Not Found] Space may not exist or token lacks permissions');
+      }
+    } else {
+      console.error('[❌ Failed to Restart]', err.message);
+    }
+  }
+}
+
+async function testHFAPI() {
+  if (!HF_TOKEN) {
+    console.error('[❌ No HF_TOKEN] Hugging Face token not set. Skipping API test.');
+    return false;
+  }
+  try {
+    console.log('[🔍 Testing Hugging Face API endpoint...]');
+    const response = await axios.get(`${HF_API_BASE}/spaces`, {
+      headers: { Authorization: `Bearer ${HF_TOKEN}` },
+      timeout: 10000,
+      params: { limit: 1 }
+    });
+    console.log('[✅ API Test] Endpoint is working');
+    return true;
+  } catch (err) {
+    console.error('[❌ API Test Failed]', err.response?.status || err.code || err.message);
+    return false;
+  }
+}
+
+// On server startup, test API and schedule restarts
+(async () => {
+  if (!HF_TOKEN) {
+    console.error('[❌ No HF_TOKEN] Hugging Face token not set. Skipping auto-restart scheduling.');
+    return;
+  }
+  console.log('[🚀 Starting] HuggingFace Space Auto-Restarter');
+  const apiWorking = await testHFAPI();
+  if (!apiWorking) {
+    console.log('[⚠️ Warning] API test failed, but proceeding anyway...');
+  }
+  // Immediately restart once on startup
+  restartSpace();
+  // Then restart every 5 minutes
+  setInterval(restartSpace, 5 * 60 * 1000);
+  console.log('[⏰ Scheduled] Space will restart every 5 minutes');
+})();
+
 const { Pool } = pg;
 
 const app = express();
@@ -110,7 +189,7 @@ async function checkPromptCensorship(prompt) {
     return false;
   } catch (e) {
     console.error('Gemini censorship API error:', e.message);
-    // fallback: treat as not censored if eror
+    // fallback: treat as not censored if error
     return false;
   }
 }