Spaces:

nitrox
/

ten

Runtime error

App Files Files Community

3v324v23 commited on Mar 28

Commit

3c7d3b0

1 Parent(s): d7f0eff

Добавлен Playground UI в контейнер для полноценного интерфейса TEN Agent

Browse files

Files changed (2) hide show

Dockerfile +15 -5
app.py +53 -19

Dockerfile CHANGED Viewed

@@ -30,6 +30,11 @@ ENV PATH=$PATH:/usr/local/go/bin
 ENV GOPATH=/go
 ENV PATH=$PATH:$GOPATH/bin
 WORKDIR /app
 # Клонирование репозитория TEN Agent (основная ветка)
@@ -46,18 +51,23 @@ RUN mkdir -p /app/server/bin && \
 # Копирование файла .env с настройками
 COPY .env /app/.env
-# Сборка TEN Agent
 RUN cd /app/server && \
     go mod tidy && \
     go mod download && \
     go build -o bin/api main.go && \
     chmod +x bin/api
-# Копирование приложения Python для запуска веб-интерфейса
 COPY app.py /app/
 # Экспозиция портов
-EXPOSE 7860 8080
-# Запуск Python скрипта для интеграции с Hugging Face
-CMD ["python3", "app.py"]

 ENV GOPATH=/go
 ENV PATH=$PATH:$GOPATH/bin
+# Установка Node.js и pnpm (для Playground)
+RUN curl -fsSL https://deb.nodesource.com/setup_18.x | bash - && \
+    apt-get install -y nodejs && \
+    npm install -g pnpm
 WORKDIR /app
 # Клонирование репозитория TEN Agent (основная ветка)
 # Копирование файла .env с настройками
 COPY .env /app/.env
+# Сборка TEN Agent сервера
 RUN cd /app/server && \
     go mod tidy && \
     go mod download && \
     go build -o bin/api main.go && \
     chmod +x bin/api
+# Сборка Playground UI
+WORKDIR /app/playground
+RUN pnpm install && \
+    NEXT_PUBLIC_EDIT_GRAPH_MODE=false pnpm build
+# Копирование приложения Python для запуска всех сервисов
 COPY app.py /app/
 # Экспозиция портов
+EXPOSE 7860 8080 3000
+# Запуск Python скрипта
+CMD ["python3", "/app/app.py"]

app.py CHANGED Viewed

@@ -4,31 +4,56 @@ import subprocess
 import sys
 import time
 from pathlib import Path
 def main():
-    # Запустить TEN-Agent
-    server_process = None
     try:
-        # Путь к исполняемому файлу API сервера
         api_binary = Path("/app/server/bin/api")
-        # Проверяем существование файла
         if not api_binary.exists():
             print(f"ERROR: API binary not found at {api_binary}", file=sys.stderr)
             return 1
         # Запускаем API сервер
-        print("Starting TEN-Agent server...")
-        server_process = subprocess.Popen([str(api_binary)])
-        # Запускаем простой HTTP сервер для отображения информации
         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')
                     self.end_headers()
                     html_content = """
@@ -36,17 +61,24 @@ def main():
                     <html>
                     <head>
                         <title>TEN Agent - Hugging Face Space</title>
                         <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; }
                         </style>
                     </head>
                     <body>
                         <h1>TEN Agent запущен успешно!</h1>
                         <div class="info">
                             <p>TEN Agent API сервер работает на порту 8080.</p>
                         </div>
                         <h2>API эндпоинты:</h2>
                         <ul>
@@ -59,14 +91,9 @@ def main():
                     </html>
                     """
-                    self.wfile.write(html_content.encode())
-                else:
-                    self.send_response(404)
-                    self.send_header('Content-type', 'text/plain')
-                    self.end_headers()
-                    self.wfile.write(b'Not Found')
-        # Запускаем HTTP сервер
         port = 7860  # Hugging Face Space обычно ожидает сервер на порту 7860
         print(f"Starting HTTP server on port {port}...")
         httpd = HTTPServer(('0.0.0.0', port), CustomHandler)
@@ -75,12 +102,19 @@ def main():
     except KeyboardInterrupt:
         print("Shutting down...")
     finally:
-        # Завершаем API сервер при выходе
-        if server_process:
-            server_process.terminate()
-            server_process.wait()
     return 0
 if __name__ == "__main__":
     sys.exit(main())

 import sys
 import time
 from pathlib import Path
+import signal
 def main():
+    processes = []
     try:
+        # Пути к исполняемым файлам
         api_binary = Path("/app/server/bin/api")
+        playground_dir = Path("/app/playground")
+        # Проверяем существование файлов
         if not api_binary.exists():
             print(f"ERROR: API binary not found at {api_binary}", file=sys.stderr)
             return 1
+        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)
+        # Запускаем простой HTTP сервер для Hugging Face
         from http.server import HTTPServer, SimpleHTTPRequestHandler
         class CustomHandler(SimpleHTTPRequestHandler):
             def do_GET(self):
+                # Перенаправляем корневой запрос на Playground
                 if self.path == '/':
+                    self.send_response(301)
+                    self.send_header('Location', 'http://localhost:3000')
+                    self.end_headers()
+                else:
                     self.send_response(200)
+                    self.send_header('Content-type', 'text/html; charset=utf-8')
                     self.end_headers()
                     html_content = """
                     <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; }
+                            .link { font-size: 18px; margin-top: 30px; text-align: center; }
+                            .link a { display: inline-block; background: #0d6efd; color: white; padding: 10px 20px; text-decoration: none; border-radius: 5px; }
                         </style>
                     </head>
                     <body>
                         <h1>TEN Agent запущен успешно!</h1>
                         <div class="info">
                             <p>TEN Agent API сервер работает на порту 8080.</p>
+                            <p>Playground UI доступен на порту 3000.</p>
+                        </div>
+                        <div class="link">
+                            <a href="http://localhost:3000">Открыть Playground UI</a>
                         </div>
                         <h2>API эндпоинты:</h2>
                         <ul>
                     </html>
                     """
+                    self.wfile.write(html_content.encode('utf-8'))
+        # Запускаем 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)
     except KeyboardInterrupt:
         print("Shutting down...")
     finally:
+        # Завершаем все процессы при выходе
+        for proc in processes:
+            try:
+                proc.terminate()
+                proc.wait(timeout=5)
+            except:
+                proc.kill()
     return 0
 if __name__ == "__main__":
+    # Корректная обработка сигналов завершения
+    signal.signal(signal.SIGINT, lambda sig, frame: sys.exit(0))
+    signal.signal(signal.SIGTERM, lambda sig, frame: sys.exit(0))
     sys.exit(main())