Update app.py

app.py

@@ -1,42 +1,37 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Use a Spaces-friendly chat model
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+client = InferenceClient("microsoft/phi-4")
 
 def respond(message, history):
-    # Build K-pop specific system prompt
-    system_prompt = (
-        "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.\n\n"
+    # 🔹 Changed system prompt to force K-pop only responses
+    messages = [{
+        "role": "system",
+        "content": (
+            "You are a friendly chatbot who ONLY talks about K-pop. "
+        
+        )
+    }]
+    
+    if history:
+        messages.extend(history)
+        
+    messages.append({"role": "user", "content": message})
+    
+    response = client.chat_completion(
+        messages,
+        max_tokens=200
     )
-
-    # Turn chat history into a plain text conversation
-    conversation = ""
-    for user_msg, bot_msg in history:
-        conversation += f"User: {user_msg}\nBot: {bot_msg}\n"
-    conversation += f"User: {message}\nBot:"
-
-    # Combine instructions and conversation
-    prompt = system_prompt + conversation
-
-    # Generate response
-    response = client.text_generation(
-        prompt,
-        max_new_tokens=150,
-        do_sample=True,
-        temperature=0.7
-    )
-
-    return response.strip()
+    
+    return response['choices'][0]['message']['content'].strip()
 
 chatbot = gr.ChatInterface(
     respond,
+    type="messages",
     title="K-pop Chatbot",
     description="Ask me anything about K-pop! 🎤🎶"
 )
 
 chatbot.launch()
 
+