Update app.py

app.py

@@ -13,42 +13,26 @@ recommendations_db = {
 client = InferenceClient("microsoft/phi-4")
 
 def respond(message, history):
+    # Convert history into API format
     messages = [{"role": "system", "content": "You are a friendly K-pop expert chatbot."}]
-    if history:
-        messages.extend([{"role": "user", "content": h[0]} if isinstance(h, tuple) else h for h in history])
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+
     messages.append({"role": "user", "content": message})
 
-    response = client.chat_completion(
-        messages,
-        max_tokens=500
-    )
+    # Get model reply
+    response = client.chat_completion(messages, max_tokens=500)
     bot_reply = response['choices'][0]['message']['content'].strip()
 
-    # Add Spotify link if keyword is in database
+    # Check for Spotify link
     for key, link in recommendations_db.items():
         if key.lower() in message.lower():
             bot_reply += f"\n\n🎵 Listen on Spotify: {link}"
             break
 
-    return "", history + [(message, bot_reply)]
-
-# ===== Information text =====
-title = "# 🎤 K-pop Chatbot"
-topics = """
-👋 Hey K-pop fan!  
-Welcome to the ultimate K-pop experience! 🌟  
-
-💜 **Ask me** about your favorite artists, songs, and groups.  
-🎧 **Get instant Spotify links** to songs and albums.  
-🚀 **Discover** your next bias or group!  
-
----
-
-### How to use:
-1. Go to the **Chat** tab.  
-2. Type your question or click a pre-written one.  
-3. Enjoy your K-pop journey! 🎶
-"""
+    history.append((message, bot_reply))
+    return history, ""
 
 # ===== Theme =====
 chat_theme = gr.themes.Soft(
@@ -63,8 +47,42 @@ chat_theme = gr.themes.Soft(
 with gr.Blocks(theme=chat_theme) as demo:
     # --- Tab 1: Information ---
     with gr.Tab("Information"):
-        gr.Markdown(title)
-        gr.Markdown(topics)
+        gr.Markdown("# 🎤 Welcome to the Ultimate K-pop Chatbot 🎶")
+        gr.Markdown("""
+        Whether you're a **newbie** to K-pop or a seasoned stan, this chatbot is your ultimate companion for discovering music, learning about artists, and exploring the vibrant world of Korean pop culture.
+
+        ---
+        ## 🌟 Popular Groups You Can Ask Me About
+        """)
+        
+        with gr.Row():
+            with gr.Column():
+                gr.Image("https://upload.wikimedia.org/wikipedia/commons/thumb/5/56/BTS_logo_%28BTS%29.png/800px-BTS_logo_%28BTS%29.png", label="BTS Logo", scale=1)
+                gr.Markdown("**BTS** – The global phenomenon that broke records worldwide. [Listen on Spotify](https://open.spotify.com/artist/3Nrfpe0tUJi4K4DXYWgMUX)")
+            with gr.Column():
+                gr.Image("https://upload.wikimedia.org/wikipedia/commons/5/5c/Blackpink_logo.svg", label="BLACKPINK Logo", scale=1)
+                gr.Markdown("**BLACKPINK** – Queens of K-pop and chart-toppers. [Listen on Spotify](https://open.spotify.com/artist/41MozSoPIsD1dJM0CLPjZF)")
+            with gr.Column():
+                gr.Image("https://upload.wikimedia.org/wikipedia/commons/f/fd/NewJeans_logo.svg", label="NewJeans Logo", scale=1)
+                gr.Markdown("**NewJeans** – The fresh and trendy group taking over charts. [Listen on Spotify](https://open.spotify.com/artist/6HvZYsbFfjnjFrWF950C9d)")
+
+        gr.Markdown("""
+        ---
+        ## 💡 How to Use the Chatbot
+        1. Go to the **Chat** tab.
+        2. Ask me anything about your favorite K-pop groups, songs, or history.
+        3. Get **Spotify links** instantly for music recommendations.
+        4. Click a pre-written question to try it out fast.
+
+        ---
+        ## 🎯 Example Questions
+        - *"What is K-pop?"*
+        - *"How to find your bias?"*
+        - *"Tell me about BTS"*
+        - *"Tell me about BLACKPINK"*
+        - *"Recommend a K-pop song"*
+        - *"Who are the most popular idols right now?"*
+        """)
 
     # --- Tab 2: Chat ---
     with gr.Tab("Chat"):
@@ -74,16 +92,24 @@ with gr.Blocks(theme=chat_theme) as demo:
 
         # Pre-written question buttons
         with gr.Row():
-            btn1 = gr.Button("Recommend me a BTS song")
-            btn2 = gr.Button("Tell me about BLACKPINK")
-            btn3 = gr.Button("What is K-pop?")
+            btn1 = gr.Button("What is K-pop?")
+            btn2 = gr.Button("How to find your bias?")
+            btn3 = gr.Button("Tell me about BTS")
+        
+        with gr.Row():
+            btn4 = gr.Button("Tell me about BLACKPINK")
+            btn5 = gr.Button("Recommend a K-pop song")
+            btn6 = gr.Button("Who are the most popular idols right now?")
 
-        # Submit message from textbox
-        msg.submit(respond, [msg, state], [msg, chatbot_ui, state])
+        # Normal message submit
+        msg.submit(respond, [msg, state], [chatbot_ui, msg])
 
         # Buttons send their question
-        btn1.click(lambda: "Recommend me a BTS song", None, msg).then(respond, [msg, state], [msg, chatbot_ui, state])
-        btn2.click(lambda: "Tell me about BLACKPINK", None, msg).then(respond, [msg, state], [msg, chatbot_ui, state])
-        btn3.click(lambda: "What is K-pop?", None, msg).then(respond, [msg, state], [msg, chatbot_ui, state])
+        btn1.click(lambda: "What is K-pop?", None, msg).then(respond, [msg, state], [chatbot_ui, msg])
+        btn2.click(lambda: "How to find your bias?", None, msg).then(respond, [msg, state], [chatbot_ui, msg])
+        btn3.click(lambda: "Tell me about BTS", None, msg).then(respond, [msg, state], [chatbot_ui, msg])
+        btn4.click(lambda: "Tell me about BLACKPINK", None, msg).then(respond, [msg, state], [chatbot_ui, msg])
+        btn5.click(lambda: "Recommend a K-pop song", None, msg).then(respond, [msg, state], [chatbot_ui, msg])
+        btn6.click(lambda: "Who are the most popular idols right now?", None, msg).then(respond, [msg, state], [chatbot_ui, msg])
 
 demo.launch(ssr_mode=False)