judith.ibanez commited on
Commit Β·
1c5ed00
1
Parent(s): 15e4ee8
add some modifications
Browse files
app.py
CHANGED
|
@@ -365,14 +365,310 @@ def recognize_music_gradio(pdf_file):
|
|
| 365 |
traceback.print_exc()
|
| 366 |
return None
|
| 367 |
|
| 368 |
-
# Create Gradio interface
|
| 369 |
-
|
| 370 |
-
|
| 371 |
-
|
| 372 |
-
|
| 373 |
-
|
| 374 |
-
|
| 375 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 376 |
|
| 377 |
# Removed run_gradio function - now running directly in main thread
|
| 378 |
|
|
|
|
| 365 |
traceback.print_exc()
|
| 366 |
return None
|
| 367 |
|
| 368 |
+
# Create enhanced Gradio interface with custom CSS and better UX
|
| 369 |
+
custom_css = """
|
| 370 |
+
.gradio-container {
|
| 371 |
+
max-width: 1200px !important;
|
| 372 |
+
margin: auto !important;
|
| 373 |
+
}
|
| 374 |
+
|
| 375 |
+
.main-header {
|
| 376 |
+
text-align: center;
|
| 377 |
+
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
|
| 378 |
+
color: white;
|
| 379 |
+
padding: 2rem;
|
| 380 |
+
border-radius: 15px;
|
| 381 |
+
margin-bottom: 2rem;
|
| 382 |
+
box-shadow: 0 8px 32px rgba(0,0,0,0.1);
|
| 383 |
+
}
|
| 384 |
+
|
| 385 |
+
.feature-card {
|
| 386 |
+
background: white;
|
| 387 |
+
border-radius: 12px;
|
| 388 |
+
padding: 1.5rem;
|
| 389 |
+
margin: 1rem 0;
|
| 390 |
+
box-shadow: 0 4px 16px rgba(0,0,0,0.1);
|
| 391 |
+
border-left: 4px solid #667eea;
|
| 392 |
+
}
|
| 393 |
+
|
| 394 |
+
.upload-area {
|
| 395 |
+
border: 2px dashed #667eea;
|
| 396 |
+
border-radius: 12px;
|
| 397 |
+
padding: 2rem;
|
| 398 |
+
text-align: center;
|
| 399 |
+
background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
|
| 400 |
+
transition: all 0.3s ease;
|
| 401 |
+
}
|
| 402 |
+
|
| 403 |
+
.upload-area:hover {
|
| 404 |
+
border-color: #764ba2;
|
| 405 |
+
background: linear-gradient(135deg, #e3f2fd 0%, #bbdefb 100%);
|
| 406 |
+
}
|
| 407 |
+
|
| 408 |
+
.status-success {
|
| 409 |
+
background: linear-gradient(135deg, #4caf50 0%, #45a049 100%);
|
| 410 |
+
color: white;
|
| 411 |
+
padding: 1rem;
|
| 412 |
+
border-radius: 8px;
|
| 413 |
+
margin: 1rem 0;
|
| 414 |
+
}
|
| 415 |
+
|
| 416 |
+
.status-error {
|
| 417 |
+
background: linear-gradient(135deg, #f44336 0%, #d32f2f 100%);
|
| 418 |
+
color: white;
|
| 419 |
+
padding: 1rem;
|
| 420 |
+
border-radius: 8px;
|
| 421 |
+
margin: 1rem 0;
|
| 422 |
+
}
|
| 423 |
+
|
| 424 |
+
.info-box {
|
| 425 |
+
background: linear-gradient(135deg, #e3f2fd 0%, #bbdefb 100%);
|
| 426 |
+
border-radius: 8px;
|
| 427 |
+
padding: 1rem;
|
| 428 |
+
margin: 1rem 0;
|
| 429 |
+
border-left: 4px solid #2196f3;
|
| 430 |
+
}
|
| 431 |
+
"""
|
| 432 |
+
|
| 433 |
+
def create_gradio_interface():
|
| 434 |
+
with gr.Blocks(css=custom_css, title="πΌ Audiveris Music Score Recognition", theme=gr.themes.Soft()) as interface:
|
| 435 |
+
# Header
|
| 436 |
+
gr.HTML("""
|
| 437 |
+
<div class="main-header">
|
| 438 |
+
<h1>πΌ Audiveris Music Score Recognition</h1>
|
| 439 |
+
<p style="font-size: 1.2em; margin-top: 1rem; opacity: 0.9;">
|
| 440 |
+
Transform your PDF music scores into editable MusicXML files using advanced AI recognition
|
| 441 |
+
</p>
|
| 442 |
+
</div>
|
| 443 |
+
""")
|
| 444 |
+
|
| 445 |
+
# Main content area
|
| 446 |
+
with gr.Row():
|
| 447 |
+
with gr.Column(scale=1):
|
| 448 |
+
gr.HTML("""
|
| 449 |
+
<div class="feature-card">
|
| 450 |
+
<h3>β¨ Features</h3>
|
| 451 |
+
<ul style="line-height: 1.8;">
|
| 452 |
+
<li>π΅ High-accuracy music notation recognition</li>
|
| 453 |
+
<li>π PDF to MusicXML conversion</li>
|
| 454 |
+
<li>πΉ Supports complex musical scores</li>
|
| 455 |
+
<li>β‘ Fast processing with Audiveris engine</li>
|
| 456 |
+
<li>πΎ Downloadable results</li>
|
| 457 |
+
</ul>
|
| 458 |
+
</div>
|
| 459 |
+
""")
|
| 460 |
+
|
| 461 |
+
gr.HTML("""
|
| 462 |
+
<div class="info-box">
|
| 463 |
+
<h4>π How to use:</h4>
|
| 464 |
+
<ol style="line-height: 1.6;">
|
| 465 |
+
<li>Upload your PDF music score</li>
|
| 466 |
+
<li>Click "π΅ Convert to MusicXML"</li>
|
| 467 |
+
<li>Wait for processing to complete</li>
|
| 468 |
+
<li>Download your MusicXML file</li>
|
| 469 |
+
</ol>
|
| 470 |
+
</div>
|
| 471 |
+
""")
|
| 472 |
+
|
| 473 |
+
with gr.Column(scale=2):
|
| 474 |
+
# File upload section
|
| 475 |
+
gr.HTML("<h3 style='text-align: center; color: #667eea;'>π Upload Your Music Score</h3>")
|
| 476 |
+
|
| 477 |
+
pdf_input = gr.File(
|
| 478 |
+
file_types=[".pdf"],
|
| 479 |
+
label="Select PDF File",
|
| 480 |
+
file_count="single",
|
| 481 |
+
height=200,
|
| 482 |
+
elem_classes=["upload-area"]
|
| 483 |
+
)
|
| 484 |
+
|
| 485 |
+
# Processing button
|
| 486 |
+
convert_btn = gr.Button(
|
| 487 |
+
"π΅ Convert to MusicXML",
|
| 488 |
+
variant="primary",
|
| 489 |
+
size="lg",
|
| 490 |
+
scale=1
|
| 491 |
+
)
|
| 492 |
+
|
| 493 |
+
# Status and progress
|
| 494 |
+
status_display = gr.HTML(visible=False)
|
| 495 |
+
progress_bar = gr.Progress()
|
| 496 |
+
|
| 497 |
+
# Output section
|
| 498 |
+
gr.HTML("<h3 style='text-align: center; color: #667eea; margin-top: 2rem;'>π₯ Download Results</h3>")
|
| 499 |
+
|
| 500 |
+
output_file = gr.File(
|
| 501 |
+
label="MusicXML Output",
|
| 502 |
+
visible=False,
|
| 503 |
+
height=100
|
| 504 |
+
)
|
| 505 |
+
|
| 506 |
+
# Processing info
|
| 507 |
+
processing_info = gr.Textbox(
|
| 508 |
+
label="Processing Details",
|
| 509 |
+
lines=8,
|
| 510 |
+
visible=False,
|
| 511 |
+
interactive=False
|
| 512 |
+
)
|
| 513 |
+
|
| 514 |
+
# Footer
|
| 515 |
+
gr.HTML("""
|
| 516 |
+
<div style="text-align: center; margin-top: 3rem; padding: 2rem; background: #f8f9fa; border-radius: 12px;">
|
| 517 |
+
<p style="color: #666; margin: 0;">
|
| 518 |
+
Powered by <strong>Audiveris</strong> β’ Built with β€οΈ using Gradio
|
| 519 |
+
</p>
|
| 520 |
+
<p style="color: #888; font-size: 0.9em; margin-top: 0.5rem;">
|
| 521 |
+
For best results, use high-quality PDF scans with clear musical notation
|
| 522 |
+
</p>
|
| 523 |
+
</div>
|
| 524 |
+
""")
|
| 525 |
+
|
| 526 |
+
# Enhanced processing function with better feedback
|
| 527 |
+
def process_with_feedback(pdf_file, progress=gr.Progress()):
|
| 528 |
+
if pdf_file is None:
|
| 529 |
+
return (
|
| 530 |
+
gr.HTML("<div class='status-error'>β Please upload a PDF file first!</div>", visible=True),
|
| 531 |
+
None,
|
| 532 |
+
gr.Textbox(visible=False),
|
| 533 |
+
gr.File(visible=False)
|
| 534 |
+
)
|
| 535 |
+
|
| 536 |
+
try:
|
| 537 |
+
# Show processing status
|
| 538 |
+
progress(0.1, desc="π Analyzing PDF file...")
|
| 539 |
+
|
| 540 |
+
status_html = """
|
| 541 |
+
<div class='status-success'>
|
| 542 |
+
<h4>π Processing your music score...</h4>
|
| 543 |
+
<p>File: <strong>{}</strong></p>
|
| 544 |
+
<p>Size: <strong>{:.2f} MB</strong></p>
|
| 545 |
+
<p>Please wait while Audiveris analyzes your score...</p>
|
| 546 |
+
</div>
|
| 547 |
+
""".format(
|
| 548 |
+
pdf_file.name.split('/')[-1],
|
| 549 |
+
os.path.getsize(pdf_file.name) / (1024*1024)
|
| 550 |
+
)
|
| 551 |
+
|
| 552 |
+
progress(0.3, desc="π΅ Running Audiveris recognition...")
|
| 553 |
+
|
| 554 |
+
# Process the file
|
| 555 |
+
result_file = recognize_music_gradio(pdf_file)
|
| 556 |
+
|
| 557 |
+
progress(0.9, desc="β
Finalizing results...")
|
| 558 |
+
|
| 559 |
+
if result_file and os.path.exists(result_file):
|
| 560 |
+
# Success
|
| 561 |
+
success_html = """
|
| 562 |
+
<div class='status-success'>
|
| 563 |
+
<h4>β
Conversion completed successfully!</h4>
|
| 564 |
+
<p>π Output: <strong>{}</strong></p>
|
| 565 |
+
<p>π Size: <strong>{:.2f} KB</strong></p>
|
| 566 |
+
<p>π Your MusicXML file is ready for download!</p>
|
| 567 |
+
</div>
|
| 568 |
+
""".format(
|
| 569 |
+
os.path.basename(result_file),
|
| 570 |
+
os.path.getsize(result_file) / 1024
|
| 571 |
+
)
|
| 572 |
+
|
| 573 |
+
# Processing details
|
| 574 |
+
details = f"""β
CONVERSION SUCCESSFUL
|
| 575 |
+
|
| 576 |
+
π Input File: {pdf_file.name.split('/')[-1]}
|
| 577 |
+
π Input Size: {os.path.getsize(pdf_file.name) / (1024*1024):.2f} MB
|
| 578 |
+
|
| 579 |
+
π΅ Output File: {os.path.basename(result_file)}
|
| 580 |
+
π Output Size: {os.path.getsize(result_file) / 1024:.2f} KB
|
| 581 |
+
|
| 582 |
+
β±οΈ Processing completed at: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
|
| 583 |
+
|
| 584 |
+
πΌ Your PDF music score has been successfully converted to MusicXML format!
|
| 585 |
+
You can now download the file and use it in music notation software like MuseScore, Finale, or Sibelius."""
|
| 586 |
+
|
| 587 |
+
progress(1.0, desc="π Complete!")
|
| 588 |
+
|
| 589 |
+
return (
|
| 590 |
+
gr.HTML(success_html, visible=True),
|
| 591 |
+
gr.File(result_file, visible=True),
|
| 592 |
+
gr.Textbox(details, visible=True),
|
| 593 |
+
gr.File(visible=True)
|
| 594 |
+
)
|
| 595 |
+
else:
|
| 596 |
+
# Failure
|
| 597 |
+
error_html = """
|
| 598 |
+
<div class='status-error'>
|
| 599 |
+
<h4>β Conversion failed</h4>
|
| 600 |
+
<p>The music recognition process encountered an error.</p>
|
| 601 |
+
<p>Please check that your PDF contains clear musical notation and try again.</p>
|
| 602 |
+
</div>
|
| 603 |
+
"""
|
| 604 |
+
|
| 605 |
+
error_details = f"""β CONVERSION FAILED
|
| 606 |
+
|
| 607 |
+
π Input File: {pdf_file.name.split('/')[-1]}
|
| 608 |
+
π Input Size: {os.path.getsize(pdf_file.name) / (1024*1024):.2f} MB
|
| 609 |
+
|
| 610 |
+
β οΈ Error: No output file was generated by Audiveris
|
| 611 |
+
β±οΈ Failed at: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
|
| 612 |
+
|
| 613 |
+
π‘ Troubleshooting tips:
|
| 614 |
+
β’ Ensure your PDF contains clear, high-quality musical notation
|
| 615 |
+
β’ Check that the PDF is not password-protected
|
| 616 |
+
β’ Try with a different PDF file
|
| 617 |
+
β’ Make sure the musical notation is not handwritten"""
|
| 618 |
+
|
| 619 |
+
return (
|
| 620 |
+
gr.HTML(error_html, visible=True),
|
| 621 |
+
None,
|
| 622 |
+
gr.Textbox(error_details, visible=True),
|
| 623 |
+
gr.File(visible=False)
|
| 624 |
+
)
|
| 625 |
+
|
| 626 |
+
except Exception as e:
|
| 627 |
+
# Exception handling
|
| 628 |
+
error_html = f"""
|
| 629 |
+
<div class='status-error'>
|
| 630 |
+
<h4>β Processing Error</h4>
|
| 631 |
+
<p>An unexpected error occurred: <code>{str(e)}</code></p>
|
| 632 |
+
<p>Please try again or contact support if the problem persists.</p>
|
| 633 |
+
</div>
|
| 634 |
+
"""
|
| 635 |
+
|
| 636 |
+
error_details = f"""β PROCESSING ERROR
|
| 637 |
+
|
| 638 |
+
π Input File: {pdf_file.name.split('/')[-1] if pdf_file else 'Unknown'}
|
| 639 |
+
β οΈ Error: {str(e)}
|
| 640 |
+
β±οΈ Failed at: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
|
| 641 |
+
|
| 642 |
+
π§ Technical Details:
|
| 643 |
+
{str(e)}
|
| 644 |
+
|
| 645 |
+
Please try again with a different file or contact support."""
|
| 646 |
+
|
| 647 |
+
return (
|
| 648 |
+
gr.HTML(error_html, visible=True),
|
| 649 |
+
None,
|
| 650 |
+
gr.Textbox(error_details, visible=True),
|
| 651 |
+
gr.File(visible=False)
|
| 652 |
+
)
|
| 653 |
+
|
| 654 |
+
# Connect the button to the processing function
|
| 655 |
+
convert_btn.click(
|
| 656 |
+
fn=process_with_feedback,
|
| 657 |
+
inputs=[pdf_input],
|
| 658 |
+
outputs=[status_display, output_file, processing_info, output_file],
|
| 659 |
+
show_progress=True
|
| 660 |
+
)
|
| 661 |
+
|
| 662 |
+
# Auto-hide status when new file is uploaded
|
| 663 |
+
pdf_input.change(
|
| 664 |
+
fn=lambda: (gr.HTML(visible=False), gr.Textbox(visible=False), gr.File(visible=False)),
|
| 665 |
+
outputs=[status_display, processing_info, output_file]
|
| 666 |
+
)
|
| 667 |
+
|
| 668 |
+
return interface
|
| 669 |
+
|
| 670 |
+
# Create the enhanced interface
|
| 671 |
+
gradio_interface = create_gradio_interface()
|
| 672 |
|
| 673 |
# Removed run_gradio function - now running directly in main thread
|
| 674 |
|