Spaces:
Sleeping
Sleeping
Update report_generator.py
Browse files- report_generator.py +10 -10
report_generator.py
CHANGED
|
@@ -162,7 +162,7 @@ def create_report(audio_data, output_path):
|
|
| 162 |
)
|
| 163 |
|
| 164 |
# ============================================================
|
| 165 |
-
# 4. ENERGY
|
| 166 |
# ============================================================
|
| 167 |
|
| 168 |
ax_bar = fig.add_subplot(gs[2, :])
|
|
@@ -177,7 +177,8 @@ def create_report(audio_data, output_path):
|
|
| 177 |
e["2k_8khz"], e["8k_12khz"], e["12k_16khz"], e["above_16khz"]
|
| 178 |
]
|
| 179 |
|
| 180 |
-
colors = ["#2C3E50", "#E74C3C", "#E67E22",
|
|
|
|
| 181 |
|
| 182 |
bars = ax_bar.bar(bands, vals, color=colors, edgecolor="black", alpha=0.85)
|
| 183 |
|
|
@@ -196,10 +197,10 @@ def create_report(audio_data, output_path):
|
|
| 196 |
|
| 197 |
issues = audio_data["issues"]
|
| 198 |
|
| 199 |
-
|
| 200 |
|
| 201 |
if not issues:
|
| 202 |
-
|
| 203 |
else:
|
| 204 |
icons = {
|
| 205 |
"CRITICAL": "π΄",
|
|
@@ -208,20 +209,20 @@ def create_report(audio_data, output_path):
|
|
| 208 |
"LOW": "π’"
|
| 209 |
}
|
| 210 |
for issue, sev, desc in issues:
|
| 211 |
-
|
| 212 |
-
|
| 213 |
|
| 214 |
if spec["spectral_notches"]:
|
| 215 |
-
|
| 216 |
"",
|
| 217 |
f"π΅ Spectral Notches: {len(spec['spectral_notches'])}",
|
| 218 |
]
|
| 219 |
for i, n in enumerate(spec["spectral_notches"][:5], 1):
|
| 220 |
-
|
| 221 |
|
| 222 |
ax_issues.text(
|
| 223 |
0.05, 0.95,
|
| 224 |
-
"\n".join(
|
| 225 |
fontsize=10.8,
|
| 226 |
va="top",
|
| 227 |
family="monospace",
|
|
@@ -267,7 +268,6 @@ def create_report(audio_data, output_path):
|
|
| 267 |
f"Low: {s['low']}",
|
| 268 |
]
|
| 269 |
|
| 270 |
-
# Add synthetic visual block (your Option 3)
|
| 271 |
score_lines += [
|
| 272 |
"",
|
| 273 |
"βββββββββββββββββββββββ",
|
|
|
|
| 162 |
)
|
| 163 |
|
| 164 |
# ============================================================
|
| 165 |
+
# 4. ENERGY BAR CHART
|
| 166 |
# ============================================================
|
| 167 |
|
| 168 |
ax_bar = fig.add_subplot(gs[2, :])
|
|
|
|
| 177 |
e["2k_8khz"], e["8k_12khz"], e["12k_16khz"], e["above_16khz"]
|
| 178 |
]
|
| 179 |
|
| 180 |
+
colors = ["#2C3E50", "#E74C3C", "#E67E22",
|
| 181 |
+
"#F39C12", "#2ECC71", "#3498DB", "#9B59B6"]
|
| 182 |
|
| 183 |
bars = ax_bar.bar(bands, vals, color=colors, edgecolor="black", alpha=0.85)
|
| 184 |
|
|
|
|
| 197 |
|
| 198 |
issues = audio_data["issues"]
|
| 199 |
|
| 200 |
+
issue_lines = ["DETECTED ISSUES", "β" * 80]
|
| 201 |
|
| 202 |
if not issues:
|
| 203 |
+
issue_lines.append("β
No significant issues detected.")
|
| 204 |
else:
|
| 205 |
icons = {
|
| 206 |
"CRITICAL": "π΄",
|
|
|
|
| 209 |
"LOW": "π’"
|
| 210 |
}
|
| 211 |
for issue, sev, desc in issues:
|
| 212 |
+
issue_lines.append(f"{icons.get(sev,'βͺ')} [{sev}] {issue}")
|
| 213 |
+
issue_lines.append(f" β {desc}")
|
| 214 |
|
| 215 |
if spec["spectral_notches"]:
|
| 216 |
+
issue_lines += [
|
| 217 |
"",
|
| 218 |
f"π΅ Spectral Notches: {len(spec['spectral_notches'])}",
|
| 219 |
]
|
| 220 |
for i, n in enumerate(spec["spectral_notches"][:5], 1):
|
| 221 |
+
issue_lines.append(f" {i}. {n['freq']:.1f} Hz (Depth {n['depth_db']:.1f} dB)")
|
| 222 |
|
| 223 |
ax_issues.text(
|
| 224 |
0.05, 0.95,
|
| 225 |
+
"\n".join(issue_lines),
|
| 226 |
fontsize=10.8,
|
| 227 |
va="top",
|
| 228 |
family="monospace",
|
|
|
|
| 268 |
f"Low: {s['low']}",
|
| 269 |
]
|
| 270 |
|
|
|
|
| 271 |
score_lines += [
|
| 272 |
"",
|
| 273 |
"βββββββββββββββββββββββ",
|