Spaces:

nitrox
/

ten

Runtime error

App Files Files Community

3v324v23 commited on Mar 28

Commit

2455e56

1 Parent(s): 86f3bf2

Исправлена проблема с правами Go при сборке и прокси для работы Playground UI

Browse files

Files changed (2) hide show

Dockerfile +11 -8
app.py +73 -66

Dockerfile CHANGED Viewed

@@ -22,22 +22,25 @@ RUN apt-get clean && apt-get update && apt-get install -y --no-install-recommend
     ca-certificates \
     && apt-get clean && rm -rf /var/lib/apt/lists/* && rm -rf /tmp/*
-# Установка Go 1.21
 RUN wget https://golang.org/dl/go1.21.0.linux-amd64.tar.gz && \
     tar -C /usr/local -xzf go1.21.0.linux-amd64.tar.gz && \
-    rm go1.21.0.linux-amd64.tar.gz
 ENV PATH=$PATH:/usr/local/go/bin
-ENV GOPATH=/go
 ENV PATH=$PATH:$GOPATH/bin
-# Установка Node.js 20.x (вместо 18.x) и pnpm для Playground
 RUN curl -fsSL https://deb.nodesource.com/setup_20.x | bash - && \
     apt-get install -y nodejs && \
     npm install -g pnpm
-# Создаем пользователя для запуска приложения с правильными правами
-RUN useradd -m -s /bin/bash tenuser
 WORKDIR /app
 # Клонирование репозитория TEN Agent (основная ветка)
@@ -70,7 +73,7 @@ RUN cd /app/server && \
 # Сборка Playground UI с правами пользователя
 WORKDIR /app/playground
-ENV PNPM_HOME="/app/.pnpm-store"
 ENV PATH="$PNPM_HOME:$PATH"
 RUN pnpm install --no-frozen-lockfile && \
     NEXT_PUBLIC_EDIT_GRAPH_MODE=false pnpm build

     ca-certificates \
     && apt-get clean && rm -rf /var/lib/apt/lists/* && rm -rf /tmp/*
+# Создаем пользователя для запуска приложения с правильными правами
+RUN useradd -m -s /bin/bash tenuser
+# Установка Go 1.21 с правильными правами доступа
 RUN wget https://golang.org/dl/go1.21.0.linux-amd64.tar.gz && \
     tar -C /usr/local -xzf go1.21.0.linux-amd64.tar.gz && \
+    rm go1.21.0.linux-amd64.tar.gz && \
+    mkdir -p /home/tenuser/go && \
+    chown -R tenuser:tenuser /home/tenuser/go
 ENV PATH=$PATH:/usr/local/go/bin
+ENV GOPATH=/home/tenuser/go
 ENV PATH=$PATH:$GOPATH/bin
+# Установка Node.js 20.x и pnpm для Playground
 RUN curl -fsSL https://deb.nodesource.com/setup_20.x | bash - && \
     apt-get install -y nodejs && \
     npm install -g pnpm
 WORKDIR /app
 # Клонирование репозитория TEN Agent (основная ветка)
 # Сборка Playground UI с правами пользователя
 WORKDIR /app/playground
+ENV PNPM_HOME="/home/tenuser/.pnpm-store"
 ENV PATH="$PNPM_HOME:$PATH"
 RUN pnpm install --no-frozen-lockfile && \
     NEXT_PUBLIC_EDIT_GRAPH_MODE=false pnpm build

app.py CHANGED Viewed

@@ -22,83 +22,90 @@ def main():
         if not playground_dir.exists():
             print(f"ERROR: Playground directory not found at {playground_dir}", file=sys.stderr)
             return 1
-        # Запускаем простой HTTP сервер для Hugging Face
         from http.server import HTTPServer, SimpleHTTPRequestHandler
-        class CustomHandler(SimpleHTTPRequestHandler):
             def do_GET(self):
-                if self.path == '/':
-                    self.send_response(200)
-                    self.send_header('Content-type', 'text/html; charset=utf-8')
-                    self.end_headers()
-                    html_content = """
-                    <!DOCTYPE html>
-                    <html>
-                    <head>
-                        <title>TEN Agent - Hugging Face Space</title>
-                        <meta charset="utf-8">
-                        <style>
-                            body { font-family: Arial, sans-serif; line-height: 1.6; max-width: 800px; margin: 0 auto; padding: 20px; }
-                            h1 { color: #333; }
-                            .info { background: #f8f9fa; border-left: 4px solid #28a745; padding: 15px; margin-bottom: 20px; }
-                            .endpoint { background: #e9ecef; padding: 10px; border-radius: 5px; font-family: monospace; }
-                            .api { margin-top: 30px; }
-                        </style>
-                    </head>
-                    <body>
-                        <h1>TEN Agent запущен успешно!</h1>
-                        <div class="info">
-                            <p>TEN Agent API сервер работает на порту 8080.</p>
-                            <p>В связи с ограничениями Hugging Face Space, полноценный интерфейс Playground UI не может быть запущен напрямую.</p>
-                        </div>
-                        <div class="api">
-                            <h2>API эндпоинты:</h2>
-                            <ul>
-                                <li><span class="endpoint">GET /health</span> - проверка состояния сервера</li>
-                                <li><span class="endpoint">GET /list</span> - список запущенных агентов</li>
-                                <li><span class="endpoint">GET /graphs</span> - доступные графы</li>
-                            </ul>
-                        </div>
-                        <h2>Инструкция по локальному использованию</h2>
-                        <p>Для использования полного интерфейса, подключите локальный Playground к этому API:</p>
-                        <ol>
-                            <li>Клонируйте репозиторий: <code>git clone https://github.com/TEN-framework/TEN-Agent.git</code></li>
-                            <li>Перейдите в директорию playground: <code>cd TEN-Agent/playground</code></li>
-                            <li>Установите зависимости: <code>pnpm install</code></li>
-                            <li>Запустите Playground с подключением к API: <code>AGENT_SERVER_URL=https://nitrox-ten.hf.space pnpm dev</code></li>
-                            <li>Откройте в браузере: <code>http://localhost:3000</code></li>
-                        </ol>
-                        <p>См. <a href="https://github.com/TEN-framework/TEN-Agent" target="_blank">документацию TEN Agent</a> для получения дополнительной информации.</p>
-                    </body>
-                    </html>
-                    """
-                    self.wfile.write(html_content.encode('utf-8'))
-                elif self.path.startswith('/health') or self.path.startswith('/list') or self.path.startswith('/graphs'):
-                    # Проксирование API запросов к API серверу
-                    self.send_response(301)
-                    self.send_header('Location', f'http://localhost:8080{self.path}')
                     self.end_headers()
-                else:
-                    self.send_response(404)
                     self.send_header('Content-type', 'text/plain; charset=utf-8')
                     self.end_headers()
-                    self.wfile.write(b'Not Found')
-        # Запускаем API сервер
-        print("Starting TEN-Agent API server on port 8080...")
-        api_server_process = subprocess.Popen([str(api_binary)])
-        processes.append(api_server_process)
-        # Запускаем HTTP сервер для Hugging Face
-        port = 7860  # Hugging Face Space обычно ожидает сервер на порту 7860
-        print(f"Starting HTTP server on port {port}...")
-        httpd = HTTPServer(('0.0.0.0', port), CustomHandler)
         httpd.serve_forever()
     except KeyboardInterrupt:

         if not playground_dir.exists():
             print(f"ERROR: Playground directory not found at {playground_dir}", file=sys.stderr)
             return 1
+        # Запускаем API сервер
+        print("Starting TEN-Agent API server on port 8080...")
+        api_server_process = subprocess.Popen([str(api_binary)])
+        processes.append(api_server_process)
+        # Даем время API серверу запуститься
+        time.sleep(3)
+        # Запускаем Playground UI
+        print("Starting Playground UI on port 3000...")
+        playground_env = os.environ.copy()
+        playground_env["AGENT_SERVER_URL"] = "http://localhost:8080"  # Подключаемся к локальному API серверу
+        playground_process = subprocess.Popen(
+            ["pnpm", "start", "--", "-p", "3000"],
+            cwd=str(playground_dir),
+            env=playground_env
+        )
+        processes.append(playground_process)
+        # Запускаем прокси для Hugging Face
         from http.server import HTTPServer, SimpleHTTPRequestHandler
+        import http.client
+        import socketserver
+        class ProxyHandler(SimpleHTTPRequestHandler):
             def do_GET(self):
+                # Направляем запросы к Playground UI
+                if self.path == '/' or self.path.startswith('/about') or self.path.startswith('/_next') or self.path.startswith('/favicon.ico'):
+                    self._proxy_request('localhost', 3000, self.path)
+                # Направляем API запросы к API серверу
+                elif self.path.startswith('/health') or self.path.startswith('/list') or self.path.startswith('/graphs'):
+                    self._proxy_request('localhost', 8080, self.path)
+                else:
+                    # По умолчанию отправляем на Playground
+                    self._proxy_request('localhost', 3000, self.path)
+            def do_POST(self):
+                # Все POST запросы направляем в API
+                self._proxy_request('localhost', 8080, self.path)
+            def _proxy_request(self, host, port, path):
+                try:
+                    # Создаем соединение с целевым сервером
+                    conn = http.client.HTTPConnection(host, port)
+                    # Копируем заголовки запроса
+                    headers = {}
+                    for header in self.headers:
+                        headers[header] = self.headers[header]
+                    # Получаем данные запроса
+                    content_length = int(self.headers.get('Content-Length', 0))
+                    body = self.rfile.read(content_length) if content_length > 0 else None
+                    # Отправляем запрос на целевой сервер
+                    conn.request(self.command, path, body=body, headers=headers)
+                    resp = conn.getresponse()
+                    # Отправляем ответ клиенту
+                    self.send_response(resp.status)
+                    for header in resp.headers:
+                        self.send_header(header, resp.headers[header])
                     self.end_headers()
+                    # Отправляем тело ответа
+                    self.wfile.write(resp.read())
+                    conn.close()
+                except Exception as e:
+                    print(f"Proxy error: {e}", file=sys.stderr)
+                    self.send_response(500)
                     self.send_header('Content-type', 'text/plain; charset=utf-8')
                     self.end_headers()
+                    self.wfile.write(f"Proxy error: {e}".encode('utf-8'))
+        # Запускаем HTTP прокси-сервер
+        port = 7860  # Hugging Face Space ожидает сервер на этом порту
+        print(f"Starting proxy server on port {port}, forwarding to Playground UI and API...")
+        # Разрешаем повторное использование адреса и порта
+        class ReuseAddressServer(socketserver.TCPServer):
+            allow_reuse_address = True
+        httpd = ReuseAddressServer(('0.0.0.0', port), ProxyHandler)
         httpd.serve_forever()
     except KeyboardInterrupt: