Update app.py

app.py

@@ -1,19 +1,34 @@
 import os
 import io
+import re
 import numpy as np
 from flask import Flask, render_template, request, send_file
 from PIL import Image
 import xml.etree.ElementTree as ET
 
-# Попытка импортировать moviepy для обработки видео
 try:
-    from moviepy.editor import VideoFileClip, ImageSequenceClip
+    from moviepy.editor import VideoFileClip
     HAS_MOVIEPY = True
 except ImportError:
     HAS_MOVIEPY = False
 
 app = Flask(__name__)
 
+# Рекурсивная функция для удаления жестко заданных цветов внутри SVG-файлов
+def strip_fill_colors(elem):
+    # Удаляем fill если он не "none" (none обычно используется для обводки)
+    if 'fill' in elem.attrib and elem.attrib['fill'].lower() != 'none':
+        del elem.attrib['fill']
+    
+    # Удаляем fill из инлайн-стилей (style="fill: #000; ...")
+    if 'style' in elem.attrib:
+        style_str = elem.attrib['style']
+        style_str = re.sub(r'fill:\s*(?!none)[^;]+;?', '', style_str, flags=re.IGNORECASE)
+        elem.attrib['style'] = style_str
+
+    for child in elem:
+        strip_fill_colors(child)
+
 @app.route('/')
 def index():
     return render_template('index.html')
@@ -27,13 +42,14 @@ def download_svg():
         invert = request.form.get('invert') == 'true'
         bg_color = request.form.get('bg_color', '#000000')
         
+        canvas_w = float(request.form.get('canvas_w', 1400))
+        canvas_h = float(request.form.get('canvas_h', 900))
+        
         fill_colors = request.form.getlist('fill_colors[]')
         svg_files = request.files.getlist('shape_svg[]')
         raster = request.files.get('raster_image')
         
-        canvas_w, canvas_h = (1400, 900)
         img_data = None
-        
         if raster:
             img = Image.open(raster).convert('L').resize((density, int(canvas_h / (canvas_w / density))))
             img_data = np.array(img)
@@ -50,12 +66,20 @@ def download_svg():
         parsed_patterns = []
         for i, f in enumerate(svg_files):
             try:
-                svg_root = ET.fromstring(f.read().decode('utf-8'))
+                # Читаем файл, убираем теги неймспейсов для корректной работы
+                content = f.read().decode('utf-8')
+                content = re.sub(r'\sxmlns="[^"]+"', '', content, count=1)
+                svg_root = ET.fromstring(content)
+                
+                # ОЧИЩАЕМ ИСХОДНЫЕ ЦВЕТА ПАТТЕРНА
+                strip_fill_colors(svg_root)
+                
                 parsed_patterns.append({
                     'elements': list(svg_root),
                     'color': fill_colors[i] if i < len(fill_colors) else '#ffffff'
                 })
-            except:
+            except Exception as e:
+                print("SVG Parse Error:", e)
                 continue
 
         step = canvas_w / density
@@ -91,7 +115,7 @@ def download_svg():
                 else:
                     ET.SubElement(dwg, 'rect', {
                         'x': str(x - size/2), 'y': str(y - size/2),
-                        'width': str(size), 'height': str(size), 'fill': fill_colors[0] if fill_colors else '#fff'
+                        'width': str(size), 'height': str(size), 'fill': fill_colors[0] if fill_colors else '#000'
                     })
 
         svg_io = io.BytesIO(ET.tostring(dwg))
@@ -104,47 +128,83 @@ def download_svg():
 @app.route('/process_video', methods=['POST'])
 def process_video():
     if not HAS_MOVIEPY:
-        return "MoviePy not installed on server. Run: pip install moviepy", 400
+        return "MoviePy not installed", 400
         
     try:
