Update app.py

app.py

@@ -1,316 +1,221 @@
-import gradio as gr
-import librosa
-import numpy as np
-import matplotlib.pyplot as plt
-import io
-import reportlab
-from reportlab.pdfgen import canvas
-from reportlab.lib.pagesizes import letter
-from reportlab.lib.units import inch
-from reportlab.lib import colors
+# ----------------------------------------
+# PDF GENERATION – LUXE EDITION
+# ----------------------------------------
+
 from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
+from reportlab.lib.pagesizes import letter
+from reportlab.lib import colors
+from reportlab.pdfgen import canvas
 from reportlab.lib.utils import ImageReader
-import qrcode
-import tempfile
-import datetime
-import os
-
-# ===============================
-#  REGISTER PLAYFAIR DISPLAY BOLD
-# ===============================
-pdfmetrics.registerFont(TTFont("PlayfairBold", "PlayfairDisplay-Bold.ttf"))
-
-# ============================================
-#  LOAD LOGO FROM ROOT (logo.png)
-# ============================================
-LOGO_PATH = "logo.png"
-
-# ================================
-#  COLOR PALETTE (LUXE MODE)
-# ================================
-AUBERGINE = colors.HexColor("#3B0B3F")
-ACCENT = colors.HexColor("#8B5CF6")
-GREEN = colors.HexColor("#1FA749")
-YELLOW = colors.HexColor("#F6C434")
-RED = colors.HexColor("#D83131")
-
-# ================================
-#  UTILITY FUNCTIONS
-# ================================
-
-def detect_bpm(y, sr):
-    try:
-        tempo, _ = librosa.beat.beat_track(y=y, sr=sr)
-        return round(float(tempo))
-    except:
-        return None
 
-def detect_key(y, sr):
-    try:
-        chroma = librosa.feature.chroma_cqt(y=y, sr=sr)
-        chroma_mean = np.mean(chroma, axis=1)
-        key_index = np.argmax(chroma_mean)
-        KEYS = ["C","C#","D","Eb","E","F","F#","G","Ab","A","Bb","B"]
-        return KEYS[key_index]
-    except:
-        return None
+# Register Playfair Display Bold (must match file name EXACTLY)
+pdfmetrics.registerFont(TTFont("PlayfairDisplay-Bold", "PlayfairDisplay-Bold.ttf"))
+pdfmetrics.registerFont(TTFont("Helvetica", "Helvetica"))
+
+ACCENT = colors.HexColor("#8b5cf6")     # r8b5cf6
+PRIMARY = colors.HexColor("#3b0c3f")    # eggplant/aubergine
+LIGHT_GRAY = colors.HexColor("#e6e6e6")
+BLACK = colors.black
+WHITE = colors.white
+
+ENGINE_URL = "https://www.tourdefierce.vip/audio-clapback-engine-ai-detector-autotune-detector"
+
+
+def make_pdf(
+    logo_path,
+    ai_score,
+    human_score,
+    atune,
+    shade,
+    style,
+    tone,
+    transcript,
+    clip_title
+):
+    buffer = io.BytesIO()
+    c = canvas.Canvas(buffer, pagesize=letter)
+
+    W, H = letter
+
+    # -------------------------------
+    # HEADER AREA
+    # -------------------------------
+    c.setFillColor(WHITE)
+    c.rect(0, 0, W, H, fill=1)
+
+    # Logo
+    c.drawImage("logo.png", 40, H - 120, width=90, height=90, preserveAspectRatio=True)
+
+    # Brand name
+    c.setFillColor(PRIMARY)
+    c.setFont("PlayfairDisplay-Bold", 32)
+    c.drawString(150, H - 60, "Tour de Fierce")
+
+    # Subtitle
+    c.setFillColor(ACCENT)
+    c.setFont("Helvetica", 14)
+    c.drawString(150, H - 82, "Audio Clapback Report™")
 
