Update app.py

app.py

@@ -1,54 +1,34 @@
-import gradio as gr
-from transformers import pipeline
-
-# Example group recommendation database
-GROUP_DB = [
-    {"group": "Red Velvet", "style": "pop"},
-    {"group": "TWICE", "style": "pop"},
-    {"group": "BLACKPINK", "style": "pop"},
-    {"group": "BTS", "style": "pop"},
-    # Add more groups if you want!
-]
-
-# Use the BlenderBot small model (CPU & RAM friendly)
-chatbot = pipeline(
-    "text2text-generation", 
-    model="facebook/blenderbot-400M-distill", 
-    device=-1  # Force CPU (for Hugging Face CPU Spaces)
-)
-
-def custom_kpop_response(message, history):
-    """Gradio expects (message, history) and returns (response, history)."""
-    user_input_lower = message.lower()
-    greeting = ""
-
-    # Greeting logic
-    if any(greet in user_input_lower for greet in ["hi", "hello", "hey"]):
-        greeting = "Hi there! Welcome to K-pop world! "
-
-    # Custom K-pop pop recommendation
-    if ("taylor swift" in user_input_lower or 
-        ("pop" in user_input_lower and "kpop" in user_input_lower)):
-        recommendation = (
-            "If you like Taylor Swift's pop style, try K-pop groups like Red Velvet, "
-            "TWICE, or BLACKPINK! They're known for catchy melodies and energetic choreography."
-        )
-        bot_reply = greeting + recommendation
-        history.append((message, bot_reply))
-        return "", history
-
-    # Fallback to BlenderBot for other questions
-    result = chatbot(message, max_length=100)
-    bot_reply = greeting + result[0]['generated_text']
-    history.append((message, bot_reply))
-    return "", history
-
-iface = gr.ChatInterface(
-    fn=custom_kpop_response,
-    title="K-Pop Genie Chatbot",
-    description="Say hi! Ask for K-pop groups, or songs similar to your Western favorites."
-)
-
-if __name__ == "__main__":
-    iface.launch()
+iimport gradio as gr
+from huggingface_hub import InferenceClient
+
+client = InferenceClient("microsoft/phi-4")
+
+def respond(message, history):
+    messages = [{"role": "system", "content": "You are a friendly kpop chatbot."}]
+    if history:
+        messages.extend(history)
+    messages.append({"role": "user", "content": message})
+    response = client.chat_completion(messages, max_tokens=100)
+    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()