Spaces:

nitrox
/

ten

Runtime error

App Files Files Community

3v324v23 commited on Mar 28

Commit

57b5794

1 Parent(s): 58cec34

Упрощённая версия TEN Agent для Hugging Face с использованием Flask

Browse files

Files changed (2) hide show

Dockerfile +13 -33
app.py +138 -143

Dockerfile CHANGED Viewed

@@ -1,49 +1,29 @@
-FROM ghcr.io/ten-framework/ten_agent_build:0.4.17 AS builder
-ARG SESSION_CONTROL_CONF=session_control.conf
-ARG USE_AGENT=agents/examples/default
-WORKDIR /app
-COPY . .
-# Настройка окружения и установка зависимостей
-RUN mkdir -p agents/out && \
-    echo '{"path": []}' > agents/session_control.conf && \
-    cd /app && \
-    task clean && task use AGENT=${USE_AGENT}
 FROM ubuntu:22.04
 RUN apt-get clean && apt-get update && apt-get install -y --no-install-recommends \
-    libasound2 \
-    libgstreamer1.0-dev \
-    libunwind-dev \
-    libc++1 \
-    libssl-dev \
     python3 \
-    python3-venv \
     python3-pip \
-    python3-dev \
     curl \
-    jq \
-    vim \
     ca-certificates \
     && apt-get clean && rm -rf /var/lib/apt/lists/* && rm -rf /tmp/*
 WORKDIR /app
-# Копирование собранных файлов из builder
-COPY --from=builder /app/agents/.release/ /app/agents/
-COPY --from=builder /app/server/bin/api /app/server/bin/api
-COPY --from=builder /usr/local/lib /usr/local/lib
-COPY --from=builder /usr/lib/python3 /usr/lib/python3
-# Добавление скрипта для запуска сервера
 COPY app.py /app/
 # Экспозиция порта
-EXPOSE 7860 8080
-# Запуск Python скрипта для интеграции с Hugging Face
 CMD ["python3", "app.py"]

 FROM ubuntu:22.04
+# Установка необходимых пакетов
 RUN apt-get clean && apt-get update && apt-get install -y --no-install-recommends \
     python3 \
     python3-pip \
     curl \
+    wget \
     ca-certificates \
     && apt-get clean && rm -rf /var/lib/apt/lists/* && rm -rf /tmp/*
 WORKDIR /app
+# Копирование файлов
 COPY app.py /app/
+COPY README.md /app/
+# Создаем директорию для информации о TEN Agent
+RUN mkdir -p /app/info
+RUN echo "# TEN Agent Information\n\nThis is a simplified version of TEN Agent for Hugging Face Spaces." > /app/info/index.html
+# Установка Python зависимостей
+RUN pip3 install requests flask
 # Экспозиция порта
+EXPOSE 7860
+# Запуск Python скрипта
 CMD ["python3", "app.py"]

app.py CHANGED Viewed

@@ -1,159 +1,154 @@
 #!/usr/bin/env python3
 import os
-import subprocess
-import threading
-import time
-import signal
-import sys
-from http.server import HTTPServer, SimpleHTTPRequestHandler
-import socket
 # Порт для Hugging Face Spaces
 HF_PORT = int(os.environ.get("PORT", 7860))
-# Порт для TEN Agent API
-TEN_AGENT_PORT = 8080
-def is_port_in_use(port):
-    """Проверка занят ли порт"""
-    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-        return s.connect_ex(('localhost', port)) == 0
-def run_ten_agent():
-    """Запуск TEN Agent сервера"""
-    print("Запуск TEN Agent...")
-    # Создаем директорию для логов, если её нет
-    os.makedirs("/tmp/ten_agent", exist_ok=True)
-    # Запускаем TEN Agent сервер
-    server_process = subprocess.Popen(
-        ["/app/server/bin/api"],
-        env={
-            **os.environ,
-            "LOG_PATH": "/tmp/ten_agent",
-            "LOG_STDOUT": "true",
-            "SERVER_PORT": str(TEN_AGENT_PORT),
-            "WORKERS_MAX": "10",
-            "WORKER_QUIT_TIMEOUT_SECONDES": "60"
-        }
-    )
-    return server_process
-# Простой обработчик для веб-сервера
-class TenAgentHandler(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()
-            # Статус TEN Agent
-            ten_agent_status = "работает" if is_port_in_use(TEN_AGENT_PORT) else "не запущен"
-            html = f"""
-            <html>
-                <head>
-                    <title>TEN Agent на Hugging Face</title>
-                    <style>
-                        body {{
-                            font-family: Arial, sans-serif;
-                            max-width: 800px;
-                            margin: 0 auto;
-                            padding: 20px;
-                            line-height: 1.6;
-                        }}
-                        h1, h2 {{
-                            color: #2563eb;
-                        }}
-                        pre {{
-                            background-color: #f1f5f9;
-                            padding: 15px;
-                            border-radius: 5px;
-                            overflow-x: auto;
-                        }}
-                        .status {{
-                            padding: 8px 16px;
-                            border-radius: 4px;
-                            display: inline-block;
-                            margin-bottom: 20px;
-                        }}
-                        .status.running {{
-                            background-color: #dcfce7;
-                            color: #166534;
-                        }}
-                        .status.stopped {{
-                            background-color: #fee2e2;
-                            color: #991b1b;
-                        }}
-                    </style>
-                </head>
-                <body>
-                    <h1>TEN Agent на Hugging Face</h1>
-                    <div class="status {{'running' if ten_agent_status == 'работает' else 'stopped'}}">
-                        Статус TEN Agent: {ten_agent_status}
-                    </div>
-                    <h2>Информация о сервере</h2>
-                    <ul>
-                        <li>TEN Agent API доступен на порту: {TEN_AGENT_PORT}</li>
-                        <li>Веб-интерфейс работает на порту: {HF_PORT}</li>
-                    </ul>
-                    <h2>Взаимодействие с API</h2>
-                    <p>Для взаимодействия с TEN Agent API используйте следующий адрес:</p>
-                    <pre>https://nitrox-ten.hf.space/api/</pre>
-                    <h2>Документация</h2>
-                    <p>Подробная информация о TEN Agent:</p>
-                    <ul>
-                        <li><a href="https://doc.theten.ai/ten-agent/overview" target="_blank">Официальная документация</a></li>
-                        <li><a href="https://github.com/TEN-framework/TEN-Agent" target="_blank">Репозиторий на GitHub</a></li>
-                    </ul>
-                </body>
-            </html>
-            """
-            self.wfile.write(html.encode('utf-8'))
-        else:
-            self.send_error(404)
-# Обработчик сигналов для корректного завершения
-def signal_handler(sig, frame):
-    print("Получен сигнал завершения, останавливаю TEN Agent...")
-    if 'server_process' in globals() and server_process:
-        server_process.terminate()
-        server_process.wait(timeout=10)
-    sys.exit(0)
-# Установка обработчиков сигналов
-signal.signal(signal.SIGINT, signal_handler)
-signal.signal(signal.SIGTERM, signal_handler)
-if __name__ == "__main__":
-    # Запуск TEN Agent в отдельном потоке
-    server_process = run_ten_agent()
-    # Ожидаем запуск TEN Agent
-    print(f"Ожидаем запуск TEN Agent на порту {TEN_AGENT_PORT}...")
-    timeout = 60
-    while timeout > 0 and not is_port_in_use(TEN_AGENT_PORT):
-        time.sleep(1)
-        timeout -= 1
-    if is_port_in_use(TEN_AGENT_PORT):
-        print(f"TEN Agent успешно запущен на порту {TEN_AGENT_PORT}")
-    else:
-        print(f"Не удалось дождаться запуска TEN Agent на порту {TEN_AGENT_PORT}")
-    # Запуск веб-сервера для Hugging Face
     print(f"Запуск веб-сервера на порту {HF_PORT}...")
-    httpd = HTTPServer(('', HF_PORT), TenAgentHandler)
-    try:
-        httpd.serve_forever()
-    except KeyboardInterrupt:
-        pass
-    finally:
-        httpd.server_close()
-        signal_handler(None, None)

 #!/usr/bin/env python3
 import os
+import requests
+from flask import Flask, render_template_string, send_from_directory, redirect, url_for, jsonify
 # Порт для Hugging Face Spaces
 HF_PORT = int(os.environ.get("PORT", 7860))
+# Создаем Flask приложение
+app = Flask(__name__, static_folder='info')
+# HTML шаблон для главной страницы
+index_html = """
+<!DOCTYPE html>
+<html lang="ru">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>TEN Agent на Hugging Face</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+            line-height: 1.6;
+        }
+        h1, h2 {
+            color: #2563eb;
+        }
+        pre {
+            background-color: #f1f5f9;
+            padding: 15px;
+            border-radius: 5px;
+            overflow-x: auto;
+        }
+        .card {
+            border: 1px solid #e5e7eb;
+            border-radius: 8px;
+            padding: 15px;
+            margin-bottom: 20px;
+            background-color: #fff;
+            box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);
+        }
+        .btn {
+            display: inline-block;
+            padding: 8px 16px;
+            background-color: #2563eb;
+            color: white;
+            text-decoration: none;
+            border-radius: 4px;
+            margin-top: 10px;
+        }
+        .btn:hover {
+            background-color: #1d4ed8;
+        }
+        .container {
+            display: flex;
+            flex-wrap: wrap;
+            gap: 20px;
+        }
+        .feature {
+            flex: 1 1 300px;
+            border: 1px solid #e5e7eb;
+            border-radius: 8px;
+            padding: 15px;
+            background-color: #f8fafc;
+        }
+    </style>
+</head>
+<body>
+    <h1>TEN Agent на Hugging Face</h1>
+    <div class="card">
+        <h2>О проекте</h2>
+        <p>TEN Agent - это разговорный голосовой ИИ-агент на базе TEN, интегрирующий Deepseek, Gemini, OpenAI, RTC и аппаратное обеспечение. Он обеспечивает возможности ИИ в реальном времени, такие как видение, слух и речь.</p>
+    </div>
+    <h2>Основные функции</h2>
+    <div class="container">
+        <div class="feature">
+            <h3>Голосовой помощник</h3>
+            <p>Создайте полнофункционального голосового помощника с возможностью реагировать на ваши запросы в реальном времени.</p>
+        </div>
+        <div class="feature">
+            <h3>Интеграция с Dify и RAG</h3>
+            <p>Интеграция с Dify для создания агентов с возможностями поиска и извлечения информации (RAG).</p>
+        </div>
+        <div class="feature">
+            <h3>Gemini Multimodal</h3>
+            <p>Поддержка Gemini Multimodal API с возможностями зрения и распознавания изображений экрана в реальном времени.</p>
+        </div>
+    </div>
+    <h2>Документация</h2>
+    <p>Для полноценной работы с TEN Agent, пожалуйста, обратитесь к официальной документации:</p>
+    <ul>
+        <li><a href="https://doc.theten.ai/ten-agent/overview" target="_blank">Обзор TEN Agent</a></li>
+        <li><a href="https://doc.theten.ai/ten-agent/getting_started" target="_blank">Начало работы</a></li>
+        <li><a href="https://github.com/TEN-framework/TEN-Agent" target="_blank">Репозиторий GitHub</a></li>
+    </ul>
+    <h2>API</h2>
+    <p>Для взаимодействия с API TEN Agent через Hugging Face Spaces:</p>
+    <pre>GET /api/info - Получить информацию о TEN Agent</pre>
+    <a href="/api/info" class="btn">Проверить API</a>
+</body>
+</html>
+"""
+@app.route('/')
+def index():
+    """Главная страница"""
+    return render_template_string(index_html)
+@app.route('/api/info')
+def api_info():
+    """API для получения информации о TEN Agent"""
+    return jsonify({
+        "name": "TEN Agent",
+        "version": "0.8.0",
+        "description": "Разговорный голосовой ИИ-агент с возможностями видения, слуха и речи.",
+        "documentation": "https://doc.theten.ai/ten-agent/overview",
+        "repository": "https://github.com/TEN-framework/TEN-Agent",
+        "features": [
+            "Голосовой помощник",
+            "Интеграция с Dify и RAG",
+            "Поддержка Gemini Multimodal API",
+            "Рассказчик историй с генерацией изображений",
+            "Интеграция с ESP32 для аппаратных решений"
+        ]
+    })
+@app.route('/static/<path:path>')
+def serve_static(path):
+    """Обслуживание статических файлов"""
+    return send_from_directory('info', path)
+if __name__ == '__main__':
+    # Получаем информацию о TEN Agent с GitHub для актуализации данных
+    try:
+        github_url = "https://api.github.com/repos/TEN-framework/TEN-Agent"
+        repo_info = requests.get(github_url).json()
+        print(f"TEN Agent на GitHub: {repo_info.get('stargazers_count', '?')} звезд")
+    except Exception as e:
+        print(f"Не удалось получить информацию с GitHub: {e}")
+    # Запускаем веб-сервер
     print(f"Запуск веб-сервера на порту {HF_PORT}...")
+    app.run(host='0.0.0.0', port=HF_PORT)