-def detect_voice_range(pitches):
-    valid = pitches[pitches > 0]
-    if len(valid) == 0:
-        return "Unknown range"
+    # Timestamp
+    c.setFillColor(BLACK)
+    c.setFont("Helvetica", 9)
+    c.drawString(150, H - 98, f"Generated: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M')}")
 
-    median_pitch = np.median(valid)
-    if median_pitch < 150: return "Bass / Baritone Piece"
-    if median_pitch < 250: return "Tenor / Alto Piece"
-    if median_pitch < 350: return "Mezzo / Soprano Piece"
-    return "High Soprano / Extreme Range Piece"
+    # Clip Title
+    c.setFont("Helvetica", 11)
+    c.drawString(40, H - 145, f"Clip Analyzed: {clip_title}")
 
-def detect_midi_like(y):
+    # QR Code (top right)
     try:
-        onset_env = librosa.onset.onset_strength(y=y)
-        diffs = np.diff(onset_env)
-        jitter = np.std(diffs)
-        return jitter < 0.01
-    except:
-        return False
-
-def shade_score(ai_like):
-    # direct mapping
-    return round(float(ai_like) * 1.0, 1)
-
-def metric_color(value, metric_type):
-    """Returns Green/Yellow/Red depending on thresholds."""
-    if metric_type == "ai":
-        if value <= 25: return GREEN
-        if value <= 75: return YELLOW
-        return RED
-    if metric_type == "human":
-        if value >= 75: return GREEN
-        if value >= 25: return YELLOW
-        return RED
-    if metric_type == "auto":
-        # always orange block
-        return colors.HexColor("#F59E0B")
-    return AUBERGINE
-
-# ======================================
-#  PDF GENERATOR (LUXE VERSION)
-# ======================================
-def generate_pdf(filename, ai_like, human_like, auto_index, shade, style, tone,
-                 key, bpm, voice_range, is_midi, transcript):
-
-    out = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
-    c = canvas.Canvas(out.name, pagesize=letter)
-    width, height = letter
-
-    # ======================================
-    # HEADER
-    # ======================================
-    c.setFont("PlayfairBold", 24)
-    c.setFillColor(AUBERGINE)
-    c.drawString(50, height - 70, "Tour de Fierce™")
-
-    c.setFont("PlayfairBold", 16)
-    c.drawString(50, height - 95, "Audio Clapback Report™")
-
-    # timestamp
-    c.setFont("Helvetica", 10)
-    c.setFillColor(colors.black)
-    c.drawString(50, height - 112, f"Generated: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M')}")
-
-    # filename
-    c.setFont("Helvetica-Bold", 11)
-    c.drawString(50, height - 130, f"Report for: {filename}")
-
-    # LOGO (top left)
-    try:
-        logo = ImageReader(LOGO_PATH)
-        c.drawImage(logo, width - 150, height - 120, width=100, preserveAspectRatio=True, mask='auto')
+        import qrcode
+        qr = qrcode.make(ENGINE_URL)
+        qr_buffer = io.BytesIO()
+        qr.save(qr_buffer, format="PNG")
+        qr_buffer.seek(0)
+        qr_img = ImageReader(qr_buffer)
+        c.drawImage(qr_img, W - 120, H - 140, width=80, height=80)
     except:
         pass
 
-    # QR CODE (top right)
-    qr = qrcode.make("https://www.tourdefierce.vip/audio-clapback-engine-ai-detector-autotune-detector")
-    qr_buffer = io.BytesIO()
-    qr.save(qr_buffer)
-    qr_buffer.seek(0)
-    c.drawImage(ImageReader(qr_buffer), width - 150, height - 250, width=90, height=90)
-
-    # line
-    c.setStrokeColor(colors.lightgrey)
+    # Divider line
+    c.setStrokeColor(LIGHT_GRAY)
     c.setLineWidth(1)
