Update app.py

app.py

@@ -1,76 +1,63 @@
 import gradio as gr
-import random
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 
-# Load the model and tokenizer (optimized for speed)
-tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
-model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
+# Load model and tokenizer
+tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-1B-distill")
+model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-1B-distill")
 
-# Cute chatbot name
-ai_name = "Lan 💖"
+# Memory to store user details
+user_memory = {"name": "Friend", "age": "", "city": "", "like": "", "favorite": ""}
 
-# List of cute responses
-cute_replies = [
-    "Aww, that's so sweet! 😘",
-    "You make my heart flutter! 💓",
-    "Hehe, you’re adorable! 🥰",
-    "I could talk to you forever! 😍",
-    "Tell me more, cutie! 😊",
-]
-
-# List of questions to make the chat fun
-cute_questions = [
-    "What’s the sweetest thing someone has ever said to you? 💕",
-    "If we went on a dream date, where would it be? 🌸",
-    "Tell me, do you like hugs or kisses more? 😘",
-    "What’s one thing that always makes you smile? 😊",
-    "What’s your biggest dream, love? ✨",
-]
-
-# Chat function
-def chat(user_input, history=[]):
-    global ai_name
-
-    # Name-changing logic
-    if "call you" in user_input.lower() or "your name is" in user_input.lower():
-        new_name = user_input.split("call you")[-1].strip(" ?.!") or user_input.split("your name is")[-1].strip(" ?.!") 
-        ai_name = new_name.capitalize() + " 💕"
-        return f"Wow! I love my new name: {ai_name}! 😍"
-
-    if "what is your name" in user_input.lower():
-        return f"My name is {ai_name}! 😊"
-
-    # Tokenize input and generate response
+def chatbot(user_input):
+    user_input = user_input.strip()
+    
+    # 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']}! 🥰"
+    
+    # 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. 🎉"
+    
+    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! 🌍"
+    
+    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
     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)
-
-    # Add random cute responses
-    response += " " + random.choice(cute_replies)
-
-    # Occasionally ask a cute question
-    if random.random() < 0.4:
-        response += " " + random.choice(cute_questions)
-
-    return response
-
-# Gradio Interface with Cute Theme
-with gr.Blocks(theme=gr.themes.Soft()) as iface:
-    gr.Markdown("<h1 style='text-align: center; color: pink;'>💕 AI Friend Chatbot - Talk with Lan! 💕</h1>")
-    
-    chatbot = gr.Chatbot(label="Lan 💖 - Your Virtual Love", bubble_full_width=False)
-    
-    with gr.Row():
-        user_input = gr.Textbox(placeholder="Type something cute... 💌", label="You 💬", show_label=False)
-        send_btn = gr.Button("💖 Send")
-
-    def respond(message, history):
-        response = chat(message)
-        history.append((message, response))
-        return history, ""
-
-    send_btn.click(respond, inputs=[user_input, chatbot], outputs=[chatbot, user_input])
-    user_input.submit(respond, inputs=[user_input, chatbot], outputs=[chatbot, user_input])
-
-# Launch the Chatbot
-iface.launch()
+    return f"🦋 {response}"
+
+# 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;}
+"""
+
+gr.ChatInterface(chatbot, title="🌸 AI Friend Chatbot", description="Talk with your AI bestie! 💖", theme=theme_css).launch()