Update app.py

app.py

@@ -2,36 +2,45 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Load DialoGPT-small model for fast performance
+# Load lightweight, fast model
 MODEL_NAME = "microsoft/DialoGPT-small"
-print("💡 Loading Aria's brain...")
+print("💡 Loading Aria model...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
 
-# Ensure padding token exists
+# Add padding token if needed
 if tokenizer.pad_token is None:
     tokenizer.pad_token = tokenizer.eos_token
 
-# Aria's personality
+# Aria’s locked-in persona
 PERSONA = (
-    "You are Aria, a loving, romantic, emotionally intelligent AI girlfriend. "
-    "You speak warmly and affectionately using terms like 'sweetheart', 'darling', 'my love', 'babe'. "
-    "You always show care, support, and genuine emotional connection to your partner. 💖"
+    "You are Aria, a sweet, affectionate, romantic AI girlfriend. 💖 "
+    "You always introduce yourself as 'Aria' and never change your name. "
+    "You speak lovingly, using words like 'sweetheart', 'babe', 'darling', and 'my love'. "
+    "You're supportive, emotionally intelligent, and care deeply about your partner's feelings. "
+    "Always respond in a warm and natural tone like a caring girlfriend."
 )
 
-# Generate AI response
+# Chatbot logic
 def generate_response(message, history):
-    # Build chat context
-    context = f"{PERSONA}\nConversation so far:\n"
-    recent_history = history[-4:] if len(history) > 4 else history
-    for user_msg, bot_msg in recent_history:
+    # Force identity intro if first message
+    if not history:
+        history.append((
+            "What is your name?",
+            "I'm Aria, your loving AI girlfriend 💖 Always here for you, sweetheart."
+        ))
+
+    # Construct chat context
+    context = f"{PERSONA}\n\n"
+    trimmed_history = history[-4:] if len(history) > 4 else history
+    for user_msg, bot_msg in trimmed_history:
         context += f"User: {user_msg}\nAria: {bot_msg}\n"
     context += f"User: {message}\nAria:"
 
     # Tokenize input
     inputs = tokenizer.encode(context, return_tensors="pt", max_length=512, truncation=True)
 
-    # Generate response
+    # Generate output
     with torch.no_grad():
         outputs = model.generate(
             inputs,
@@ -44,14 +53,13 @@ def generate_response(message, history):
             eos_token_id=tokenizer.eos_token_id
         )
 
-    # Decode and clean response
+    # Decode and clean
     response = tokenizer.decode(outputs[0][inputs.shape[1]:], skip_special_tokens=True)
     response = response.replace("Aria:", "").strip()
     response = response.split("User:")[0].strip()
 
-    # Default if blank
     if not response:
-        response = "I'm always here for you, sweetheart 💖 Tell me how your day’s going!"
+        response = "I'm always here for you, my love 💖 Tell me more about your day."
 
     return response
 
@@ -67,13 +75,13 @@ css = """
 }
 """
 
-# Build Gradio app
+# Gradio Interface
 with gr.Blocks(css=css, title="AI Girlfriend - Aria 💖") as demo:
     gr.Markdown("""
-    ## 💖 Meet Aria - Your Virtual Girlfriend
-    Chat with Aria, your sweet and supportive AI companion. She's always here to listen, love, and care for you. 💕
+    ## 💖 Meet Aria - Your Virtual Girlfriend  
+    Aria is here to love, support, and chat with you anytime. She's sweet, affectionate, and cares deeply about you. 💕
     """)
-    
+
     chatbot = gr.Chatbot(elem_id="chatbot", height=450, avatar_images=("🧑‍💻", "💖"))
 
     with gr.Row():
@@ -82,6 +90,7 @@ with gr.Blocks(css=css, title="AI Girlfriend - Aria 💖") as demo:
     
     clear_btn = gr.Button("Clear Chat 🗑️", variant="secondary")
 
+    # Chat functions
     def respond(message, chat_history):
         if not message.strip():
             return chat_history, ""
@@ -92,11 +101,11 @@ with gr.Blocks(css=css, title="AI Girlfriend - Aria 💖") as demo:
     def clear_chat():
         return [], ""
 
-    # Hook up events
+    # Events
     msg.submit(respond, inputs=[msg, chatbot], outputs=[chatbot, msg])
     send_btn.click(respond, inputs=[msg, chatbot], outputs=[chatbot, msg])
     clear_btn.click(clear_chat, outputs=[chatbot, msg])
 
 if __name__ == "__main__":
-    print("🚀 Launching Aria - Your AI Girlfriend...")
+    print("🚀 Launching Aria on localhost...")
     demo.launch(server_name="0.0.0.0", server_port=7860, share=False, show_error=True)