Update app.py

app.py

@@ -1,57 +1,39 @@
 import gradio as gr
-
-# --- Chatbot logic ---
-def get_song_recommendation(user_input, history):
-    genre = user_input.lower()
-    songs = {
-        "pop": "Try listening to 'As It Was' by Harry Styles.",
-        "rock": "Try 'Smells Like Teen Spirit' by Nirvana.",
-        "jazz": "Try 'So What' by Miles Davis.",
-        "hip hop": "Try 'HUMBLE.' by Kendrick Lamar.",
-        "classical": "Try 'Clair de Lune' by Debussy."
-    }
-
-    response = "I didn't catch the genre. Please try pop, rock, jazz, hip hop, or classical."
-    for key in songs:
-        if key in genre:
-            response = songs[key]
-            break
-
-    history.append((user_input, response))
-    return "", history
-
-# --- Pages ---
-
-def page_home():
-    return gr.Markdown("## 🏠 Home\nWelcome to the Music Finder website!")
-
-def page_about():
-    return gr.Markdown("## ℹ️ About\nThis app helps you find songs by chatting based on your favorite genre.")
-
-def page_chatbot():
-    with gr.Blocks() as demo:
-        gr.Markdown("## 💬 Song Finder Chatbot\nType your favorite genre and get a song recommendation!")
-
-        chatbot = gr.Chatbot()
-        msg = gr.Textbox(placeholder="Enter a genre like pop, rock, jazz...")
-        clear_btn = gr.Button("Clear Chat")
-
-        msg.submit(get_song_recommendation, [msg, chatbot], [msg, chatbot])
-        clear_btn.click(lambda: [], None, chatbot, queue=False)
-
-    return demo
-
-def page_top_genres():
-    return gr.Markdown("## 🎶 Top Genres\n- Pop\n- Rock\n- Jazz\n- Hip Hop\n- Classical")
-
-def page_contact():
-    return gr.Markdown("## 📬 Contact\nMade with ❤️. Reach me at: [your-email@example.com]")
-
-# --- Main App with Tabs ---
-demo = gr.TabbedInterface(
-    interface_list=[page_home, page_about, page_chatbot, page_top_genres, page_contact],
-    tab_names=["Home", "About", "Chatbot", "Top Genres", "Contact"]
+from huggingface_hub import InferenceClient
+
+client = InferenceClient("microsoft/phi-4")
+
+def respond(message, history):
+    # System prompt specialized for K-pop
+    messages = [
+        {
+            "role": "system",
+            "content": (
+                "You are a friendly chatbot who ONLY talks about K-pop. "
+                "You are an expert in K-pop groups, idols, music, albums, choreography, fandom culture, "
+                "and industry news. If a user asks about anything unrelated to K-pop, politely redirect "
+                "the conversation back to K-pop."
+            )
+        }
+    ]
+    
+    if history:
+        messages.extend(history)
+        
+    messages.append({"role": "user", "content": message})
+    
+    response = client.chat_completion(
+        messages,
+        max_tokens=150
+    )
+    
+    return response['choices'][0]['message']['content'].strip()
+
+chatbot = gr.ChatInterface(
+    respond, 
+    type="messages", 
+    title="K-pop Chatbot",
+    description="Ask me anything about K-pop! 🎤🎶"
 )
 
-demo.launch()
-
+chatbot.launch()