Update app.py

app.py

@@ -72,27 +72,59 @@ def analyze_audio(audio_file, progress=gr.Progress()):
         # SCORING
         # ======================================================
         progress(0.82, desc="Scoring...")
-
+        
         # Count issues by severity
         critical = sum(1 for _, sev, _ in issues if sev == "CRITICAL")
         high = sum(1 for _, sev, _ in issues if sev == "HIGH")
         medium = sum(1 for _, sev, _ in issues if sev == "MEDIUM")
+        low = sum(1 for _, sev, _ in issues if sev == "LOW")
         
-        # Original scoring formula preserved exactly
+        # Score formula
         score = 100 - (critical * 30) - (high * 15) - (medium * 5)
         score = max(0, score)
         
-        # QUALITY LABELS (original logic)
+        # Grade + Quality Label (same logic as your report)
         if score >= 90:
-            grade, quality, color = "A", "EXCELLENT", "🟢"
+            grade, quality = "A", "EXCELLENT"
+            color = "🟢"
+            recommendation = "Excellent for TTS dataset"
         elif score >= 75:
-            grade, quality, color = "B", "GOOD", "🟢"
+            grade, quality = "B", "GOOD"
+            color = "🟢"
+            recommendation = "Very good quality; suitable for TTS"
         elif score >= 60:
-            grade, quality, color = "C", "FAIR", "🟡"
+            grade, quality = "C", "FAIR"
+            color = "🟡"
+            recommendation = "Usable but may contain processing artifacts"
         elif score >= 40:
-            grade, quality, color = "D", "POOR", "🟠"
+            grade, quality = "D", "POOR"
+            color = "🟠"
+            recommendation = "Not recommended for TTS (heavy processing)"
         else:
-            grade, quality, color = "F", "CRITICAL", "🔴"
+            grade, quality = "F", "CRITICAL"
+            color = "🔴"
+            recommendation = "Severely degraded or processed; avoid for TTS"
+        
+        # Cleanliness score (needed by report_generator)
+        cleanliness_score = max(0, 100 - (medium * 5 + low * 3))
+        
+        # Processing severity index (needed by report_generator)
+        processing_severity = (critical * 3) + (high * 2) + medium
+        
+        audio_data["score"] = {
+            "score": score,
+            "grade": grade,
+            "quality": quality,
+            "recommendation": recommendation,
+            "cleanliness_score": cleanliness_score,
+            "processing_severity": processing_severity,
+            "critical": critical,
+            "high": high,
+            "medium": medium,
+            "low": low,
+            "color": color
+        }
+        
 
         # ======================================================
         # CREATE REPORT PNG