Update app.py

app.py

@@ -3,32 +3,43 @@ from huggingface_hub import InferenceClient
 
 client = InferenceClient("microsoft/phi-4")
 
+# Step 1: Add a function that checks for genres in the user message
+def recommend_songs(message):
+    genre_map = {
+        "pop": ["'Feel Special' - TWICE", "'Dynamite' - BTS"],
+        "r&b": ["'Instagram' - DEAN", "'You, Clouds, Rain' - Heize"],
+        "hip hop": ["'Daechwita' - Agust D", "'ON' - BTS"],
+        "dance": ["'Next Level' - aespa", "'Hot' - Seventeen"],
+        "rock": ["'Time of Our Life' - DAY6"]
+    }
+
+    for genre, songs in genre_map.items():
+        if genre in message.lower():
+            return f"Here are some {genre} K-pop songs you might like:\n" + "\n".join(songs)
+    return None
+
+# Step 2: Update your respond function to check for recommendations first
 def respond(message, history):
-    messages = [{"role": "system", "content": "You are a friendly kpop chatbot."}]
+    # Check if we can recommend songs based on genre keywords
+    quick_rec = recommend_songs(message)
+    if quick_rec:
+        return quick_rec  # Return genre-based recommendations immediately
+
+    # Otherwise, call the language model for a general response
+    messages = [{"role": "system", "content": "You are a friendly chatbot who specializes in K-pop. Recommend songs, explain K-pop terms, and introduce groups based on user preferences."}]
+    
     if history:
         messages.extend(history)
+        
     messages.append({"role": "user", "content": message})
-    response = client.chat_completion(messages, max_tokens=100)
+    
+    response = client.chat_completion(
+        messages,
+        max_tokens=150  # you can increase tokens if you want longer replies
+    )
+    
     return response['choices'][0]['message']['content'].strip()
 
-spotify_iframe = """
-<div style="max-width: 720px; margin: 12px auto;">
-  <iframe
-    data-testid="embed-iframe"
-    style="border-radius:12px"
-    src="https://open.spotify.com/embed/playlist/2EoheVFjqIxgJMb8VnDRtZ?utm_source=generator&theme=0"
-    width="100%"
-    height="352"
-    frameborder="0"
-    allow="autoplay; clipboard-write; encrypted-media; fullscreen; picture-in-picture"
-    loading="lazy">
-  </iframe>
-</div>
-"""
-
-with gr.Blocks() as demo:
-    chatbot = gr.ChatInterface(respond, type="messages", title="Kpop Song Hunters")
-    gr.HTML(spotify_iframe)
-
-demo.launch()
+chatbot = gr.ChatInterface(respond, type="messages", title="KpopBot")
 
+chatbot.launch()