Spaces:

nitrox
/

ten

Runtime error

App Files Files Community

3v324v23 commited on Mar 28

Commit

db0df50

1 Parent(s): 9f18430

Восстановление Playground UI в контейнере с сохранением исправления для /go

Browse files

Files changed (2) hide show

Dockerfile +9 -56
app.py +50 -58

Dockerfile CHANGED Viewed

@@ -69,66 +69,19 @@ RUN cd /app/server && \
     go build -o bin/api main.go && \
     chmod +x bin/api
-# Создаем простой HTML-шаблон для информационной страницы
-RUN mkdir -p /app/static && \
-    echo '<!DOCTYPE html>\n\
-    <html lang="en">\n\
-    <head>\n\
-    <meta charset="UTF-8">\n\
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">\n\
-    <title>TEN Agent API</title>\n\
-    <style>\n\
-    body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }\n\
-    h1 { color: #333; }\n\
-    .container { border: 1px solid #ddd; padding: 20px; border-radius: 5px; }\n\
-    code { background-color: #f5f5f5; padding: 2px 5px; border-radius: 3px; }\n\
-    pre { background-color: #f5f5f5; padding: 10px; border-radius: 5px; overflow-x: auto; }\n\
-    .endpoint { margin-bottom: 20px; }\n\
-    .method { font-weight: bold; color: #0066cc; }\n\
-    </style>\n\
-    </head>\n\
-    <body>\n\
-    <h1>TEN Agent API Server</h1>\n\
-    <div class="container">\n\
-    <p>Сервер TEN Agent API успешно запущен!</p>\n\
-    <p>Вы можете взаимодействовать с API через следующие эндпоинты:</p>\n\
-    \n\
-    <div class="endpoint">\n\
-    <span class="method">GET</span> <code>/health</code>\n\
-    <p>Проверка состояния API сервера</p>\n\
-    </div>\n\
-    \n\
-    <div class="endpoint">\n\
-    <span class="method">GET</span> <code>/list</code>\n\
-    <p>Получение списка доступных графов и модулей</p>\n\
-    </div>\n\
-    \n\
-    <div class="endpoint">\n\
-    <span class="method">GET</span> <code>/graphs</code>\n\
-    <p>Получение списка графов</p>\n\
-    </div>\n\
-    \n\
-    <div class="endpoint">\n\
-    <span class="method">POST</span> <code>/run</code>\n\
-    <p>Запуск графа или модуля</p>\n\
-    </div>\n\
-    </div>\n\
-    \n\
-    <h2>Локальный запуск Playground UI</h2>\n\
-    <div class="container">\n\
-    <p>Для использования полноценного интерфейса TEN Agent Playground локально:</p>\n\
-    <pre>git clone https://github.com/TEN-framework/TEN-Agent.git\ncd TEN-Agent/playground\npnpm install\nAGENT_SERVER_URL=https://nitrox-ten.hf.space pnpm dev</pre>\n\
-    <p>Затем откройте <code>http://localhost:3000</code> в браузере.</p>\n\
-    </div>\n\
-    </body>\n\
-    </html>' > /app/static/index.html
 # Установка точки входа
 COPY app.py /app/app.py
 RUN chmod +x /app/app.py
-# Открываем порт
-EXPOSE 8080
-# Запускаем API сервер
 ENTRYPOINT ["python3", "/app/app.py"]

     go build -o bin/api main.go && \
     chmod +x bin/api
+# Сборка Playground UI
+ENV PNPM_HOME="/app/.pnpm-store"
+ENV PATH="$PNPM_HOME:$PATH"
+RUN cd /app/playground && \
+    pnpm install --no-frozen-lockfile && \
+    NEXT_PUBLIC_EDIT_GRAPH_MODE=false pnpm build
 # Установка точки входа
 COPY app.py /app/app.py
 RUN chmod +x /app/app.py
+# Открываем порты
+EXPOSE 7860 8080 3000
+# Запускаем API сервер и Playground
 ENTRYPOINT ["python3", "/app/app.py"]

app.py CHANGED Viewed

@@ -5,74 +5,66 @@ import sys
 import time
 from pathlib import Path
 import signal
-import http.server
-import socketserver
 import threading
-# Настройка путей
-API_BINARY = Path("/app/server/bin/api")
-STATIC_DIR = Path("/app/static")
-# Проверка наличия файлов
-if not API_BINARY.exists():
-    print(f"ERROR: API binary not found at {API_BINARY}", file=sys.stderr)
-    sys.exit(1)
-# Класс для обработки HTTP запросов
-class RequestHandler(http.server.SimpleHTTPRequestHandler):
-    def __init__(self, *args, **kwargs):
-        # Устанавливаем корневую директорию для статических файлов
-        super().__init__(*args, directory=STATIC_DIR, **kwargs)
-    def do_GET(self):
-        # Перенаправление корневого пути на статическую страницу
-        if self.path == '/':
-            self.path = '/index.html'
-        # Передача всех остальных запросов в SimpleHTTPRequestHandler
-        return super().do_GET()
-# Запуск API сервера в отдельном потоке
-def run_api_server():
-    print("Starting TEN-Agent API server on port 8080...")
-    api_process = subprocess.Popen([str(API_BINARY)])
-    return api_process
-# Запуск HTTP сервера для статических файлов
-def run_http_server():
-    try:
-        # Сначала пытаемся запустить на порте 7860 (для HuggingFace)
-        PORT = 7860
-        httpd = socketserver.TCPServer(("", PORT), RequestHandler)
-        print(f"Serving static files on port {PORT}")
-        httpd.serve_forever()
-    except OSError:
-        # Если порт 7860 занят, используем 8000
-        PORT = 8000
-        httpd = socketserver.TCPServer(("", PORT), RequestHandler)
-        print(f"Serving static files on port {PORT}")
-        httpd.serve_forever()
 def main():
-    # Запуск API сервера
-    api_process = run_api_server()
-    # Запуск HTTP сервера в отдельном потоке
-    http_thread = threading.Thread(target=run_http_server, daemon=True)
-    http_thread.start()
     try:
-        # Ожидание сигнала остановки
-        api_process.wait()
     except KeyboardInterrupt:
         print("Shutting down...")
     finally:
         # Завершение процессов
-        if api_process and api_process.poll() is None:
-            api_process.terminate()
-            api_process.wait(timeout=5)
     return 0
 if __name__ == "__main__":
     sys.exit(main())

 import time
 from pathlib import Path
 import signal
 import threading
 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_process = subprocess.Popen([str(api_binary)])
+        processes.append(api_process)
+        # Даем API серверу время на запуск
+        time.sleep(3)
+        # Запускаем Playground UI на порту 7860 (порт Hugging Face)
+        print("Starting Playground UI on port 7860...")
+        os.environ["PORT"] = "7860"
+        os.environ["AGENT_SERVER_URL"] = "http://localhost:8080"
+        playground_process = subprocess.Popen(
+            ["pnpm", "start"],
+            cwd=str(playground_dir),
+            env=os.environ
+        )
+        processes.append(playground_process)
+        # Ожидаем завершения процессов
+        for proc in processes:
+            proc.wait()
     except KeyboardInterrupt:
         print("Shutting down...")
+    except Exception as e:
+        print(f"Error: {e}", file=sys.stderr)
     finally:
         # Завершение процессов
+        for proc in processes:
+            if proc and proc.poll() is None:
+                proc.terminate()
+                try:
+                    proc.wait(timeout=5)
+                except subprocess.TimeoutExpired:
+                    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())