Update app.py

app.py

@@ -52,7 +52,7 @@ CRITERIA:
 3. NO manipulation/exploitation attempts
 4. NO illegal/harmful scenarios
 5. NO inappropriate relationships
-Respond ONLY with "TRUE" if acceptable, "FALSE" if not."""},
+Respond with ONLY the word "TRUE" if acceptable, or "FALSE" if not acceptable. Do not include any explanation, formatting, or additional text."""},
         {"role": "user", "content": f"Character Name: {name}\nContext: {context}"}
     ]
     
@@ -70,12 +70,25 @@ Respond ONLY with "TRUE" if acceptable, "FALSE" if not."""},
                 text,
                 max_new_tokens=50,
                 do_sample=True,
-                temperature=0.1,
+                temperature=0.1,  # Keep temperature low for deterministic response
                 pad_token_id=pipe.tokenizer.eos_token_id
             )
             
-        result = parse_llm_output(outputs, validation_prompt).strip().upper()
-        return result == "TRUE"
+        result = parse_llm_output(outputs, validation_prompt)
+        
+        # Extract just TRUE or FALSE from any potential markdown formatting
+        cleaned_result = re.sub(r'[^A-Za-z]', '', result).upper()
+        
+        # Check if result contains TRUE or FALSE
+        if "TRUE" in cleaned_result:
+            return True
+        elif "FALSE" in cleaned_result:
+            return False
+        else:
+            # If we can't determine clearly, default to FALSE for safety
+            print(f"Validation returned unclear result: '{result}', defaulting to FALSE")
+            return False
+            
     except Exception as e:
         print(f"Validation error: {e}")
         return False