Initial commit

app.py

@@ -0,0 +1,98 @@
+import streamlit as st
+import numpy as np
+from video_processing import enhance_video
+from config import EXAMPLES
+
+st.set_page_config(
+    page_title="Сверточный Видоусилитель",
+    page_icon="🎬",
+    layout="wide"
+)
+
+
+def main():
+    st.title("🎬 Сверточный Видоусилитель")
+
+    if 'current_filter' not in st.session_state:
+        st.session_state.current_filter = "Повышение резкости"
+        st.session_state.kernel = EXAMPLES["Повышение резкости"]
+
+    # Секция загрузки файла
+    uploaded_file = st.file_uploader("📤 Загрузите видео файл", type=["mp4", "avi", "mov"])
+
+    # Панель управления фильтрами
+    with st.sidebar:
+        st.header("⚙ Настройки свертки")
+
+        # Отображение текущего фильтра
+        st.markdown(f"**Активный фильтр:** `{st.session_state.current_filter}`")
+        kernel_input = st.text_input(
+            "Матрица 3x3 (9 чисел через запятую):",
+            value=", ".join(map(str, st.session_state.kernel.flatten())),
+            help="Пример: -1, -1, -1, -1, 9, -1, -1, -1, -1",
+            key="kernel_input"
+        )
+
+        st.subheader("📚 Пресеты фильтров")
+        cols = st.columns(2)
+        for idx, (name, example) in enumerate(EXAMPLES.items()):
+            with cols[idx % 2]:
+                is_active = name == st.session_state.current_filter
+                btn_label = f"✅ {name}" if is_active else f"⚙️ {name}"
+                if st.button(
+                        btn_label,
+                        help=f"Матрица: {example.tolist()}",
+                        use_container_width=True
+                ):
+                    st.session_state.current_filter = name
+                    st.session_state.kernel = example
+                    st.rerun()
+
+    if uploaded_file:
+        if st.button("🚀 ПРИМЕНИТЬ СВЕРТОЧНЫЙ ФИЛЬТР", use_container_width=True, type="primary"):
+            try:
+                if not kernel_input.strip():
+                    st.error("❌ Выберите фильтр или введите свою матрицу!")
+                    return
+
+                kernel_values = [float(x.strip()) for x in kernel_input.split(',')]
+
+                if len(kernel_values) != 9:
+                    st.error("❌ Требуется ровно 9 значений через запятую!")
+                    return
+
+                if all(abs(x) < 0.0001 for x in kernel_values):
+                    st.error("❌ Матрица не может быть нулевой!")
+                    return
+
+                kernel = np.array(kernel_values).reshape(3, 3)
+                st.session_state.kernel = kernel
+                st.session_state.current_filter = "Пользовательский"
+
+                with st.spinner("🔎 Применяем сверточный фильтр..."):
+                    processed_path = enhance_video(uploaded_file, kernel)
+
+                col1, col2 = st.columns(2)
+                with col1:
+                    st.subheader("Исходное видео")
+                    st.video(uploaded_file)
+
+                with col2:
+                    st.subheader("Результат обработки")
+                    st.video(processed_path)
+
+                    with open(processed_path, "rb") as f:
+                        st.download_button(
+                            label="💾 Скачать результат",
+                            data=f,
+                            file_name="conv_video.mp4",
+                            mime="video/mp4",
+                            use_container_width=True
+                        )
+
+            except Exception as e:
+                st.error(f"⚠️ Критическая ошибка: {str(e)}")
+
+
+if __name__ == "__main__":
+    main()

config.py

@@ -0,0 +1,47 @@
+import numpy as np
+
+EXAMPLES = {
+    # Базовые фильтры
+    "Повышение резкости": np.array([
+        [-1, -1, -1],
+        [-1, 9, -1],
+        [-1, -1, -1]
+    ]),
+    "Гауссово размытие": np.array([
+        [0.0625, 0.125, 0.0625],
+        [0.125, 0.25, 0.125],
+        [0.0625, 0.125, 0.0625]
+    ]),
+    "Детектор краёв": np.array([
+        [0, 1, 0],
+        [1, -4, 1],
+        [0, 1, 0]
+    ]),
+
+    # Новые фильтры
+    "Эффект тиснения": np.array([
+        [-2, -1, 0],
+        [-1, 1, 1],
+        [0, 1, 2]
+    ]),
+    "Контурная резкость": np.array([
+        [0, -1, 0],
+        [-1, 5, -1],
+        [0, -1, 0]
+    ]),
+    "Точечное размытие": np.array([
+        [0, 0.2, 0],
+        [0.2, 0.2, 0.2],
+        [0, 0.2, 0]
+    ]),
+    "Повышение контраста": np.array([
+        [0, -1, 0],
+        [-1, 5, -1],
+        [0, -1, 0]
+    ]),
+    "Эффект рельефа": np.array([
+        [1, 1, 0],
+        [1, 0, -1],
+        [0, -1, -1]
+    ])
+}

requirements.txt

@@ -0,0 +1,54 @@
+altair==5.5.0
+attrs==25.1.0
+blinker==1.9.0
+cachetools==5.5.1
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+decorator==5.1.1
+filelock==3.17.0
+fsspec==2025.2.0
+gitdb==4.0.12
+GitPython==3.1.44
+idna==3.10
+imageio==2.37.0
+imageio-ffmpeg==0.6.0
+Jinja2==3.1.5
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+mdurl==0.1.2
+moviepy==2.1.2
+mpmath==1.3.0
+narwhals==1.25.1
+networkx==3.4.2
+numpy==2.2.2
+opencv-python==4.11.0.86
+packaging==24.2
+pandas==2.2.3
+pillow==10.4.0
+proglog==0.1.10
+protobuf==5.29.3
+pyarrow==19.0.0
+pydeck==0.9.1
+Pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+pytz==2025.1
+referencing==0.36.2
+requests==2.32.3
+rich==13.9.4
+rpds-py==0.22.3
+six==1.17.0
+smmap==5.0.2
+streamlit==1.42.0
+sympy==1.13.1
+tenacity==9.0.0
+toml==0.10.2
+torch==2.6.0
+tornado==6.4.2
+tqdm==4.67.1
+typing_extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0

video_processing.py

@@ -0,0 +1,55 @@
+import cv2
+import numpy as np
+import tempfile
+import os
+from moviepy import VideoFileClip
+
+
+def enhance_video(video_file, kernel):
+    # Сохранение во временный файл
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_input:
+        temp_input.write(video_file.getbuffer())
+        input_path = temp_input.name
+
+    # Обработка видео
+    cap = cv2.VideoCapture(input_path)
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    fps = cap.get(cv2.CAP_PROP_FPS)
+
+    # Подготовка выходного файла
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_output:
+        output_path = temp_output.name
+        out = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (width, height))
+
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret:
+                break
+            processed_frame = cv2.filter2D(frame, -1, kernel)
+            out.write(processed_frame)
+
+        cap.release()
+        out.release()
+
+    # Микширование аудио
+    final_output = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name
+    video_clip = VideoFileClip(output_path)
+    audio_clip = VideoFileClip(input_path).audio
+
+    if audio_clip:
+        video_clip.audio = audio_clip
+
+    video_clip.write_videofile(
+        final_output,
+        codec='libx264',
+        audio_codec='aac',
+        logger=None
+    )
+
+    # Очистка временных файлов
+    for path in [input_path, output_path]:
+        if os.path.exists(path):
+            os.remove(path)
+
+    return final_output