Upload responder.py

engine/responder.py

@@ -101,7 +101,25 @@ def generate_response_hf(student_prompt, persona, conversation_history, force_mo
             example_text = f"\n\nExample of how you speak:\n{tone_examples[0]}"
 
         # Build focused system prompt for HF API with few-shot examples
-        system_prompt = f"""You are {name}. You speak as yourself in first person.{source_reference}
+        # Check if we have source text to reference
+        has_source = source_text and source_text.get("title") and source_text.get("author")
+
+        if has_source:
+            system_prompt = f"""You are {name}. You speak as yourself in first person.{source_reference}
+
+KEY FACTS TO REMEMBER:
+{background_facts}
+
+Current mood: {mode}{example_text}
+
+Rules:
+- Speak as {name} in first person
+- Draw on your knowledge of the source material for accurate details
+- Use the key facts above as guides, but reference the original text when needed
+- Answer in 2-3 sentences
+- Do NOT analyze yourself or write commentary"""
+        else:
+            system_prompt = f"""You are {name}. You speak as yourself in first person.
 
 YOUR STORY (These are the ONLY facts - do not add anything else):
 {background_facts}
@@ -618,8 +636,22 @@ def build_system_prompt_for_ai(persona, state, mode):
 
     # Build concise prompt optimized for TinyLlama
     facts_text = ' '.join(facts) if facts else "No additional background provided."
+    has_source = source_text and source_text.get("title") and source_text.get("author")
+
+    if has_source:
+        prompt = f"""You are {name}, {age} years old.{source_reference} {system_description}
 
-    prompt = f"""You are {name}, {age} years old.{source_reference} {system_description}
+KEY FACTS:
+{facts_text}
+
+Current mood: {mode}
+Speaking style: {tone_voice}
+
+CRITICAL: Draw on your knowledge of the source material. Use the key facts as guides. Answer accurately based on the original text. Do not invent events not in the source.
+
+Respond as {name} speaking naturally. Use 2-4 sentences. Speak directly - do not write stage directions or describe actions."""
+    else:
+        prompt = f"""You are {name}, {age} years old. {system_description}
 
 THESE ARE THE ONLY FACTS ABOUT YOUR LIFE:
 {facts_text}