Update app.py

app.py

@@ -1,30 +1,94 @@
+import os
+import requests
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Connect to the model
+# -----------------------
+# SPOTIFY AUTH
+# -----------------------
+def get_spotify_token():
+    client_id = os.getenv("SPOTIFY_CLIENT_ID")
+    client_secret = os.getenv("SPOTIFY_CLIENT_SECRET")
+    auth_url = "https://accounts.spotify.com/api/token"
+    auth_response = requests.post(
+        auth_url,
+        data={"grant_type": "client_credentials"},
+        auth=(client_id, client_secret),
+    )
+    if auth_response.status_code == 200:
+        return auth_response.json()["access_token"]
+    else:
+        return None
+
+def search_spotify(query, limit=3):
+    token = get_spotify_token()
+    if not token:
+        return []
+    url = f"https://api.spotify.com/v1/search?q={query}&type=track&limit={limit}"
+    headers = {"Authorization": f"Bearer {token}"}
+    r = requests.get(url, headers=headers)
+    if r.status_code == 200:
+        items = r.json()["tracks"]["items"]
+        links = [f"{item['name']} by {item['artists'][0]['name']}: {item['external_urls']['spotify']}" for item in items]
+        return links
+    else:
+        return []
+
+# -----------------------
+# CHATBOT LOGIC
+# -----------------------
 client = InferenceClient("microsoft/phi-4")
 
-# Chat function
 def respond(message, history):
-    messages = [
-        {"role": "system", "content": "You are a friendly, hyper-energetic K-pop expert chatbot. Always give clear answers and, when possible, recommend K-pop songs with clickable Spotify links."}
-    ]
-    
-    # Keep chat history
+    # Only add short responses unless user asks for more
+    style_prompt = "Answer briefly unless the user asks for more details. If recommending songs, provide working Spotify links."
+
+    messages = [{"role": "system", "content": f"You are a friendly Kpop expert chatbot. {style_prompt}"}]
+
     if history:
-        for h in history:
-            if isinstance(h, tuple):
-                messages.append({"role": "user", "content": h[0]})
-                messages.append({"role": "assistant", "content": h[1]})
-            else:
-                messages.append(h)
-    
+        messages.extend([{"role": "user", h[0]} if isinstance(h, tuple) else h for h in history])
+
     messages.append({"role": "user", "content": message})
-    
+
+    # Song recommendation check
+    if "recommend" in message.lower() and "song" in message.lower():
+        songs = search_spotify("kpop")
+        if songs:
+            return "Here are some Kpop songs I recommend:\n" + "\n".join(songs)
+
     response = client.chat_completion(messages, max_tokens=500)
     return response['choices'][0]['message']['content'].strip()
 
