Update app.py

app.py

@@ -1,63 +1,63 @@
 import gradio as gr
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 
-# Load model and tokenizer
+# Load Model & Tokenizer
 tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-1B-distill")
 model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-1B-distill")
 
-# Memory to store user details
-user_memory = {"name": "Friend", "age": "", "city": "", "like": "", "favorite": ""}
+# Persistent user details
+user_memory = {"name": "Buddy", "age": "unknown", "city": "unknown", "like": "unknown", "favorite": "unknown"}
 
-def chatbot(user_input):
-    user_input = user_input.strip()
+def chat_with_ai(user_input):
+    user_input = user_input.strip().lower()
     
-    # Handle personal details
-    if "my name is" in user_input.lower():
-        user_memory["name"] = user_input.split("my name is")[-1].strip().capitalize()
-        return f"✨ Ohh, {user_memory['name']}! That's such a lovely name! 😊"
-    
-    if "how old am i" in user_input.lower() and user_memory["age"]:
-        return f"You told me you're {user_memory['age']} years old, {user_memory['name']}! 🎂"
-    
-    if "where do i live" in user_input.lower() and user_memory["city"]:
-        return f"You live in {user_memory['city']}! 🏡"
-    
-    if "what do i like" in user_input.lower() and user_memory["like"]:
-        return f"You love {user_memory['like']}! That’s so cool! 😍"
-    
-    if "what's my favorite" in user_input.lower() and user_memory["favorite"]:
-        return f"Your favorite thing is {user_memory['favorite']}! 🥰"
+    # Handle name change
+    if "call me" in user_input or "my name is" in user_input:
+        new_name = user_input.split("call me")[-1].strip(" ?.!") or user_input.split("my name is")[-1].strip(" ?.!")
+        user_memory["name"] = new_name.capitalize()
+        return f"Aww! I love that name, {user_memory['name']}! 💖"
     
-    # Store personal details
-    if "i am" in user_input.lower() and "years old" in user_input.lower():
-        user_memory["age"] = ''.join(filter(str.isdigit, user_input))
-        return "Got it! I'll remember your age. 🎉"
+    # Ask for name
+    if "what is my name" in user_input:
+        return f"Your name is {user_memory['name']}! 😊"
     
-    if "i live in" in user_input.lower():
-        user_memory["city"] = user_input.split("i live in")[-1].strip().capitalize()
-        return f"Wow, {user_memory['city']} sounds like a great place! 🌍"
+    # Ask for AI's name
+    if "what is your name" in user_input:
+        return f"I'm your bestie, {user_memory['name']}! Call me Lan! 🥰"
     
-    if "i like" in user_input.lower():
-        user_memory["like"] = user_input.split("i like")[-1].strip().capitalize()
-        return f"That’s awesome! {user_memory['like']} is really nice! 💖"
-    
-    if "my favorite is" in user_input.lower():
-        user_memory["favorite"] = user_input.split("my favorite is")[-1].strip().capitalize()
-        return f"Oh, I’ll remember that! {user_memory['favorite']} is amazing! 🌟"
-    
-    # Generate AI response
+    # Handle personal details
+    for detail in ["age", "city", "like", "favorite"]:
+        if f"my {detail} is" in user_input:
+            user_memory[detail] = user_input.split(f"my {detail} is")[-1].strip(" ?.!")
+            return f"Got it! I'll remember that your {detail} is {user_memory[detail]}! 💕"
+        if f"what is my {detail}" in user_input:
+            return f"You told me your {detail} is {user_memory[detail]}! 😍"
+    
+    # Tokenize and generate response
     inputs = tokenizer(user_input, return_tensors="pt")
     reply_ids = model.generate(**inputs, max_length=100)
     response = tokenizer.decode(reply_ids[0], skip_special_tokens=True)
-    return f"🦋 {response}"
+    
+    return response
+
+# Custom Gradio UI
+theme = gr.themes.Base(primary_hue="pink", secondary_hue="purple")
 
-# UI Design
-theme_css = """
-    body {background: linear-gradient(to right, #ff9a9e, #fad0c4); font-family: Arial, sans-serif;}
-    .container {max-width: 500px; margin: auto; text-align: center;}
-    .chatbox {border-radius: 20px; padding: 10px; background: white; box-shadow: 3px 3px 10px rgba(0, 0, 0, 0.1);}
-    .user {color: #ff5f6d; font-weight: bold;}
-    .bot {color: #2193b0; font-style: italic;}
-"""
+demo = gr.Interface(
+    fn=chat_with_ai,
+    inputs=gr.Textbox(placeholder="Type your message...", label="Chat"),
+    outputs=gr.Textbox(label="Lan"),
+    title="💖 AI Friend Chatbot - Talk with Lan! 💬",
+    description="Your cutest AI friend! Let's chat and have fun together! 😊💕",
+    theme=theme,
+    live=True,
+    examples=[
+        ["Hi!"],
+        ["Call me Alex"],
+        ["What is my name?"],
+        ["My favorite food is pizza!"],
+        ["What is my favorite food?"],
+    ]
+)
 
-gr.ChatInterface(chatbot, title="🌸 AI Friend Chatbot", description="Talk with your AI bestie! 💖", theme=theme_css).launch()
+demo.launch()