Synced repo using 'sync_with_huggingface' Github Action

.dockerignore

@@ -0,0 +1,9 @@
+html/*
+.DS_Store
+
+.venv
+*.swp
+.mypy_cache
+.pytest_cache
+.ipynb_checkpoints
+__pycache__

.pre-commit-config.yaml

@@ -0,0 +1,13 @@
+repos:
+  - repo: https://github.com/psf/black
+    rev: 22.12.0
+    hooks:
+      - id: black
+        types: [python]
+        language_version: python3.10.11
+  - repo: https://github.com/PyCQA/isort
+    rev: 5.12.0
+    hooks:
+      - id: isort
+        types: [python]
+        language_version: python3.10.11

Dockerfile

@@ -0,0 +1,66 @@
+FROM ubuntu:22.04 AS base
+
+ARG PYTHON_VERSION=3.10
+
+ENV DEBIAN_FRONTEND=noninteractive
+ENV WORKDIR /app/
+
+WORKDIR /opt
+
+# install dev tools
+RUN apt-get update && apt-get install -y \
+  vim neovim nano \
+  git git-lfs \
+  zip unzip \
+  curl wget make build-essential xz-utils file tree \
+  sudo \
+  dnsutils \
+  tzdata language-pack-ja \
+  && apt-get clean \
+  && rm -rf /var/lib/apt/lists/*
+
+# for Japanese settings
+# ENV TZ Asia/Tokyo
+# ENV LANG ja_JP.utf8
+
+# for US settings
+ENV LANG en_US.UTF-8
+ENV LANGUAGE en_US
+
+# install Python
+RUN apt-get update && apt-get -yV upgrade && DEBIAN_FRONTEND=noninteractive apt-get -yV install \
+  build-essential libssl-dev libffi-dev \
+  python${PYTHON_VERSION} python${PYTHON_VERSION}-distutils python${PYTHON_VERSION}-dev \
+  && ln -s /usr/bin/python${PYTHON_VERSION} /usr/local/bin/python3 \
+  && ln -s /usr/bin/python${PYTHON_VERSION} /usr/local/bin/python \
+  && apt-get clean \
+  && rm -rf /var/lib/apt/lists/*
+
+## install pip
+RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
+  && python3 get-pip.py \
+  && pip3 --no-cache-dir install --upgrade pip
+
+## install Poetry
+RUN curl -sSL https://install.python-poetry.org | python3 -
+ENV PATH $PATH:/root/.local/bin
+RUN poetry config virtualenvs.create true \
+  && poetry config virtualenvs.in-project false
+
+WORKDIR ${WORKDIR}
+
+# install python packages
+COPY poetry.lock pyproject.toml ./
+COPY src ./src
+RUN poetry install --no-dev
+
+FROM base AS dev
+WORKDIR ${WORKDIR}
+
+# install python packages
+COPY poetry.lock pyproject.toml ./
+COPY src ./src
+RUN poetry install
+
+# Hugging Face Hub Settings
+CMD ["poetry", "run", "streamlit", "run", "src/app.py", "--server.port", "7860"]

compose.yaml

@@ -0,0 +1,29 @@
+version: "3.2"
+services:
+  binarual_beats_generator:
+    tty: true
+    stdin_open: true
+    user: root
+    working_dir: /app
+    build:
+      context: .
+      dockerfile: docker/cpu/Dockerfile
+      target: dev
+      # secrets:
+      #   - github_token
+      args:
+        progress: plain
+    volumes:
+      - type: bind
+        source: ./
+        target: /app
+    ports:
+      - "8501:8501"
+    command:
+      poetry run streamlit run src/app.py
+    environment:
+        PYTHONPATH: "/app/src"
+        PYTHONUNBUFFERED: 1
+# secrets:
+#   github_token:
+#     file: ${HOME}/.git-credentials

docker/cpu/Dockerfile

@@ -0,0 +1,63 @@
+FROM ubuntu:22.04 AS base
+
+ARG PYTHON_VERSION=3.10
+
+ENV DEBIAN_FRONTEND=noninteractive
+ENV WORKDIR /app/
+
+WORKDIR /opt
+
+# install dev tools
+RUN apt-get update && apt-get install -y \
+  vim neovim nano \
+  git git-lfs \
+  zip unzip \
+  curl wget make build-essential xz-utils file tree \
+  sudo \
+  dnsutils \
+  tzdata language-pack-ja \
+  && apt-get clean \
+  && rm -rf /var/lib/apt/lists/*
+
+# for Japanese settings
+# ENV TZ Asia/Tokyo
+# ENV LANG ja_JP.utf8
+
+# for US settings
+ENV LANG en_US.UTF-8
+ENV LANGUAGE en_US
+
+# install Python
+RUN apt-get update && apt-get -yV upgrade && DEBIAN_FRONTEND=noninteractive apt-get -yV install \
+  build-essential libssl-dev libffi-dev \
+  python${PYTHON_VERSION} python${PYTHON_VERSION}-distutils python${PYTHON_VERSION}-dev \
+  && ln -s /usr/bin/python${PYTHON_VERSION} /usr/local/bin/python3 \
+  && ln -s /usr/bin/python${PYTHON_VERSION} /usr/local/bin/python \
+  && apt-get clean \
+  && rm -rf /var/lib/apt/lists/*
+
+## install pip
+RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
+  && python3 get-pip.py \
+  && pip3 --no-cache-dir install --upgrade pip
+
+## install Poetry
+RUN curl -sSL https://install.python-poetry.org | python3 -
+ENV PATH $PATH:/root/.local/bin
+RUN poetry config virtualenvs.create true \
+  && poetry config virtualenvs.in-project false
+
+WORKDIR ${WORKDIR}
+
+# install python packages
+COPY poetry.lock pyproject.toml ./
+COPY src ./src
+RUN poetry install --no-dev
+
+FROM base AS dev
+WORKDIR ${WORKDIR}
+
+# install python packages
+COPY poetry.lock pyproject.toml ./
+COPY src ./src
+RUN poetry install

notebooks/helloworld.ipynb

@@ -0,0 +1,51 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "18edcb70-64a5-4d17-94c0-a86ecc435be4",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "hello world\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(\"hello world\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b90e3e1f-5aa1-43a0-b9e2-1d1bcaf9ff94",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.5"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

pyproject.toml

@@ -0,0 +1,57 @@
+[tool.poetry]
+name = "brainwave"
+version = "0.1.0"
+description = ""
+authors = ["iamtatsuki05 <tatsukio0522@gmail.com>"]
+packages = [
+    { include = "brainwave", from = "src/" },
+]
+
+[tool.poetry.dependencies]
+python = "^3.10"
+python-dotenv = "^1.0.0"
+setuptools = "^69.0.3"
+fire = "^0.5.0"
+pydantic = "^2.5.3"
+beautifulsoup4 = "^4.12.2"
+selenium = "^4.16.0"
+fastapi = "^0.108.0"
+uvicorn = "^0.25.0"
+matplotlib = "^3.5.1"
+pandas = "^1.4.2"
+seaborn = "^0.11.2"
+japanize-matplotlib = "^1.1.3"
+numpy = "^1.22.3"
+jupyterlab = "^3.3.4"
+tqdm = "^4.64.0"
+scikit-learn = "^1.1.1"
+openpyxl = "^3.1.2"
+streamlit = "^1.34.0"
+
+[tool.poetry.group.dev.dependencies]
+pytest = "^7.0.0"
+ipykernel = ">=6.13.0"
+autopep8 = ">=1.6.0"
+autoflake = ">=1.4"
+flake8 = ">=4.0.1"
+flake8-isort = ">=4.1.1"
+flake8-quotes = ">=3.3.1"
+flake8-print = ">=4.0.0"
+isort = ">=5.10.1"
+black = ">=22.10.0"
+mypy = ">=0.971"
+tox = ">=3.25.1"
+pre-commit = ">=3.3.3"
+nbstripout = "0.6.1"
+
+[tool.isort]
+line_length = 88
+multi_line_output = 3
+include_trailing_comma = true
+
+[tool.black]
+skip-string-normalization = true
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"

scripts/main.py

@@ -0,0 +1,2 @@
+if __name__ == '__main__':
+    print('Hello, World!')

src/app.py

@@ -0,0 +1,103 @@
+import base64
+from io import BytesIO
+
+import scipy.io.wavfile as wav
+import streamlit as st
+
+from brainwave.config.app import (
+    BASE_FREQUENCY_DEFAULT,
+    BASE_FREQUENCY_RANGE_MAX,
+    BASE_FREQUENCY_RANGE_MIN,
+    BASE_FREQUENCY_STEP,
+    BEAT_FREQUENCY_RANGE_MAX,
+    BEAT_FREQUENCY_RANGE_MIN,
+    BRAIN_WAVES,
+    DURATION_DEFAULT,
+    DURATION_RANGE_MAX,
+    DURATION_RANGE_MIN,
+)
+from brainwave.utils.binaural_beats import generate_binaural_beats
+
+
+def app():
+    st.title('Binaural Beats Generator')
+
+    st.subheader('Brain Wave Selection')
+    selected_wave: str = st.selectbox(
+        'Select a type of brain wave:', list(BRAIN_WAVES.keys())
+    )
+    default_beat: int = BRAIN_WAVES[selected_wave]
+
+    st.subheader('Base Frequency (Hz)')
+    base_frequency: int = st.slider(
+        'Base Frequency (Hz):',
+        min_value=BASE_FREQUENCY_RANGE_MIN,
+        max_value=BASE_FREQUENCY_RANGE_MAX,
+        value=BASE_FREQUENCY_DEFAULT,
+        step=BASE_FREQUENCY_STEP,
+    )
+    base_frequency_input: int = st.number_input(
+        '...or enter the base frequency using the keyboard:',
+        min_value=BASE_FREQUENCY_RANGE_MIN,
+        max_value=BASE_FREQUENCY_RANGE_MAX,
+        value=base_frequency,
+        step=BASE_FREQUENCY_STEP,
+    )
+
+    st.subheader('Beat Frequency (Hz)')
+    beat_frequency: int = st.slider(
+        'Select the beat frequency using the slider:',
+        min_value=BEAT_FREQUENCY_RANGE_MIN,
+        max_value=BEAT_FREQUENCY_RANGE_MAX,
+        value=default_beat,
+    )
+    beat_frequency_input: int = st.number_input(
+        '...or enter the beat frequency using the keyboard:',
+        min_value=BEAT_FREQUENCY_RANGE_MIN,
+        max_value=BEAT_FREQUENCY_RANGE_MAX,
+        value=beat_frequency,
+    )
+
+    st.subheader('Duration (seconds)')
+    duration: int = st.slider(
+        'Duration (seconds):',
+        min_value=DURATION_RANGE_MIN,
+        max_value=DURATION_RANGE_MAX,
+        value=DURATION_DEFAULT,
+    )
+    duration_input: int = st.number_input(
+        '...or enter the duration using the keyboard:',
+        min_value=DURATION_RANGE_MIN,
+        max_value=DURATION_RANGE_MAX,
+        value=duration,
+    )
+
+    base_frequency = base_frequency_input
+    beat_frequency = beat_frequency_input
+    duration = duration_input
+
+    if st.button('Generate Sound'):
+        audio_data, sample_rate = generate_binaural_beats(
+            base_frequency, beat_frequency, duration
+        )
+        wav_file: BytesIO = BytesIO()
+        wav.write(wav_file, sample_rate, audio_data)
+        wav_file.seek(0)
+
+        b64: str = base64.b64encode(wav_file.read()).decode()
+        audio_html: str = (
+            f'<audio controls loop><source src="data:audio/wav;base64,{b64}" '
+            f'type="audio/wav"></audio>'
+        )
+        st.markdown(audio_html, unsafe_allow_html=True)
+
+        href: str = (
+            f'<a href="data:file/wav;base64,{b64}" download="'
+            f'binaural_beats_{selected_wave}.wav">Download '
+            f'{selected_wave} Binaural Beat</a>'
+        )
+        st.markdown(href, unsafe_allow_html=True)
+
+
+if __name__ == '__main__':
+    app()

src/brainwave/config/app.py

@@ -0,0 +1,21 @@
+from typing import Final
+
+BRAIN_WAVES: Final[dict[str, int]] = {
+    'Alpha Waves': 10,  # Typically 8-13 Hz
+    'Beta Waves': 20,  # Typically 14-30 Hz
+    'Theta Waves': 6,  # Typically 4-7 Hz
+    'Delta Waves': 2,  # Typically 0.5-3 Hz
+    'Gamma Waves': 40,  # Typically 30-100 Hz
+}
+
+BASE_FREQUENCY_RANGE_MIN: Final[int] = 1
+BASE_FREQUENCY_RANGE_MAX: Final[int] = 500
+BASE_FREQUENCY_DEFAULT: Final[int] = 200
+BASE_FREQUENCY_STEP: Final[int] = 10
+
+BEAT_FREQUENCY_RANGE_MIN: Final[int] = 1
+BEAT_FREQUENCY_RANGE_MAX: Final[int] = 100
+
+DURATION_RANGE_MIN: Final[int] = 1
+DURATION_RANGE_MAX: Final[int] = 300
+DURATION_DEFAULT: Final[int] = 30

src/brainwave/env.py

@@ -0,0 +1,3 @@
+from pathlib import Path
+
+PACKAGE_DIR = Path(__file__).parents[2]

src/brainwave/main.py

@@ -0,0 +1,6 @@
+def main():
+    print('Hello, World!')
+
+
+if __name__ == '__main__':
+    main()

src/brainwave/utils/binaural_beats.py

@@ -0,0 +1,20 @@
+from typing import Optional
+
+import numpy as np
+
+
+def generate_binaural_beats(
+    base_freq: int,
+    beat_freq: int,
+    duration: int,
+    fs: Optional[int] = 44100,
+) -> tuple[np.ndarray, int]:
+    n_samples: int = int(fs * duration)
+    t: np.ndarray = np.arange(n_samples) / fs
+    left_ear: np.ndarray = np.sin(2 * np.pi * base_freq * t)
+    right_ear: np.ndarray = np.sin(2 * np.pi * (base_freq + beat_freq) * t)
+    stereo_sound: np.ndarray = np.vstack((left_ear, right_ear)).T
+    stereo_normalized: np.ndarray = np.int16(
+        (stereo_sound / np.max(np.abs(stereo_sound))) * 32767
+    )
+    return stereo_normalized, fs