app.py

#!/usr/bin/env python3
import os
import subprocess
import sys
import time
import json
from pathlib import Path
import signal
import threading
import shutil

def check_and_create_property_json():
    """Проверяет наличие property.json и создает его при необходимости"""
    property_path = Path("/app/agents/property.json")
    
    if not property_path.exists():
        print(f"WARNING: {property_path} не найден, создаем файл...")
        
        property_data = {
            "name": "TEN Agent Example",
            "version": "0.0.1",
            "extensions": ["openai_chatgpt"],
            "description": "A basic voice agent with OpenAI",
            "graphs": [
                {
                    "name": "Voice Agent",
                    "description": "Basic voice agent with OpenAI",
                    "file": "voice_agent.json"
                },
                {
                    "name": "Chat Agent",
                    "description": "Simple chat agent",
                    "file": "chat_agent.json"
                }
            ]
        }
        
        # Проверяем и создаем директории
        property_path.parent.mkdir(parents=True, exist_ok=True)
        
        # Записываем файл
        with open(property_path, 'w') as f:
            json.dump(property_data, f, indent=2)
            
        print(f"Файл {property_path} создан успешно")

def check_files():
    """Проверяет и выводит информацию о важных файлах"""
    files_to_check = [
        "/app/agents/property.json",
        "/app/agents/manifest.json",
        "/app/agents/voice_agent.json",
        "/app/agents/chat_agent.json",
        "/app/server/bin/api"
    ]
    
    print("\n=== Проверка критических файлов ===")
    for file_path in files_to_check:
        path = Path(file_path)
        if path.exists():
            if path.is_file():
                size = path.stat().st_size
                print(f"✅ {file_path} (размер: {size} байт)")
                
                # Если это JSON файл, выводим его содержимое
                if file_path.endswith('.json'):
                    try:
                        with open(file_path, 'r') as f:
                            content = json.load(f)
                        print(f"   Содержимое: {json.dumps(content, indent=2)}")
                    except Exception as e:
                        print(f"   Ошибка чтения JSON: {e}")
            else:
                print(f"❌ {file_path} (это директория, а не файл)")
        else:
            print(f"❌ {file_path} (файл не найден)")
    
    print("\n=== Проверка структуры директорий ===")
    print("Содержимое /app/agents:")
    subprocess.run(["ls", "-la", "/app/agents"])
    
    print("\nПроверка прав доступа:")
    subprocess.run(["stat", "/app/agents"])
    subprocess.run(["stat", "/app/agents/property.json"])

def test_api():
    """Делает запрос к API для получения списка графов"""
    import urllib.request
    import urllib.error
    
    print("\n=== Тестирование API ===")
    try:
        # Даем серверу время запуститься
        time.sleep(3)
        with urllib.request.urlopen("http://localhost:8080/graphs") as response:
            data = response.read().decode('utf-8')
            print(f"Ответ /graphs: {data}")
    except urllib.error.URLError as e:
        print(f"Ошибка запроса к API: {e}")
    except Exception as e:
        print(f"Неизвестная ошибка при запросе к API: {e}")

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
        
        # Проверяем и создаем property.json
        check_and_create_property_json()
        
        # Проверка файлов перед запуском
        check_files()
        
        # Запускаем API сервер
        print("Starting TEN-Agent API server on port 8080...")
        api_process = subprocess.Popen([str(api_binary)])
        processes.append(api_process)
        
        # Тестируем API
        test_thread = threading.Thread(target=test_api)
        test_thread.daemon = True
        test_thread.start()
        
        # Запускаем Playground UI в режиме dev на порту 7860 (порт Hugging Face)
        print("Starting Playground UI in development mode on port 7860...")
        os.environ["PORT"] = "7860"
        os.environ["AGENT_SERVER_URL"] = "http://localhost:8080"
        os.environ["NEXT_PUBLIC_EDIT_GRAPH_MODE"] = "true"  # Включаем расширенный режим редактирования
        os.environ["NEXT_PUBLIC_DISABLE_CAMERA"] = "true"  # Отключаем запрос на использование камеры
        
        playground_process = subprocess.Popen(
            ["pnpm", "dev"],
            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())