Update app.py

app.py

@@ -1,22 +1,69 @@
 # MARK: app.py
 
-import os
-import sys
-import json
-import base64
-import random
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import HTMLResponse, FileResponse, StreamingResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from fastapi.middleware.cors import CORSMiddleware
 from pathlib import Path
-from io import BytesIO
+from dominant_color import get_luminance, complementary, get_dominant_color, _is_grayscale, _rgb_to_hsv, hsv_to_rgb, rgb_to_hsv
+import subprocess
+import urllib.parse
+import os
+import json
 from mutagen import File
+from io import BytesIO
+import base64
 from PIL import Image
-from colorsys import rgb_to_hsv, hsv_to_rgb
-
-
+from colorsys import rgb_to_hls, hls_to_rgb
+import logging
+import wave
+import struct
+import math
+import random
+import shutil
+
+# command = "cls"
+# if os.name in ("nt", "dos"):  # If Machine is running on Win OS, use cls
+#     subprocess.call(command, shell=True)
+# else:  # If Machine is running on OSX/Linux
+#     subprocess.call(["printf", "\033c"])
+
+
+no_style = "\033[0m"
+bold = "\033[91m"
+black = "\033[30m"
+grey = "\033[90m"
+white = "\033[97m"
+white_bg = "\033[107m"
+blue = "\033[94m"
+blue_bg = "\033[104m"
+green = "\033[92m"
+green_bg = "\033[102m"
+yellow = "\033[93m"
+red = "\033[31m"
+red_light = "\033[91m"
+red_light_bg = "\033[101m"
+
+global logger
+logger = logging.getLogger()
+handler = logging.StreamHandler()
+logging.addLevelName(logging.DEBUG, grey + "DEBUG:   " + no_style)
+logging.addLevelName(logging.INFO, blue + "ℹ️")
+logging.addLevelName(logging.WARNING, no_style + yellow + "WARNING:")
+logging.addLevelName(logging.ERROR, no_style + red_light_bg + black + "ERROR:   " + no_style)
+logging.addLevelName(logging.CRITICAL, f"{no_style + red}CRITICAL:{no_style}")
+handler.setLevel(logging.INFO)  # DEBUG INFO WARNING ERROR CRITICAL
+# formatter = logging.Formatter(fmt=" %(levelname)-8s %(message)s", style="%")
+# formatter = logging.Formatter(fmt=" %(levelname)-8s %(message)s")
+# # formatter = AnsiColorFormatter("{levelname:<8s} {message}", style="{")
+formatter = logging.Formatter("{levelname:s} {message}", style="{")
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+logger.setLevel(logging.INFO)  # DEBUG INFO WARNING ERROR CRITICAL
+
+
+# MARK: FastAPI
 app = FastAPI()
 
 app.add_middleware(
@@ -30,8 +77,7 @@ app.add_middleware(
 SERVER_HOST = "0.0.0.0"
 SERVER_PORT = 7860
 BASE_URL = "https://sebastiankay-my-ai-songs.hf.space/"
-GET_AUDIO_FILE_URL = BASE_URL + "getfile/"
-GET_COVERART_URL = BASE_URL + "static/coverarts/"
+GET_FILE_URL = BASE_URL + "getfile/"
 
 
 # Serve static files from the "static" directory
@@ -43,67 +89,20 @@ templates = Jinja2Templates(directory="templates")
 # Configure the base directory for your music files
 app.mount("/files/music", StaticFiles(directory="files/music"), name="music")
 MUSIC_DIR = Path("files/music")
+MUSIC_DIR_2 = Path("files/music_2")
 
 
-# MARK: LUMINANCE
-def get_luminance(r, g, b):
-    """returns the luminance of a color"""
-    return 0.2126 * r + 0.7152 * g + 0.0722 * b
-
-
-# MARK: COMPLEMENTARY COLOR
-def complementary(r, g, b):
-    """returns RGB components of complementary color"""
-    hsv = rgb_to_hsv(r, g, b)
-    rgb_color = hsv_to_rgb((hsv[0] + 0.5) % 1, hsv[1], hsv[2])
-    return [round(rgb_color[0]), round(rgb_color[1]), round(rgb_color[2])]
-
+def create_peaks_json(file):
+    input_file = file
+    output_file = f"{file}.json"
+    command = f'audiowaveform -i "{input_file}" -o "{output_file}" -b 8'
+    # print("Executing command:", command)
+    subprocess.call(command, shell=True)
 
-# MARK: DOMINANT COLOR
-def get_dominant_color_2(image_path):
-    img = Image.open(image_path)
-    img = img.convert("RGB")
-    img = img.resize((100, 100), resample=0)
-    pixels = list(img.getdata())
+    with open(output_file, "r") as f:
+        peaks = json.load(f)
 
-    # Erzeuge eine Liste mit den Häufigkeiten der Farben
-    colors = []
-    for pixel in pixels:
-        r, g, b = pixel
-        h, s, v = rgb_to_hsv(r / 255, g / 255, b / 255)
-        if v > 0.5:  # Filteriere hellere Farben aus
-            continue
-        if v > 0.99:  # Filteriere Weiß aus
-            continue
-        colors.append((h, s, v))
-
-    # Ermittle die dominante Farbe
-    dominant_color = max(colors, key=lambda x: x[2])
-    dominant_color_rgb = hsv_to_rgb(dominant_color[0], dominant_color[1], dominant_color[2])
-    dominant_color_rgb = [int(c * 255) for c in dominant_color_rgb]
-    # print(dominant_color_rgb)
-
-    return dominant_color_rgb
-
-
-def get_dominant_color(image, palette_size=16):
-
-    pil_img = Image.open(image)
-
-    # Resize image to speed up processing
-    img = pil_img.copy()
-    img.thumbnail((100, 100))
-
-    # Reduce colors (uses k-means internally)
-    paletted = img.convert("P", palette=Image.ADAPTIVE, colors=palette_size)
-
-    # Find the color that occurs most often
-    palette = paletted.getpalette()
-    color_counts = sorted(paletted.getcolors(), reverse=True)
-    palette_index = color_counts[0][1]
-    dominant_color = palette[palette_index * 3 : palette_index * 3 + 3]
-
-    return dominant_color
+    return peaks["data"]
 
 
 # Function to get the list of music files with metadata
@@ -116,10 +115,17 @@ def get_music_files_with_metadata():
     music_files = []
     for file in MUSIC_DIR.iterdir():
         if file.is_file():
-            # Datei überspringen, wenn "_HIDETHIS" im Namen vorkommt
-            if "_HIDETHIS" in file.name:
+
+            if "_HIDETHIS" in file.name or file.name.endswith(".json") or file.name.endswith(".png"):
+                print(f"{yellow}Ignoring File: {no_style} {file.name}")
                 continue
 
+            filename_without_mp3 = file.name.rsplit(".", 1)[0]
+            filename_coverart = filename_without_mp3 + ".coverart.png"
+            filename_coverart_encoded = urllib.parse.quote(filename_coverart)
+            filename_peaks = filename_without_mp3 + ".peaks.json"
+            # peaks = create_peaks_json(file)
+
             try:  # Wrap metadata extraction in try-except
                 audio = File(file)
                 if audio is None:  # Handle cases where mutagen can't read the file
@@ -129,41 +135,38 @@ def get_music_files_with_metadata():
                 track_number_str = audio.get("TRCK", ["0"])[0].split("/")[0]
                 track_number = int(track_number_str) if track_number_str.isdigit() else 0
 
-                # print(audio.getall("APIC", ["Unknown"])[0].data)
-
-                cover_art_filename = file.name + ".png" if Path("static/coverarts/" + file.name + ".png").is_file() else f"no_coverart_{random.randint(1, 4)}.png"
-                print("cover_art_path:", GET_COVERART_URL + cover_art_filename)
-
                 metadata = {
                     "encoded_title": base64.b64encode(file.name.encode("utf-8")).decode("utf-8"),
-                    "filename": file.name,
                     "title": str(audio.get("TIT2", ["Unknown"])[0]),
                     "artist": str(audio.get("TPE1", ["Unknown"])[0]),
                     "genre": str(audio.get("TCON", ["Unknown"])[0]),
                     "duration": int(audio.info.length) if hasattr(audio, "info") and hasattr(audio.info, "length") else 0,
                     "track_number": track_number,
-                    "audiofile": GET_AUDIO_FILE_URL + file.name,
-                    "cover_art": file.name + ".png",
-                    "coverart": GET_COVERART_URL + cover_art_filename,
-                    "dominant_color": "128, 128, 128",
-                    "complementary_color": "128, 128, 128",
+                    "audiofile": GET_FILE_URL + file.name,
+                    "coverart": GET_FILE_URL + filename_coverart_encoded,
+                    "dominant_color_1": "128, 128, 128",
+                    "complementary_color_1": "128, 128, 128",
+                    "dominant_color_2": "128, 128, 128",
+                    "complementary_color_2": "128, 128, 128",
+                    # "peaks": peaks,
                 }
 
                 music_files.append(metadata)
 
                 # --- Cover Art Processing ---
-                cover_art_path = Path("static/coverarts") / cover_art_filename
-                
+                cover_art_path = MUSIC_DIR / filename_coverart
                 if cover_art_path.is_file():
                     try:
-                        dominant_color = get_dominant_color(cover_art_path)
-                        complementary_color = complementary(dominant_color[0], dominant_color[1], dominant_color[2])
+                        colors = get_dominant_color(cover_art_path, min_brightness=0.4, max_brightness=0.8, n_clusters=4)
+
+                        metadata["dominant_color_1"] = ", ".join(map(str, colors[0]))
+                        metadata["complementary_color_1"] = ", ".join(map(str, colors[1]))
 
-                        metadata["dominant_color"] = ", ".join(map(str, dominant_color))
-                        metadata["complementary_color"] = ", ".join(map(str, complementary_color))
+                        metadata["dominant_color_2"] = ", ".join(map(str, colors[2]))
+                        metadata["complementary_color_2"] = ", ".join(map(str, colors[3]))
 
                     except Exception as e:
-                        print(f"Error processing cover art {metadata['cover_art']}: {e}")
+                        print(f"{red_light}Error{no_style} processing cover art {metadata['coverart']}: {e}")
                 else:
                     print(f"Cover art file not found: {cover_art_path}")
                 # --- End Cover Art Processing ---
@@ -215,6 +218,12 @@ async def get_file(filename: str):
         media_type = "audio/ogg"
     elif filename.lower().endswith(".flac"):
         media_type = "audio/flac"
+    elif filename.lower().endswith(".png"):
+        media_type = "image/png"
+    elif filename.lower().endswith(".jpg") or filename.lower().endswith(".jpeg"):
+        media_type = "image/jpeg"
+    elif filename.lower().endswith(".gif"):
+        media_type = "image/gif"
 
     # def iterfile():  # (1)
     #     with open(file_path, mode="rb") as file_like:  # (2)
@@ -240,6 +249,12 @@ if __name__ == "__main__":
 
     # print(f"\n\nStarting server on: \nhttp://{SERVER_HOST}:{SERVER_PORT}")
     print()
-    print(f"API DOCS: {BASE_URL}docs\n----------------------------------\n----------------------------------")
+    logger.info("=============================================================")
+    logger.info(" ")
+    logger.info(f"API DOCS: {BASE_URL}docs")
+    logger.info(f"ENDPOINT: {BASE_URL}api/music")
+    logger.info("")
+    logger.info("=============================================================")
+    print()
     print()
     uvicorn.run(app, host=SERVER_HOST, port=SERVER_PORT)