Update app.py

app.py

@@ -1,42 +1,27 @@
 import gradio as gr
-from PIL import Image
-from transformers import AutoProcessor, VisionEncoderDecoderModel
+from music21 import converter
 
-# Load model
-processor = AutoProcessor.from_pretrained("truelitmus/omr-crnn")
-model = VisionEncoderDecoderModel.from_pretrained("truelitmus/omr-crnn")
+def process_musicxml(file):
+    score = converter.parse(file.name)
+    notes = [str(n.pitch) for n in score.flat.notes if n.isNote]
+    unique_notes = sorted(set(notes))
 
-# Study course generator
-def generate_study_course(note_string):
-    note_list = note_string.split()
-    unique_notes = sorted(set(note_list))
-
-    course = {
+    study_plan = {
         "Scales": {
             "All Notes": unique_notes,
             "Suggested Practice": [f"{note} Major Scale" for note in unique_notes]
         },
         "Practice Exercises": {
-            "Sight Reading (4-note chunks)": [note_list[i:i+4] for i in range(0, len(note_list)-3, 4)],
+            "Sight Reading (4-note chunks)": [notes[i:i+4] for i in range(0, len(notes)-3, 4)],
             "Finger Exercises": [f"{note} Arpeggio" for note in unique_notes]
         }
     }
-    return course
-
-# Main function
-def process_sheet(image: Image.Image):
-    inputs = processor(images=image.convert("RGB"), return_tensors="pt")
-    output = model.generate(**inputs)
-    notes = processor.batch_decode(output, skip_special_tokens=True)[0]
-    return generate_study_course(notes)
-
-# Gradio interface
-demo = gr.Interface(
-    fn=process_sheet,
-    inputs=gr.Image(type="pil", label="Upload Sheet Music Image"),
-    outputs=gr.JSON(label="Generated Study Course"),
-    title="🎼 Sheet Music Study AI",
-    description="Upload a sheet music image to detect notes and receive a structured study course with scales and practice suggestions.",
-)
+    return study_plan
 
-demo.launch()
+gr.Interface(
+    fn=process_musicxml,
+    inputs=gr.File(file_types=[".musicxml", ".xml"], label="Upload MusicXML File"),
+    outputs=gr.JSON(label="Study Plan"),
+    title="🎼 MusicXML Study Generator",
+    description="Upload a MusicXML file to generate scales and practice routines."
+).launch()