Добавлен прокси для статического Playground UI, работающий с HuggingFace Space

fallback.py

@@ -558,6 +558,196 @@ def run_proxy_server():
     
     return httpd
 
+def run_playground_proxy():
+    """Запускает прокси-сервер для обслуживания Playground UI и проксирования запросов к API"""
+    import http.server
+    import socketserver
+    from urllib.parse import urljoin, urlparse
+    import urllib.request
+    import urllib.error
+    
+    class PlaygroundProxyHandler(http.server.SimpleHTTPRequestHandler):
+        def log_message(self, format, *args):
+            logger.info(f"PLAYGROUND-PROXY: {format % args}")
+            
+        def do_GET(self):
+            # Проксирование запросов к API
+            if self.path.startswith('/api/'):
+                api_path = self.path[5:]  # Удаляем '/api/' из пути
+                api_url = f"http://localhost:8080/{api_path}"
+                
+                try:
+                    logger.info(f"Proxying GET request to API: {api_url}")
+                    with urllib.request.urlopen(api_url) as response:
+                        data = response.read()
+                        self.send_response(response.status)
+                        
+                        # Копируем все заголовки из ответа API
+                        for header, value in response.getheaders():
+                            if header.lower() not in ('transfer-encoding', 'connection'):
+                                self.send_header(header, value)
+                                
+                        # Устанавливаем CORS заголовки
+                        self.send_header('Access-Control-Allow-Origin', '*')
+                        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
+                        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
+                        
+                        self.end_headers()
+                        self.wfile.write(data)
+                except urllib.error.URLError as e:
+                    logger.error(f"Error proxying GET request to API: {e}")
+                    self.send_error(502, f"Error proxying to API: {e}")
+                return
+                
+            # Проксирование запросов к серверу графического редактора
+            elif self.path.startswith('/designer/'):
+                designer_path = self.path[10:]  # Удаляем '/designer/' из пути
+                designer_url = f"http://localhost:49483/{designer_path}"
+                
+                try:
+                    logger.info(f"Proxying GET request to designer: {designer_url}")
+                    with urllib.request.urlopen(designer_url) as response:
+                        data = response.read()
+                        self.send_response(response.status)
+                        
+                        # Копируем все заголовки из ответа дизайнера
+                        for header, value in response.getheaders():
+                            if header.lower() not in ('transfer-encoding', 'connection'):
+                                self.send_header(header, value)
+                        
+                        # Устанавливаем CORS заголовки
+                        self.send_header('Access-Control-Allow-Origin', '*')
+                        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
+                        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
+                        
+                        self.end_headers()
+                        self.wfile.write(data)
+                except urllib.error.URLError as e:
+                    logger.error(f"Error proxying GET request to designer: {e}")
+                    self.send_error(502, f"Error proxying to designer: {e}")
+                return
+                
+            # Перенаправление корневого пути на Playground UI
+            elif self.path == '/' or self.path == '':
+                self.send_response(302)
+                self.send_header('Location', 'https://ten-framework.github.io/TEN-Playground-Static/')
+                self.end_headers()
+                return
+                
+            # Для всех остальных запросов пытаемся обслужить статический файл
+            else:
+                self.send_error(404, "File Not Found")
+                
+        def do_POST(self):
+            # Проксирование POST запросов к API
+            if self.path.startswith('/api/'):
+                api_path = self.path[5:]  # Удаляем '/api/' из пути
+                api_url = f"http://localhost:8080/{api_path}"
+                
+                content_length = int(self.headers.get('Content-Length', 0))
+                post_data = self.rfile.read(content_length) if content_length > 0 else b''
+                
+                try:
+                    logger.info(f"Proxying POST request to API: {api_url}")
+                    request = urllib.request.Request(
+                        api_url,
+                        data=post_data,
+                        headers={k: v for k, v in self.headers.items() if k.lower() not in ('host', 'content-length')},
+                        method='POST'
+                    )
+                    
+                    with urllib.request.urlopen(request) as response:
+                        data = response.read()
+                        self.send_response(response.status)
+                        
+                        # Копируем все заголовки из ответа API
+                        for header, value in response.getheaders():
+                            if header.lower() not in ('transfer-encoding', 'connection'):
+                                self.send_header(header, value)
+                                
+                        # Устанавливаем CORS заголовки
+                        self.send_header('Access-Control-Allow-Origin', '*')
+                        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
+                        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
+                        
+                        self.end_headers()
+                        self.wfile.write(data)
+                except urllib.error.URLError as e:
+                    logger.error(f"Error proxying POST request to API: {e}")
+                    self.send_error(502, f"Error proxying to API: {e}")
+            
+            # Проксирование запросов к серверу графического редактора
+            elif self.path.startswith('/designer/'):
+                designer_path = self.path[10:]  # Удаляем '/designer/' из пути
+                designer_url = f"http://localhost:49483/{designer_path}"
+                
+                content_length = int(self.headers.get('Content-Length', 0))
+                post_data = self.rfile.read(content_length) if content_length > 0 else b''
+                
+                try:
+                    logger.info(f"Proxying POST request to designer: {designer_url}")
+                    request = urllib.request.Request(
+                        designer_url,
+                        data=post_data,
+                        headers={k: v for k, v in self.headers.items() if k.lower() not in ('host', 'content-length')},
+                        method='POST'
+                    )
+                    
+                    with urllib.request.urlopen(request) as response:
+                        data = response.read()
+                        self.send_response(response.status)
+                        
+                        # Копируем все заголовки из ответа дизайнера
+                        for header, value in response.getheaders():
+                            if header.lower() not in ('transfer-encoding', 'connection'):
+                                self.send_header(header, value)
+                                
+                        # Устанавливаем CORS заголовки
+                        self.send_header('Access-Control-Allow-Origin', '*')
+                        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
+                        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
+                        
+                        self.end_headers()
+                        self.wfile.write(data)
+                except urllib.error.URLError as e:
+                    logger.error(f"Error proxying POST request to designer: {e}")
+                    self.send_error(502, f"Error proxying to designer: {e}")
+            else:
+                self.send_error(404, "Not Found")
+                
+        def do_OPTIONS(self):
+            self.send_response(200)
+            self.send_header('Access-Control-Allow-Origin', '*')
+            self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
+            self.send_header('Access-Control-Allow-Headers', 'Content-Type, Authorization')
+            self.end_headers()
+    
+    # Запуск прокси-сервера на порту 7860 (стандартный порт HF Space)
+    port = 7860
+    httpd = socketserver.ThreadingTCPServer(("", port), PlaygroundProxyHandler)
+    logger.info(f"Playground proxy server started on port {port}")
+    logger.info(f"Access Playground at: https://nitrox-ten.hf.space/")
+    
+    # В отдельном потоке запускаем проверку доступности API сервера
+    def check_api_availability():
+        while True:
+            try:
+                # Проверка /graphs API
+                with urllib.request.urlopen("http://localhost:8080/graphs") as response:
+                    if response.status == 200:
+                        data = response.read().decode('utf-8')
+                        logger.info(f"API /graphs endpoint is available: {data}")
+            except Exception as e:
+                logger.warning(f"API check failed: {e}")
+            
+            time.sleep(30)  # Проверяем каждые 30 секунд
+    
+    api_check_thread = threading.Thread(target=check_api_availability, daemon=True)
+    api_check_thread.start()
+    
+    # Запускаем сервер в основном потоке
+    httpd.serve_forever()
+
 def run_playground():
     """Запуск Playground UI"""
     logger.info("Starting Playground UI in development mode")