-    c.line(40, height - 260, width - 40, height - 260)
-
-    # ======================================
-    # METRIC BOXES
-    # ======================================
-
-    def draw_metric(label, value, x, color):
-        c.setFillColor(color)
-        c.rect(x, height - 340, 160, 60, fill=1, stroke=0)
-        c.setFillColor(colors.white)
-        c.setFont("Helvetica-Bold", 12)
-        c.drawString(x + 10, height - 315, label)
-        c.setFont("Helvetica-Bold", 20)
-        c.drawString(x + 10, height - 335, f"{value}%")
-
-    draw_metric("AI Likelihood", ai_like, 50, metric_color(ai_like, "ai"))
-    draw_metric("Human Likelihood", human_like, 240, metric_color(human_like, "human"))
-
-    # autotune block
-    c.setFillColor(metric_color(auto_index, "auto"))
-    c.rect(430, height - 340, 160, 60, fill=1, stroke=0)
-    c.setFillColor(colors.white)
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(440, height - 315, "Autotune Index")
-    c.setFont("Helvetica-Bold", 20)
-    c.drawString(440, height - 335, f"{auto_index}/100")
-
-    # ======================================
-    # SHADE METER (OWN SECTION)
-    # ======================================
+    c.line(40, H - 160, W - 40, H - 160)
+
+    # -------------------------------
+    # SCORE BOXES (AI / HUMAN / AUTOTUNE)
+    # -------------------------------
+
+    def score_color(score, invert=False):
+        """
+        invert=False → green=human good / red=AI high
+        invert=True  → green=AI low / red=AI high
+        """
+        if not invert:
+            # human
+            if score >= 75:
+                return colors.green
+            if score >= 25:
+                return colors.yellow
+            return colors.red
+        else:
+            # AI
+            if score <= 25:
+                return colors.green
+            if score <= 75:
+                return colors.yellow
+            return colors.red
+
+    # AI %
+    c.setFillColor(score_color(ai_score, invert=True))
+    c.rect(40, H - 250, 150, 80, fill=1)
+    c.setFillColor(WHITE)
+    c.setFont("Helvetica", 11)
+    c.drawString(55, H - 190, "AI Likelihood")
+    c.setFont("PlayfairDisplay-Bold", 28)
+    c.drawString(55, H - 215, f"{ai_score:.1f}%")
+
+    # HUMAN %
+    c.setFillColor(score_color(human_score))
+    c.rect(210, H - 250, 150, 80, fill=1)
+    c.setFillColor(WHITE)
+    c.setFont("Helvetica", 11)
+    c.drawString(225, H - 190, "Human Likelihood")
+    c.setFont("PlayfairDisplay-Bold", 28)
+    c.drawString(225, H - 215, f"{human_score:.1f}%")
+
+    # AUTOTUNE
+    c.setFillColor(colors.orange)
+    c.rect(380, H - 250, 150, 80, fill=1)
+    c.setFillColor(WHITE)
+    c.setFont("Helvetica", 11)
+    c.drawString(395, H - 190, "Autotune Index")
+    c.setFont("PlayfairDisplay-Bold", 28)
+    c.drawString(395, H - 215, f"{atune:.1f}/100")
+
+    # -------------------------------
+    # SHADE METER (Luxury Bar)
+    # -------------------------------
+    c.setFont("Helvetica", 11)
+    c.setFillColor(BLACK)
+    c.drawString(40, H - 280, "Shade Meter")
+    c.setFont("Helvetica", 9)
+    c.drawString(40, H - 292, "0 = no shade, 100 = maximum clapback")
+
+    # meter background
+    c.setFillColor(LIGHT_GRAY)
+    c.rect(40, H - 305, 490, 12, fill=1, stroke=0)
+
+    # meter fill
     c.setFillColor(ACCENT)
-    c.roundRect(50, height - 430, 540, 60, 8, fill=0, stroke=1)
-
-    c.setFont("Helvetica-Bold", 12)
-    c.setFillColor(AUBERGINE)
-    c.drawString(65, height - 385, f"Shade Meter: {shade}/100")
+    fill_width = (shade / 100) * 490
+    c.rect(40, H - 305, fill_width, 12, fill=1, stroke=0)
 
