Update app.py

app.py

@@ -2,56 +2,22 @@ import gradio as gr
 import requests
 import os
 
-# -------------------------
-# Hugging Face API setup
-# -------------------------
-# Replace with your org/model name
-API_URL = "https://api-inference.huggingface.co/models/DS553-Music-Bot/chord_classifier"
-
-# Expect your HF token as an environment variable (set in HF Space secrets or GitHub Actions)
+API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-small"
 HF_TOKEN = os.environ.get("HF_TOKEN")
-
-if HF_TOKEN is None:
-    raise ValueError("❌ HF_TOKEN not set. Please add it to your Hugging Face Space or GitHub secrets.")
-
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-# -------------------------
-# Query function
-# -------------------------
-def query_chord(message: str, history: list[tuple[str, str]]):
-    """
-    Send note input to Hugging Face Inference API, return predicted chord.
-    """
-    payload = {"inputs": message}
-    try:
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
-    except Exception as e:
-        return f"❌ Request failed: {str(e)}"
-
+def chord_api(message: str, history: list[tuple[str,str]]):
+    prompt = f"What chord do these notes form? {message}"
+    payload = {"inputs": prompt}
+    response = requests.post(API_URL, headers=headers, json=payload)
     if response.status_code == 200:
-        try:
-            result = response.json()
-            # Expecting the model repo to return {"label": "Cmaj"} style JSON
-            if isinstance(result, dict) and "label" in result:
-                return f"🎵 Identified chord: **{result['label']}**"
-            else:
-                return f"⚠️ Unexpected API response: {result}"
-        except Exception as e:
-            return f"⚠️ Error parsing API response: {str(e)}"
-    else:
-        return f"❌ Error {response.status_code}: {response.text}"
-    print("DEBUG:", response.status_code, response.text)    
+        return response.json()[0]["generated_text"]
+    return f"❌ Error {response.status_code}: {response.text}"
 
-# -------------------------
-# Gradio ChatInterface
-# -------------------------
 chatbot = gr.ChatInterface(
-    fn=query_chord,
+    fn=chord_api,
     title="🎶 Chord Bot (API Version)",
-    description="Enter 2+ notes (e.g., C E G or Db F Ab C). "
-                "This version queries the Hugging Face Inference API "
-                "for chord classification."
+    description="This version calls Hugging Face’s Inference API (flan-t5-small)."
 )
 
 if __name__ == "__main__":