+        density = int(request.form.get('density', 76))
+        gap_x = float(request.form.get('gap_x', 0.2))
+        base_scale = float(request.form.get('base_scale', 1.0))
+        invert = request.form.get('invert') == 'true'
+        bg_color = request.form.get('bg_color', '#000000')
+        fallback_fill = request.form.get('fallback_fill', '#000000')
+        
+        # Получаем готовые цветные PNG паттерны с фронтенда!
+        png_files = request.files.getlist('pattern_pngs')
+        pil_patterns = [Image.open(p).convert("RGBA") for p in png_files]
+        num_p = len(pil_patterns)
+
         video_file = request.files.get('video_file')
         if not video_file:
-            return "No video file uploaded", 400
+            return "No video file", 400
 
-        # Сохраняем временно загруженное видео
         in_path = "temp_in.mp4"
         out_path = "temp_out.mp4"
         video_file.save(in_path)
-        
-        # Читаем параметры как для SVG, но применяем к каждому кадру
-        density = int(request.form.get('density', 76))
-        invert = request.form.get('invert') == 'true'
-        
         clip = VideoFileClip(in_path)
         
-        # --- Тут логика рендера кадров. 
-        # Чтобы сервер не завис, я генерирую упрощенную версию видео (эффект пикселизации/порога), 
-        # так как рендерить сотни SVG-фигур через Pillow для каждого кадра — убьет процессор.
-        def process_frame(frame):
-            # Пример базовой стилизации кадров (можно дописать рендер паттернов через PIL)
-            # Перевод в ЧБ и инверсия
-            gray = np.dot(frame[...,:3], [0.2989, 0.5870, 0.1140])
-            if invert:
-                gray = 255 - gray
+        w, h = clip.size
+        step = w / density
+        rows = int(h / step)
+        cell_size = int(step * (1 - gap_x) * base_scale)
+
+        # Ресайзим паттерны заранее под размер ячейки для скорости рендера
+        resized_patterns = [p.resize((cell_size, cell_size), Image.Resampling.LANCZOS) for p in pil_patterns]
+
+        def make_frame(get_frame, t):
+            frame = get_frame(t)
+            # Переводим кадр в ЧБ для анализа яркости
+            gray = Image.fromarray(frame).convert("L")
+            # Сжимаем кадр до размера сетки для ультра-быстрого анализа
+            gray_small = np.array(gray.resize((density, rows)))
+            
+            # Создаем чистый кадр с цветом фона
+            canvas_img = Image.new("RGB", (w, h), bg_color)
+            
+            from PIL import ImageDraw
+            draw = ImageDraw.Draw(canvas_img)
+
+            for r in range(rows):
+                for c in range(density):
+                    bright = gray_small[r, c] / 255.0
+                    if invert: bright = 1.0 - bright
+                    if bright > 0.96: continue
+                    
+                    x = int(c * step + step/2 - cell_size/2)
+                    y = int(r * step + step/2 - cell_size/2)
+                    
+                    if num_p > 0:
+                        p_idx = int(np.floor(bright * num_p))
+                        p_idx = max(0, min(num_p - 1, p_idx))
+                        pat = resized_patterns[p_idx]
+                        # Накладываем цветной PNG паттерн с поддержкой прозрачности
+                        canvas_img.paste(pat, (x, y), mask=pat)
+                    else:
+                        draw.rectangle([x, y, x + cell_size, y + cell_size], fill=fallback_fill)
             
-            # Возвращаем черно-белый массив как 3-канальный (чтобы MoviePy съел)
-            return np.stack((gray,)*3, axis=-1).astype('uint8')
+            return np.array(canvas_img)
 
-        new_clip = clip.fl_image(process_frame)
+        new_clip = clip.fl(make_frame)
         new_clip.write_videofile(out_path, codec="libx264", audio=False)
         
         clip.close()
         new_clip.close()
         os.remove(in_path)
         
-        return send_file(out_path, mimetype='video/mp4', as_attachment=True, download_name='rendered.mp4')
+        return send_file(out_path, mimetype='video/mp4', as_attachment=True, download_name='cell_clipper_render.mp4')
         
     except Exception as e:
+        import traceback
+        traceback.print_exc()
         return str(e), 500
 
 if __name__ == '__main__':