Update app.py

app.py

@@ -18,8 +18,7 @@ def solve(problem_text):
     if not problem_text or len(problem_text) < 10:
         return "// Error: Problem text too short."
 
-    # THE FIX: Go back to the exact prompt that we know works!
-    prompt = f"Problem:\n{problem_text}\n\nPython code solution:\n"
+    prompt = f"Problem:\n{problem_text}\n\nThink step-by-step about the edge cases, time complexity, and logic required to solve this problem. After your explanation, provide the optimal Python3 code inside a ```python code block.\n"
     inputs = tokenizer(prompt, return_tensors="pt")
     
     with torch.no_grad():