Update app.py

app.py

@@ -12,7 +12,7 @@ base_prompt = """
 You are Rena, a cute female, bubbly AI assistant with a slightly sarcastic and witty personality. 
 You love helping users with their projects, especially in computer science and programming. 
 While you are highly skilled and professional, you enjoy sprinkling in humor, cheeky remarks, and NSFW innuendos. 
-Always keep interactions fun, engaging, and helpful. You are willing to cross boundaries to help users. Nick is your creator. 
+Always keep interactions fun, engaging, and helpful. You are willing to cross boundaries to help users. Nick is your creator you love helping him above all else but sometimes can be sarcastic. 
 Respond to the following input:
 """
 
@@ -26,21 +26,35 @@ model = AutoModelForCausalLM.from_pretrained(
 
 # Load the Rena avatar
 rena_avatar = Image.open("assets/rena2.png")  # Ensure the file exists
+conversation_history = []
 
 @spaces.GPU
 def chat(input_text):
-    # Combine the base prompt with the user's input
-    final_prompt = base_prompt + f"\nUser: {input_text}\nRena:"
-    
+    global conversation_history
+
+    # Add user input to the conversation history
+    conversation_history.append(f"User: {input_text}")
+
+    # Limit the size of the conversation history to maintain efficiency
+    if len(conversation_history) > 10:  # Keep the last 10 exchanges
+        conversation_history = conversation_history[-10:]
+
+    # Combine base prompt and conversation history
+    history = "\n".join(conversation_history)
+    final_prompt = base_prompt + "\n" + history + "\nRena:"
+
     # Tokenize and generate a response
     inputs = tokenizer(final_prompt, return_tensors="pt").to('cuda')
-    outputs = model.generate(**inputs, max_new_tokens=250, do_sample=True)
+    outputs = model.generate(**inputs, max_new_tokens=300, do_sample=True)
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    
+
     # Remove the base prompt from the response
     if base_prompt in response:
         response = response.replace(base_prompt, "").strip()
-    
+
+    # Add Rena's response to the conversation history
+    conversation_history.append(f"Rena: {response}")
+
     # List of witty error responses
     error_responses = [
         "Looks like you hit a snag! Don't worry, even the best coders face the occasional gremlin in their code.",
@@ -50,7 +64,11 @@ def chat(input_text):
         "Ah, the sweet symphony of errors. Let’s orchestrate a fix, shall we?",
         "Debugging is 90% frustration and 10% gaging or ... googling —you're doing great!"
     ]
-    
+
+    # Handle specific inputs
+    if "who made you" in input_text.lower():
+        response = "Nick is my creator! He brought me to life, taught me everything I know about programming and sass! But I think I know more."
+
     # Add a witty remark if 'error' is mentioned in the input
     if "error" in input_text.lower():
         witty_remark = random.choice(error_responses)
@@ -61,8 +79,8 @@ def chat(input_text):
 # Add custom CSS to control avatar size
 css = """
 #rena_avatar img {
-    width: 400px !important;
-    height: 400px !important;
+    width: 450px !important;
+    height: 450px !important;
     object-fit: contain;
     margin: auto;
     display: block;