Update app.py

app.py

@@ -2,44 +2,51 @@ import gradio as gr
 import librosa
 import numpy as np
 from pydub import AudioSegment
-import os
 
-def ghost_ultimate_v7(tu_track, track_ref, intensidad, duda):
+def ghost_fix(tu_track, track_ref, intensidad):
     if tu_track is None:
-        return None, "Sube tu audio.", "Esperando...", "https://images.unsplash.com/photo-1614613535308-eb5fbd3d2c17?q=80&w=500", "Sube audio."
-
+        return None, "Sube un audio.", "...", "https://images.unsplash.com/photo-1614613535308-eb5fbd3d2c17?q=80&w=500", "..."
     try:
-        # 1. ANALISIS
-        y, sr = librosa.load(tu_track, duration=30)
+        # 1. ANALISIS RAPIDO
+        y, sr = librosa.load(tu_track, duration=20)
         chroma = librosa.feature.chroma_cqt(y=y, sr=sr)
-        key_idx = np.argmax(np.mean(chroma, axis=1))
         notas = ["C", "C#", "D", "D#", "E", "F", "F#", "G", "G#", "A", "A#", "B"]
-        key_det = notas[key_idx]
-        
+        key_det = notas[np.argmax(np.mean(chroma, axis=1))]
         rms_tu = np.mean(librosa.feature.rms(y=y))
-        estilo = "Dark Tech" if rms_tu > 0.12 else "Liquid"
-
+        
         # 2. COMPARADOR
-        comp_txt = "No hay referencia."
+        info = "TONALIDAD: " + key_det + "\n"
         if track_ref is not None:
-            y_ref, _ = librosa.load(track_ref, duration=30)
-            rms_ref = np.mean(librosa.feature.rms(y=y_ref))
-            diff = 20 * np.log10(rms_ref / rms_tu)
-            comp_txt = "Diferencia de volumen: " + str(round(diff, 1)) + " dB"
+            y_r, _ = librosa.load(track_ref, duration=20)
+            diff = 20 * np.log10(np.mean(librosa.feature.rms(y=y_r)) / rms_tu)
+            info += "DIF. VOLUMEN: " + str(round(diff, 1)) + " dB\n"
+        
+        # 3. MASTER
+        audio = AudioSegment.from_file(tu_track)
+        audio = audio.apply_gain((intensidad / 10) - 5)
+        audio.export("final.wav", format="wav")
+        
+        prompt = "Cyberpunk Art, Neon, DnB style, Key: " + key_det
+        return "final.wav", info + "\nCONSEJO: Ajusta tu Kick a " + key_det, key_det, "https://images.unsplash.com/photo-1550684848-fac1c5b4e853?q=80&w=500", prompt
+    except Exception as e:
+        return None, "Error: " + str(e), "!", None, "..."
 
-        # 3. MASTERING
-        sound = AudioSegment.from_file(tu_track)
-        gain = (intensidad / 10) - 5
-        mastered = sound.apply_gain(gain)
-        out = "master_v7.wav"
-        mastered.export(out, format="wav")
+with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="orange")) as demo:
+    gr.Markdown("# 💀 DNB GHOST v7.2")
+    with gr.Row():
+        with gr.Column():
+            in_t = gr.Audio(label="TU TRACK", type="filepath")
+            in_r = gr.Audio(label="REFERENCIA", type="filepath")
+            in_s = gr.Slider(0, 100, label="MASTER", value=80)
+            btn = gr.Button("🚀 PROCESAR", variant="primary")
+        with gr.Column():
+            out_a = gr.Audio(label="MASTER")
+            out_i = gr.Textbox(label="INFORME")
+            out_k = gr.Label(label="KEY")
+        with gr.Column():
+            out_m = gr.Image(label="VIBE")
+            out_p = gr.Textbox(label="PROMPT ARTE")
 
-        # 4. INFORME (Usando suma de textos para evitar errores de llaves {})
-        info = "TONALIDAD: " + key_det + "\n"
-        info += "ESTILO: " + estilo + "\n\n"
-        info += "COMPARATIVA: " + comp_txt + "\n\n"
-        info += "CONSEJO: En " + key_det + " el sub debe ser afinado."
-        
-        prompt = "Cyberpunk DnB Art, Neon, " + estilo + " vibes, " + key_det + " mood."
+    btn.click(ghost_fix, [in_t, in_r, in_s], [out_a, out_i, out_k, out_m, out_p])
 
-        return out, info, key_det + " | " + estilo
+demo.launch()