Upload 3 files

Dockerfile

@@ -0,0 +1,34 @@
+FROM python:3.10
+
+# Установка системных зависимостей
+RUN apt-get update && apt-get install -y \
+    libgirepository1.0-dev \
+    gobject-introspection \
+    pkg-config \
+    libdbus-1-dev \
+    libcairo2-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# Создание пользователя
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+RUN pwd
+# Установка рабочей директории
+WORKDIR /content/PolGen
+
+# Обновление pip
+RUN pip install pip==23.1
+
+# Копирование и установка зависимостей
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --ignore-installed --no-cache-dir --upgrade -r requirements.txt
+
+# Клонирование репозитория
+RUN rm -rf /content/PolGen
+RUN git clone --depth 1 https://github.com/Bebra777228/PolGen-RVC --branch v1.2.0-fix --single-branch /content/PolGen
+COPY --chown=user ./app.py /content/PolGen/app2.py
+RUN ls /content/PolGen
+RUN cd /content/PolGen
+EXPOSE 7860
+CMD ["python", "app2.py"]

app.py

@@ -0,0 +1,252 @@
+import gradio as gr
+import os
+import asyncio
+import shutil
+from datetime import datetime
+import urllib.request
+import zipfile
+import gdown
+import requests
+import logging
+
+# Настройка логирования
+logging.basicConfig(level=logging.INFO)
+
+# Константы
+f0_min = 50
+f0_max = 1100
+rvc_models_dir = './models'
+input_folder = './input'
+output_folder = './output'
+
+# Создание необходимых директорий
+os.makedirs(rvc_models_dir, exist_ok=True)
+os.makedirs(input_folder, exist_ok=True)
+os.makedirs(output_folder, exist_ok=True)
+
+# Общие функции
+async def run_command_async(command):
+    """Асинхронный запуск команд."""
+    process = await asyncio.create_subprocess_shell(
+        command,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE
+    )
+    stdout, stderr = await process.communicate()
+    if process.returncode != 0:
+        logging.error(f"Error: {stderr.decode()}")
+    return stdout.decode()
+
+def get_models_list():
+    """Получение списка моделей."""
+    models = []
+    if os.path.exists(rvc_models_dir):
+        models = [d for d in os.listdir(rvc_models_dir) if os.path.isdir(os.path.join(rvc_models_dir, d))]
+    return models
+
+# Функции для первой вкладки
+def process_uploaded_files(files):
+    """Обработка загруженных файлов."""
+    if os.path.exists(input_folder):
+        shutil.rmtree(input_folder)
+    os.makedirs(input_folder, exist_ok=True)
+    
+    for file in files:
+        shutil.copy(file.name, input_folder)
+    return f"Uploaded {len(files)} files to {input_folder}"
+
+async def convert_voice(
+    voicemodel_name,
+    pitch_vocal,
+    method_pitch,
+    hop_length,
+    index_rate,
+    filter_radius,
+    rms,
+    protect,
+    output_format
+):
+    """Конвертация голоса."""
+    current_time = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+    output_ai_vocals_folder = os.path.join(output_folder, current_time)
+    os.makedirs(output_ai_vocals_folder, exist_ok=True)
+
+    output_test = {
+        "mp3": "-ab 320k",
+        "wav": "-c:a pcm_s32le",
+        "flac": "-c:a flac -af aformat=s16"
+    }[output_format]
+
+    for filename in os.listdir(input_folder):
+        if filename.endswith(("wav", "mp3", "flac")):
+            input_path = os.path.join(input_folder, filename)
+            base_name = os.path.splitext(filename)[0]
+            
+            output_filename = f"mono_{base_name}_{current_time}_{method_pitch}_{pitch_vocal}.{output_format}"
+            output_path = os.path.join(output_ai_vocals_folder, output_filename)
+
+            # Выполнение команды RVC
+            await run_command_async(
+                f"python3 -m rvc.cli.rvc_cli -i \"{input_path}\" -m \"{voicemodel_name}\" -p {pitch_vocal} "
+                f"-ir {index_rate} -fr {filter_radius} -rms {rms} -f0 \"{method_pitch}\" "
+                f"-hop {hop_length} -pro {protect} -f0min {f0_min} -f0max {f0_max} -f \"wav\""
+            )
+            
+            # Конвертация в нужный формат
+            await run_command_async(
+                f"ffmpeg -y -i ./output/Voice_Converted.wav "
+                f"-vn -ar 44100 -ac 1 {output_test} \"{output_path}\""
+            )
+
+    return [os.path.join(output_ai_vocals_folder, f) for f in os.listdir(output_ai_vocals_folder) if f.endswith(output_format)]
+
+# Функции для второй вкладки
+def extract_zip(extraction_folder, zip_name):
+    """Распаковка архива."""
+    os.makedirs(extraction_folder, exist_ok=True)
+    with zipfile.ZipFile(zip_name, 'r') as zip_ref:
+        zip_ref.extractall(extraction_folder)
+    os.remove(zip_name)
+
+    model_file = next((f for f in os.listdir(extraction_folder) if f.endswith('.pth')), None)
+    index_file = next((f for f in os.listdir(extraction_folder) if f.endswith('.index')), None)
+
+    if model_file:
+        os.rename(os.path.join(extraction_folder, model_file), 
+                 os.path.join(extraction_folder, model_file))
+    if index_file:
+        os.rename(os.path.join(extraction_folder, index_file), 
+                 os.path.join(extraction_folder, index_file))
+
+async def download_model(url, dir_name):
+    """Скачивание модели."""
+    try:
+        zip_path = os.path.join(rvc_models_dir, f"{dir_name}.zip")
+        extraction_path = os.path.join(rvc_models_dir, dir_name)
+        
+        if os.path.exists(extraction_path):
+            return f"Error: Directory {dir_name} already exists!"
+
+        if 'drive.google.com' in url:
+            file_id = url.split("file/d/")[1].split("/")[0] if "file/d/" in url else url.split("id=")[1].split("&")[0]
+            gdown.download(id=file_id, output=zip_path, quiet=False)
+        elif 'huggingface.co' in url:
+            urllib.request.urlretrieve(url, zip_path)
+        elif 'pixeldrain.com' in url:
+            file_id = url.split("pixeldrain.com/u/")[1]
+            response = requests.get(f"https://pixeldrain.com/api/file/{file_id}")
+            with open(zip_path, 'wb') as f:
+                f.write(response.content)
+        
+        extract_zip(extraction_path, zip_path)
+        return f"Model {dir_name} successfully installed!"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Создание интерфейса
+with gr.Blocks() as demo:
+    gr.Markdown("# VBach Lite WEBUI")
+    
+    with gr.Tabs():
+        # Первая вкладка
+        with gr.TabItem("Замена вокала"):
+            with gr.Row():
+                with gr.Column():
+                    file_input = gr.File(file_count="multiple", label="Загрузить один или несколько файлов")
+                    upload_status = gr.Textbox(label="Статус загрузки")
+                    file_input.upload(
+                        fn=process_uploaded_files,
+                        inputs=file_input,
+                        outputs=upload_status
+                    )
+                    
+                    with gr.Accordion("Настройки RVC:", open=True):
+                        voicemodel_name = gr.Dropdown(
+                            choices=get_models_list(), 
+                            label="Имя модели", 
+                            value="senko" if "senko" in get_models_list() else None
+                        )
+                        refresh_btn = gr.Button("Обновить")
+                        
+                        pitch_vocal = gr.Slider(-48, 48, value=0, step=12, label="Высота тона")
+                        method_pitch = gr.Dropdown(
+                            ["fcpe", "rmvpe+", "mangio-crepe"],
+                            value="rmvpe+",
+                            label="Метод извлечения тона"
+                        )
+                        hop_length = gr.Slider(0, 255, value=73, step=1, label="Длина шага для mangio-crepe")
+                        index_rate = gr.Slider(0, 1, value=1, step=0.05, label="ИИ-акцент")
+                        filter_radius = gr.Slider(0, 7, value=7, step=1, label="Радиус фильтра")
+                        rms = gr.Slider(0, 1, value=0, step=0.1, label="Нормализация")
+                        protect = gr.Slider(0, 0.5, value=0.35, step=0.05, label="Защита согласных")
+                        output_format = gr.Dropdown(
+                            ["flac", "wav", "mp3"],
+                            value="mp3",
+                            label="Формат вывода"
+                        )
+                    
+                    convert_btn = gr.Button("Преобразовать!", variant="primary")
+                
+                with gr.Column():
+                    output_files = gr.Files(label="Аудио с преобразованным вокалом")
+
+            # Обработчики
+            refresh_btn.click(
+                fn=lambda: gr.update(choices=get_models_list()),
+                outputs=voicemodel_name
+            )
+            
+            convert_btn.click(
+                fn=convert_voice,
+                inputs=[
+                    voicemodel_name,
+                    pitch_vocal,
+                    method_pitch,
+                    hop_length,
+                    index_rate,
+                    filter_radius,
+                    rms,
+                    protect,
+                    output_format
+                ],
+                outputs=output_files
+            )
+
+        # Вторая вкладка
+        with gr.TabItem("Скачать модель"):
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("## Здесь можно скачать модель по ссылке на архив с нею")
+                    url_input = gr.Textbox(
+                        label="Ссылка на архив с моделью",
+                        placeholder="хаггингфейс.ко/модель.zip"
+                    )
+                    dir_name_input = gr.Textbox(
+                        label="Имя модели",
+                        placeholder="Имя модели"
+                    )
+                    download_btn = gr.Button("Скачать модель", variant="primary")
+                    install_status = gr.Textbox(label="Статус скачивания модели")
+                    
+                    download_btn.click(
+                        fn=download_model,
+                        inputs=[url_input, dir_name_input],
+                        outputs=install_status
+                    )
+                
+                with gr.Column():
+                    gr.Markdown("## Проверить список моделей")
+                    model_list = gr.Textbox(
+                        value="\n".join(get_models_list()),
+                        lines=10,
+                        label="Установленные модели"
+                    )
+                    refresh_models_btn = gr.Button("Обновить")
+                    
+                    refresh_models_btn.click(
+                        fn=lambda: gr.update(value="\n".join(get_models_list())),
+                        outputs=model_list
+                    )
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,156 @@
+aiofiles==23.2.1
+aiohappyeyeballs==2.4.4
+aiohttp==3.11.11
+aiosignal==1.3.2
+annotated-types==0.7.0
+antlr4-python3-runtime==4.8
+anyio==4.8.0
+async-timeout==5.0.1
+attrs==25.1.0
+audioread==3.0.1
+beautifulsoup4==4.12.3
+bitarray==3.0.0
+blinker==1.4
+certifi==2024.12.14
+cffi==1.17.1
+charset-normalizer==3.4.1
+click==8.1.8
+colorama==0.4.6
+contourpy==1.3.1
+cryptography==3.4.8
+cycler==0.12.1
+Cython==3.0.11
+dbus-python==1.2.18
+decorator==5.1.1
+distro==1.7.0
+edge-tts==7.0.0
+einops==0.8.0
+exceptiongroup==1.2.2
+fairseq==0.12.2
+faiss-cpu==1.7.3
+fastapi==0.112.4
+ffmpeg-python==0.2.0
+ffmpy==0.5.0
+filelock==3.17.0
+fonttools==4.55.7
+frozenlist==1.5.0
+fsspec==2024.12.0
+future==1.0.0
+gdown==5.2.0
+gradio==4.43.0
+gradio_client==1.3.0
+h11==0.14.0
+httpcore==1.0.7
+httplib2==0.20.2
+httpx==0.28.1
+huggingface-hub==0.28.0
+hydra-core==1.0.7
+hyper-connections==0.1.9
+idna==3.10
+importlib-metadata==4.6.4
+importlib_resources==6.5.2
+jeepney==0.7.1
+Jinja2==3.1.5
+joblib==1.4.2
+keyring==23.5.0
+kiwisolver==1.4.8
+launchpadlib==1.10.16
+lazr.restfulclient==0.14.4
+lazr.uri==1.0.6
+lazy_loader==0.4
+librosa==0.10.2.post1
+llvmlite==0.43.0
+local-attention==1.11.1
+lxml==5.3.0
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.10.0
+mdurl==0.1.2
+mega.py==1.0.8
+more-itertools==8.10.0
+mpmath==1.3.0
+msgpack==1.1.0
+multidict==6.1.0
+networkx==3.4.2
+numba==0.60.0
+numpy==1.23.5
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
+oauthlib==3.2.0
+omegaconf==2.0.6
+orjson==3.10.15
+packaging==24.2
+pandas==2.2.3
+pathlib==1.0.1
+pedalboard==0.9.16
+pillow==10.4.0
+platformdirs==4.3.6
+pooch==1.8.2
+portalocker==3.1.1
+praat-parselmouth==0.4.5
+propcache==0.2.1
+protobuf==5.29.3
+pycparser==2.22
+pycryptodome==3.21.0
+pydantic==2.10.6
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.19.1
+PyGObject==3.42.1
+PyJWT==2.3.0
+pyparsing==2.4.7
+PySocks==1.7.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2024.2
+pyworld==0.3.4
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.3
+resampy==0.4.3
+rich==13.9.4
+ruff==0.9.3
+sacrebleu==2.5.1
+scikit-learn==1.6.1
+scipy==1.15.1
+SecretStorage==3.3.1
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+soundfile==0.13.1
+soupsieve==2.6
+soxr==0.5.0.post1
+srt==3.5.3
+starlette==0.38.6
+sympy==1.13.1
+tabulate==0.9.0
+tenacity==5.1.5
+tensorboardX==2.6.2.2
+threadpoolctl==3.5.0
+tomlkit==0.12.0
+torch==2.5.1
+torchaudio==2.5.1
+torchcrepe==0.0.23
+tqdm==4.67.1
+triton==3.1.0
+typer==0.15.1
+typing_extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0
+uvicorn==0.34.0
+wadllib==1.3.6
+websockets==12.0
+wget==3.2
+yarl==1.18.3
+zipp==1.0.0