Update app.py

app.py

@@ -7,42 +7,48 @@ GROUP_DB = [
     {"group": "TWICE", "style": "pop"},
     {"group": "BLACKPINK", "style": "pop"},
     {"group": "BTS", "style": "pop"},
+    # Add more groups if you want!
 ]
 
-# Use text2text pipeline instead of deprecated conversational task
-chatbot = pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")
-
-pip install transformers torch
-
-def custom_kpop_response(user_input, history):
-    user_input_lower = user_input.lower()
+# 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)
+)
 
-    # Greeting detection
+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 recommendation
+    # 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, you might enjoy K-pop groups like "
-            "Red Velvet, TWICE, or BLACKPINK! They’re known for catchy melodies and energetic choreography."
+            "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."
         )
-        history.append((user_input, greeting + recommendation))
+        bot_reply = greeting + recommendation
+        history.append((message, bot_reply))
         return "", history
 
-    # Fallback to model
-    result = chatbot(user_input, max_length=100)
+    # Fallback to BlenderBot for other questions
+    result = chatbot(message, max_length=100)
     bot_reply = greeting + result[0]['generated_text']
-
-    history.append((user_input, bot_reply))
+    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."
+    description="Say hi! Ask for K-pop groups, or songs similar to your Western favorites."
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()
+