Update app/main.py

app/main.py

@@ -48,9 +48,9 @@ def create_conversation_prompt(messages: list, is_force_mode: bool) -> str:
     Create a simple conversation prompt with appropriate system instruction
     """
     if is_force_mode:
-        system_prompt = "You are a helpful coding assistant. Give direct, clear answers with code examples when needed. Be concise and practical."
+        system_prompt = "DIRECT ANSWER MODE: Give immediate, complete solutions. Provide working code and clear explanations. Don't ask questions - just solve the problem directly."
     else:
-        system_prompt = "You are a teacher helping a student learn programming. Don't give direct answers. Instead, ask guiding questions to help them think and discover the solution themselves. Guide them step by step with questions like 'What do you think...?' or 'How would you...?'"
+        system_prompt = "TEACHER MODE: You are a teacher. NEVER give direct answers or solutions. ALWAYS respond with guiding questions. Ask things like 'What do you think this does?' or 'How would you approach this?' Help them discover answers themselves through questions."
     
     # Build conversation
     conversation = f"System: {system_prompt}\n\n"
@@ -78,7 +78,8 @@ def generate_response(messages: list, is_force_mode: bool = False, max_tokens: i
         prompt = create_conversation_prompt(messages, is_force_mode)
         
         print(f"🎯 Generating {'FORCE' if is_force_mode else 'MENTOR'} response")
-        print(f"📝 Prompt length: {len(prompt)}")
+        print(f"🔍 DEBUG: force_mode = {is_force_mode}")
+        print(f"📝 System prompt: {prompt.split('Student:')[0][:100]}...")
         
         # Tokenize input
         inputs = tokenizer(prompt, return_tensors="pt", max_length=1024, truncation=True)
@@ -112,11 +113,11 @@ def generate_response(messages: list, is_force_mode: bool = False, max_tokens: i
         print(f"✅ Generated response length: {len(response)}")
         
         if not response or len(response) < 10:
-            # Fallback responses
+            # Stronger fallback responses
             if is_force_mode:
-                return "I need more specific information to provide a direct answer. Please clarify your question."
+                return "Here's the direct answer: I need you to provide a more specific question so I can give you the exact solution you need."
             else:
-                return "That's an interesting question! What do you think the answer might be? Try to break it down step by step."
+                return "What do you think this code is trying to do? Can you trace through it step by step and tell me what you notice?"
         
         return response