@@ -659,18 +849,33 @@ def run_playground():
             time.sleep(3)
             if playground_process.poll() is not None:
                 logger.error(f"All Playground UI launch methods failed")
-                raise RuntimeError(f"All Playground UI launch methods failed")
+                logger.info("Starting playground proxy as fallback")
+                
+                # Запускаем прокси-сервер для Playground
+                playground_thread = threading.Thread(target=run_playground_proxy, daemon=True)
+                playground_thread.start()
+                
+                # Создаем заглушку для процесса
+                playground_process = subprocess.Popen(
+                    ["tail", "-f", "/dev/null"],
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.PIPE,
+                )
         else:
-            logger.info("Skipping Playground launch as it's not critical for API functionality")
-            logger.info("API server and proxy server will continue running")
+            logger.info("Starting playground proxy as fallback")
+            
+            # Запускаем прокси-сервер для Playground
+            playground_thread = threading.Thread(target=run_playground_proxy, daemon=True)
+            playground_thread.start()
+            
             # Создаем заглушку для процесса
             playground_process = subprocess.Popen(
-                ["tail", "-f", "/dev/null"],  # Команда, которая никогда не завершится
+                ["tail", "-f", "/dev/null"],
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,
             )
     
-    logger.info("Playground UI started successfully or skipped")
+    logger.info("Playground UI started successfully or proxy is running")
     return playground_process
 
 def main():