Update index.js

index.js

@@ -14,14 +14,20 @@ app.all('*', async (req, res) => {
   if (req.url === '/') return;
 
   try {
-    const url = `https://generativelanguage.googleapis.com${req.url}`;
+    // Убираем первый слэш и декодируем URL
+    const targetUrl = decodeURIComponent(req.url.substring(1));
     
+    // Проверяем что это валидный URL
+    if (!targetUrl.startsWith('http')) {
+      return res.status(400).send('Invalid URL');
+    }
+
     // Копируем и фильтруем заголовки
     const headers = {...req.headers};
     delete headers['host'];
     delete headers['content-length'];
 
-    const response = await fetch(url, {
+    const response = await fetch(targetUrl, {
       method: req.method,
       headers,
       body: req.method !== 'GET' ? req.body : undefined
@@ -34,13 +40,18 @@ app.all('*', async (req, res) => {
       return;
     }
 
+    // Копируем все заголовки из ответа
+    response.headers.forEach((value, key) => {
+      res.setHeader(key, value);
+    });
+
     // Regular response
     const data = await response.buffer();
     res.status(response.status);
-    res.set('Content-Type', response.headers.get('content-type'));
     res.send(data);
 
   } catch (error) {
+    console.error('Proxy error:', error);
     res.status(500).send(error.message);
   }
 });