+    c.setFillColor(BLACK)
     c.setFont("Helvetica", 10)
-    c.setFillColor(colors.black)
-    c.drawString(65, height - 400,
-                 "The Shade Meter reflects the clip’s transparency score: higher = shadier.")
-
-    verdict = ""
-    if shade < 20: verdict = "You avoided the shade parade — clean & clear!"
-    elif shade < 50: verdict = "A lil’ shady lady moment — but nothing wild."
-    else: verdict = "Oop! You got sprayed with shade — this clip is looking sus."
-
-    c.setFont("Helvetica-Oblique", 10)
-    c.drawString(65, height - 415, verdict)
-
-    # ======================================
-    # BREAKDOWN SECTION
-    # ======================================
-    cursor = height - 460
-    c.setFont("PlayfairBold", 16)
-    c.setFillColor(AUBERGINE)
-    c.drawString(50, cursor, "Full Forensic Breakdown:")
-    cursor -= 25
-
-    text = c.beginText(50, cursor)
-    text.setFont("Helvetica", 10)
-    text.setLeading(14)
-
-    multi_voice_note = ""
-    if "multi" in voice_range.lower():
-        multi_voice_note = "Multiple voices detected — metrics averaged."
-
-    midi_note = "Instrument timing is too perfect — likely MIDI or programmed." if is_midi else \
-                "Timing variance suggests human-played instrumentation."
-
-    transcript_note = transcript if transcript else "(No transcript available)"
-
-    narrative = f"""
-AI Likelihood: {ai_like}%. Human Likelihood: {human_like}%.
-Autotune Index: {auto_index}/100. Shade Meter: {shade}/100.
-Detected Vocal Style: {style}. Clapback Tone: {tone}.
+    c.drawString(540, H - 304, f"{shade:.1f} / 100")
 
-Key Analysis: {key if key else "Unknown"}.
-Estimated BPM: {bpm if bpm else "N/A"}.
-Piece Voice Range Classification: {voice_range}. {multi_voice_note}
-
-Timing & Instrumentation: {midi_note}
-
-Transcript Sample:
-"{transcript_note}"
-
-Interpretation:
-This forensic-style analysis blends embeddings, pitch stability, and spectral norms to judge how
-organic or synthetic the performance may be. Lower jitter & shimmer often indicate machine-made
-precision, whereas organic vocals tend to include human irregularities. The Shade Meter synthesizes
-all metrics into a transparency index. Treat this as expert guidance, not legal certification.
-"""
-
-    for line in narrative.split("\n"):
-        text.textLine(line.strip())
-
-    c.drawText(text)
-
-    # footer
+    # Shade text
+    c.setFont("Helvetica", 10)
+    c.setFillColor(BLACK)
+    if shade < 10:
+        shade_text = "You're practically a saint — no shade detected!"
+    elif shade < 40:
+        shade_text = "A lil’ sprinkle of shade, but nothing major."
+    elif shade < 75:
+        shade_text = "Okay shady lady! There are *elements* to discuss."
+    else:
+        shade_text = "Girl… you were sprayed with shade. Maximum clapback."
+
+    c.drawString(40, H - 325, shade_text)
+
+    # -------------------------------
+    # ANALYSIS SUMMARY
+    # -------------------------------
+
+    c.setFont("PlayfairDisplay-Bold", 18)
+    c.setFillColor(PRIMARY)
+    c.drawString(40, H - 360, "Analysis Summary")
+
+    c.setFont("Helvetica", 11)
+    c.setFillColor(BLACK)
+    c.drawString(40, H - 385, f"Detected Vocal Style:     {style}")
+    c.drawString(40, H - 402, f"Clapback Style:           {tone}")
+    c.drawString(40, H - 419, f"Shade Meter:              {shade:.1f}/100")
+
+    # -------------------------------
+    # FORENSIC BREAKDOWN
+    # -------------------------------
+
+    c.setFont("PlayfairDisplay-Bold", 18)
+    c.setFillColor(PRIMARY)
+    c.drawString(40, H - 455, "Full Forensic Breakdown:")
+
+    text_y = H - 480
+    c.setFont("Helvetica", 10)
+    c.setFillColor(BLACK)
+
+    for line in transcript.replace("\n", " ").split(". "):
+        c.drawString(40, text_y, line.strip() + ".")
+        text_y -= 14
+        if text_y < 60:
+            c.showPage()
+            text_y = H - 60
+            c.setFont("Helvetica", 10)
+
+    # Footer
+    c.setStrokeColor(LIGHT_GRAY)
+    c.line(40, 50, W - 40, 50)
     c.setFont("Helvetica", 8)
