Revert "Call HF directly"

This reverts commit 65fafc532ea43f954bd54f302bc19ba0d4c06ae3.

auth-service.js

@@ -1,14 +1,27 @@
 const express = require('express');
+const jwt = require('jsonwebtoken');
+
 const app = express();
 const port = 4001;
 
-app.get('/hf-token', (req, res) => {
-  const token = process.env.HF_API_KEY;
-  if (!token) {
-    console.error('HF_API_KEY environment variable is not set');
-    return res.status(500).json({ error: 'Token not configured' });
+const privateKey = process.env.JWT_PRIVATE_KEY;
+if (!privateKey) {
+  console.error('JWT_PRIVATE_KEY environment variable is required');
+  process.exit(1);
+}
+
+app.get('/auth-token', (req, res) => {
+  try {
+    const token = jwt.sign({}, privateKey, {
+      algorithm: process.env.JWT_ALGORITHM || 'RS256',
+      expiresIn: '1h'
+    });
+    
+    res.json({ token });
+  } catch (error) {
+    console.error('Error generating token:', error);
+    res.status(500).json({ error: 'Failed to generate token' });
   }
-  res.json({ token });
 });
 
 app.listen(port, () => {

document-authoring.js

@@ -20,28 +20,14 @@ async function getAuthToken() {
   }
 }
 
-async function getHFToken() {
-  try {
-    const response = await fetch('/api/hf-token');
-    if (!response.ok) {
-      throw new Error('Failed to fetch HF token');
-    }
-    const { token } = await response.json();
-    return token;
-  } catch (error) {
-    console.error('Error getting HF token:', error);
-    throw error;
-  }
-}
-
 async function checkServicesStatus() {
   try {
-    const token = await getHFToken();
-    const response = await fetch('https://api-inference.huggingface.co/models/google/gemma-2-2b-it/v1/chat/completions', {
+    const token = await getAuthToken();
+    const response = await fetch('/inference/api/v1/chat/completions', {
       method: 'POST',
       headers: {
         'Content-Type': 'application/json',
-        'Authorization': 'Bearer ' + token
+        'Authorization': `Token token=${token}`
       },
       body: JSON.stringify({
         model: 'gemma-2b',
@@ -75,6 +61,7 @@ async function checkServicesStatus() {
         statusIndicator.style.color = '#dc3545';
       } else {
         statusIndicator.innerHTML = `<span class="spinner"></span> Initializing AI services...`;
+        // Schedule next check only if we haven't exceeded retries
         setTimeout(checkServicesStatus, 2000);
       }
     }
@@ -147,12 +134,12 @@ script.onload = async () => {
 
   async function translate(content, targetLang, sourceLang = 'English') {
     try {
-      const token = await getHFToken();
-      const response = await fetch('https://api-inference.huggingface.co/models/google/gemma-2-2b-it/v1/chat/completions', {
+      const token = await getAuthToken();
+      const response = await fetch('/inference/api/v1/chat/completions', {
         method: 'POST',
         headers: {
           'Content-Type': 'application/json',
-          'Authorization': 'Bearer ' + token
+          'Authorization': `Token token=${token}`
         },
         body: JSON.stringify({
           messages: [
@@ -175,6 +162,7 @@ ${targetLang}: "Las empresas usan Nutrient para..."
 ${content}`
             }
           ],
+          model: "gemma-2b",
           stream: false
         })
       });

nginx.conf

@@ -29,11 +29,5 @@ http {
             proxy_set_header Host $host;
             proxy_set_header X-Real-IP $remote_addr;
         }
-
-        location /api/hf-token {
-            proxy_pass http://localhost:4001/hf-token;
-            proxy_set_header Host $host;
-            proxy_set_header X-Real-IP $remote_addr;
-        }
     }
 }