Update app.py

app.py

@@ -1,29 +1,60 @@
 import gradio as gr
-from chord_identifier import identify_chord
 from huggingface_hub import InferenceClient
 
-# Load Hugging Face API client (example: flan-t5-small)
+# --- Rule-based chord dictionary ---
+CHORDS = {
+    frozenset(["C", "E", "G"]): "C Major",
+    frozenset(["A", "C#", "E"]): "A Major",
+    frozenset(["A", "C", "E"]): "A Minor",
+    frozenset(["D", "F#", "A"]): "D Major",
+    frozenset(["E", "G#", "B"]): "E Major",
+    frozenset(["G", "B", "D"]): "G Major",
+    frozenset(["F", "A", "C"]): "F Major",
+}
+
+def identify_chord(notes):
+    """
+    Identify chord name from a list of notes using dictionary lookup.
+    """
+    key = frozenset([n.upper() for n in notes])
+    return CHORDS.get(key, "Unknown Chord")
+
+# --- Hugging Face model client (API-based) ---
 client = InferenceClient("google/flan-t5-small")
 
 def predict_chord(notes: str):
-    # Split comma-separated input
-    note_list = [n.strip() for n in notes.split(",")]
+    """
+    Takes a comma-separated string of notes and predicts the chord.
+    First tries rule-based, then falls back to the LLM.
+    """
+    if not notes.strip():
+        return "Please enter 2 or more notes."
+
+    # Clean and split notes
+    note_list = [n.strip() for n in notes.split(",") if n.strip()]
+    if len(note_list) < 2:
+        return "Please enter at least 2 notes."
 
-    # Try rule-based
+    # Rule-based chord lookup
     chord = identify_chord(note_list)
     if chord != "Unknown Chord":
         return chord
 
-    # Fall back to LLM for reasoning
+    # Fall back to LLM
     prompt = f"Identify the musical chord made of notes: {', '.join(note_list)}"
-    response = client.text_generation(prompt, max_new_tokens=20)
-    return response.strip()
+    try:
+        response = client.text_generation(prompt, max_new_tokens=20)
+        return response.strip()
+    except Exception as e:
+        return f"Error calling Hugging Face API: {e}"
 
+# --- Gradio Interface ---
 demo = gr.Interface(
     fn=predict_chord,
     inputs=gr.Textbox(lines=1, placeholder="Enter notes, e.g., C,E,G"),
     outputs="text",
-    title="Chord Identifier (API-based)"
+    title="Chord Identifier (API-based)",
+    description="Enter two or more musical notes (comma-separated) and get the chord name."
 )
 
 if __name__ == "__main__":