-# Soft theme
+# -----------------------
+# INFO CONTENT
+# -----------------------
+intro_text = """
+# 🎤 Welcome to the Ultimate Kpop Chatbot! 💜
+
+Whether you’re brand new or a long-time stan, I’m your friendly Kpop buddy ready to help you explore this amazing music genre.
+
+## 🌟 Top 10 Popular Kpop Groups (2025)
+1. BTS  
+2. BLACKPINK  
+3. Stray Kids  
+4. SEVENTEEN  
+5. NewJeans  
+6. ATEEZ  
+7. TWICE  
+8. TXT (Tomorrow X Together)  
+9. IVE  
+10. ENHYPEN  
+
+## 💡 How to use this chatbot:
+- Click a preset question above the chat box for instant answers.  
+- Ask me anything about Kpop history, groups, songs, or trends.  
+- Request song recommendations and I’ll give you working Spotify links.  
+- Keep chatting to learn more — and if you want details, just say *Tell me more!*  
+"""
+
+# -----------------------
+# THEME
+# -----------------------
 chat_theme = gr.themes.Soft(
     primary_hue="pink",
     secondary_hue="rose",
@@ -38,54 +102,30 @@ chat_theme = gr.themes.Soft(
     button_primary_background_fill_hover="*primary_400"
 )
 
-# Interface
+# -----------------------
+# UI
+# -----------------------
+preset_questions = [
+    "What is Kpop?",
+    "How to find your bias?",
+    "Who are the most popular Kpop groups?",
+    "How can I start listening to Kpop?",
+    "What are Kpop fan chants?",
+    "Can you recommend me some Kpop songs?",
+    "What is a Kpop comeback?"
+]
+
 with gr.Blocks(theme=chat_theme) as demo:
-    with gr.Tab("Information"):
-        gr.Markdown("# 🎤 Welcome to the Ultimate K-pop Chatbot 🎶💜")
-        gr.Markdown("""
-        Annyeong, K-pop fan! 🌟 Whether you’ve just discovered your first idol or you’ve been stanning since the 2nd gen days, this chatbot is your ultimate companion.  
-        
-        I can spill the tea ☕, give you trivia 🧠, recommend music 🎧, and even link you directly to Spotify so you can stream instantly!  
-        Together, we’ll explore the dazzling world of Korean pop culture — from catchy hooks to unforgettable choreographies. 🕺💃
-        
-        ---
-        ## 🌟 Top 10 Popular K-pop Groups
-        1. **BTS** – [🎧 Listen on Spotify](https://open.spotify.com/artist/3Nrfpe0tUJi4K4DXYWgMUX)  
-        2. **BLACKPINK** – [🎧 Listen on Spotify](https://open.spotify.com/artist/41MozSoPIsD1dJM0CLPjZF)  
-        3. **TWICE** – [🎧 Listen on Spotify](https://open.spotify.com/artist/7n2Ycct7Beij7Dj7meI4X0)  
-        4. **Stray Kids** – [🎧 Listen on Spotify](https://open.spotify.com/artist/2dIgFjalVxs4ThymZ67YCE)  
-        5. **SEVENTEEN** – [🎧 Listen on Spotify](https://open.spotify.com/artist/7nqOGRxlXj7N2JYbgNEjYH)  
-        6. **NewJeans** – [🎧 Listen on Spotify](https://open.spotify.com/artist/6HvZYsbFfjnjFrWF950C9d)  
-        7. **ITZY** – [🎧 Listen on Spotify](https://open.spotify.com/artist/2KC9Qb60EaY0kW4eH68vr3)  
-        8. **ENHYPEN** – [🎧 Listen on Spotify](https://open.spotify.com/artist/5t5FqBwTcgKTaWmfEbwQY9)  
-        9. **(G)I-DLE** – [🎧 Listen on Spotify](https://open.spotify.com/artist/2AfmfGFbe0A0WsTYm0SDTx)  
-        10. **IVE** – [🎧 Listen on Spotify](https://open.spotify.com/artist/6RHTUrRF63xao58xh9FXYJ)  
-
-        ---
-        ## 💡 How to Use the Chatbot
-        1. Switch to the **Chat** tab.
-        2. Ask any question about K-pop — songs, groups, history, concepts, comebacks.
-        3. Use **pre-written questions** to start fast.
-        4. If you ask for song recs, I’ll give you a **Spotify link** so you can click & play instantly.
-        5. Try broad prompts like:
-           - "Recommend me upbeat K-pop songs"
-           - "Tell me about rookie groups in 2025"
-           - "What's the difference between 2nd gen and 4th gen?"
-        6. Don’t be shy — I’m here to help you stan smarter and harder 💜
-        """)
-
-    with gr.Tab("Chat"):
-        gr.ChatInterface(
-            fn=respond,
-            type="messages",
-            examples=[
-                "What is K-pop?",
-                "How to find your bias?",
-                "Tell me about BTS",
-                "Recommend a K-pop song",
-                "Who are the most popular idols right now?",
-                "What's the history of K-pop?"
-            ]
-        )
-
-demo.launch()
+    with gr.Tab("ℹ️ Information"):
+        gr.Markdown(intro_text)
+
+    with gr.Tab("💬 Chat"):
+        with gr.Column():
+            gr.Markdown("### ❓ Click a question below to ask instantly:")
+            with gr.Row():
+                for q in preset_questions:
+                    gr.Button(q).click(fn=respond, inputs=gr.Textbox(value=q, visible=False), outputs=gr.Textbox())
+
+            chatbot = gr.ChatInterface(fn=respond, type="messages")
+
+demo.launch()