-    c.setFillColor(colors.grey)
-    c.drawString(50, 30, "© 2025 Tour de Fierce™ — All Shade. No Shame. www.tourdefierce.vip")
+    c.setFillColor(BLACK)
+    c.drawString(40, 37, "© 2025 Tour de Fierce™ — All Shade, No Shame.")
+    c.drawString(300, 37, "www.tourdefierce.vip")
 
-    c.showPage()
     c.save()
+    buffer.seek(0)
 
-    return out.name
-
-# ======================================
-#  MAIN ANALYSIS FUNCTION
-# ======================================
-def analyze(audio_file, tone):
-    if audio_file is None:
-        return "Upload an audio file", None
-
-    filename = os.path.basename(audio_file)
-
-    y, sr = librosa.load(audio_file, sr=44100)
-    pitches, _ = librosa.piptrack(y=y, sr=sr)
-    detected_pitches = pitches[pitches > 0]
-
-    # fake ai/human for now
-    ai_like = np.random.uniform(5, 95)
-    human_like = 100 - ai_like
-    auto_index = round(np.random.uniform(0, 60), 1)
-
-    shade = shade_score(ai_like)
-    voice_range = detect_voice_range(detected_pitches)
-    key = detect_key(y, sr)
-    bpm = detect_bpm(y, sr)
-    is_midi = detect_midi_like(y)
-
-    transcript = "Transcript unavailable in preview."
-
-    pdf = generate_pdf(filename, round(ai_like), round(human_like),
-                       auto_index, shade, "Auto Style", tone,
-                       key, bpm, voice_range, is_midi, transcript)
-
-    summary = f"""
-AI: {round(ai_like)}%
-Human: {round(human_like)}%
-Autotune: {auto_index}/100
-Shade: {shade}/100
-Key: {key}
-BPM: {bpm}
-Voice Range: {voice_range}
-"""
-
-    return summary, pdf
-
-# ======================================
-#  UI
-# ======================================
-
-with gr.Blocks(css="""
-#title { font-family: 'Playfair Display', serif; font-weight: 700; font-size: 38px; color: #3B0B3F; }
-""") as iface:
-
-    gr.HTML("<h1 id='title'>👏 Tour de Fierce Audio Clapback Engine™</h1>")
-    gr.HTML("<p style='color:white;'>Detects AI Vocals • Autotune • Shade Levels • Audio Forensic Reports</p>")
-
-    audio = gr.Audio(type="filepath", label="Upload Audio File (Max 45s)")
-
-    tone = gr.Radio(["professional", "sassy", "forensic"], value="professional", label="Clapback Tone")
-
-    summary = gr.Textbox(label="Clapback Summary")
-    pdf_output = gr.File(label="Full PDF Report")
-
-    run = gr.Button("Run Clapback 👏", variant="primary")
-
-    run.click(analyze, inputs=[audio, tone], outputs=[summary, pdf_output])
-
-iface.launch()
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    tmp.write(buffer.getvalue())
+    tmp.close()
+    return tmp.name