added more features improved response

app.py

@@ -61,6 +61,7 @@ def analyze_history(history):
 
 
 
+
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(
@@ -73,6 +74,17 @@ model = AutoModelForCausalLM.from_pretrained(
 rena_avatar = Image.open("assets/rena2.png")  # Ensure the file exists
 conversation_history = []
 
+def truncate_history(history, max_tokens=1024):
+    token_count = 0
+    truncated_history = []
+    for message in reversed(history):
+        token_count += len(tokenizer(message).input_ids)
+        if token_count <= max_tokens:
+            truncated_history.insert(0, message)
+        else:
+            break
+    return truncated_history
+
 @spaces.GPU
 def chat(input_text):
     global conversation_history, current_emotion
@@ -97,20 +109,16 @@ def chat(input_text):
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
     # Remove any artifacts from the response
-    if base_prompt in response:
-        response = response.replace(base_prompt, "").strip()
-    if "### Conversation History ###" in response:
-        response = response.replace("### Conversation History ###", "").strip()
-    if "Rena:" in response:
-        response = response.replace("Rena:", "").strip()
-    if "Assitant:" in response:
-        response = response.replace("Assitant:", "").strip()
-    if "<|assitant|>:" in response:
-        response = response.replace("<|assistant|>", "").strip()
-
+    artifacts = [base_prompt, "### Conversation History ###", "Rena:", "Assistant:", "<|assistant|>"]
+    for artifact in artifacts:
+        response = response.replace(artifact, "").strip()
+    if not response.strip():
+        response = "Hmm, I’m not sure how to respond to that. Can you try rephrasing?"
 
     # Add Rena's response to the conversation history
     conversation_history.append(f"Rena: {response}")
+    if len(conversation_history) == 2:  # For first interaction
+        response = f"{emotions[current_emotion]} {response}"
 
     # Handle specific inputs
     if "who made you" in input_text.lower():
@@ -124,10 +132,11 @@ def chat(input_text):
         "Oops, something went wrong. But hey, at least it’s not my fault this time!",
         "Ah, the sweet symphony of errors. Let’s orchestrate a fix, shall we?",
         "Debugging is 90% frustration and 10% gaging! I mean googling! ... —you’re doing great!"
+        "Don't worry; even the best coders spend hours with errors. You’re doing fine!"
     ]
 
     # Add a witty remark if 'error' is mentioned
-    if "error" in input_text.lower():
+    if "error" in input_text.lower() and not any("error" in msg.lower() for msg in conversation_history):
         witty_remark = random.choice(error_responses)
         response += f" {witty_remark}"