add some drama

utils/color_model.py

@@ -12,6 +12,10 @@ model_path = hf_hub_download(
 model = tf.keras.models.load_model(model_path, compile=False)
 
 
+# -------------------------------------------------
+# Model Inference
+# -------------------------------------------------
+
 def infer_color(vad):
     input_data = np.array([[
         vad["V"],
@@ -22,26 +26,68 @@ def infer_color(vad):
     ]], dtype=np.float32)
 
     output = model.predict(input_data, verbose=0)[0]
-
     r, g, b, e, i = output
 
     return {
         "R": float(r),
         "G": float(g),
         "B": float(b),
-        "E": float(e),
-        "I": float(i)
+        "E": float(e),  # Energy
+        "I": float(i)   # Intensity
     }
 
-def scale_rgb(rgb):
+
+# -------------------------------------------------
+# Cinematic Drama Rendering
+# -------------------------------------------------
+
+def apply_cinematic_blend(model_output, drama):
+    """
+    drama:
+        0.0  → near white
+        1.0  → cinematic target
+        1.5  → expressive peak
+    """
+
+    color = np.array([
+        model_output["R"],
+        model_output["G"],
+        model_output["B"]
+    ])
+
+    energy = model_output["E"]
+    intensity = model_output["I"]
+
+    white = np.array([1.0, 1.0, 1.0])
+
+    # Nonlinear cinematic curve
+    curve = drama ** 2.2
+
+    # Blend strength influenced by model intensity
+    strength = np.clip(curve * intensity, 0.0, 1.0)
+
+    # Blend toward white baseline
+    blended = white * (1 - strength) + color * strength
+
+    # Subtle energy-based brightness shaping
+    brightness_boost = 0.9 + (0.25 * energy)
+    blended = blended * brightness_boost
+
+    blended = np.clip(blended, 0.0, 1.0)
+
     return {
-        "R": int(max(0, min(255, rgb["R"] * 255))),
-        "G": int(max(0, min(255, rgb["G"] * 255))),
-        "B": int(max(0, min(255, rgb["B"] * 255))),
-        "E": rgb["E"],
-        "I": rgb["I"]
+        "R": int(blended[0] * 255),
+        "G": int(blended[1] * 255),
+        "B": int(blended[2] * 255),
+        "E": energy,
+        "I": intensity
     }
 
+
+# -------------------------------------------------
+# Render HTML Block
+# -------------------------------------------------
+
 def render_color(rgb):
     return f"""
     <div style="
@@ -50,7 +96,7 @@ def render_color(rgb):
         border-radius:18px;
         background: rgb({rgb['R']},{rgb['G']},{rgb['B']});
         box-shadow: 0px 6px 32px rgba(0,0,0,0.25);
-        transition: all 0.3s ease-in-out;
+        transition: all 0.35s cubic-bezier(.4,0,.2,1);
     "></div>
     """