Update app.py

app.py

@@ -19,7 +19,8 @@ def solve(problem_text):
         return "// Error: Problem text too short."
 
     # UPDATED PROMPT: Explicitly forbid comments
-    prompt = f"Problem:\n{problem_text}\n\nProvide ONLY the Python3 code. DO NOT include any comments, explanations, or docstrings.\nPython code solution:\n"
+    # We ask for a step-by-step algorithm to prevent one-liners, but still ban comments
+    prompt = f"Problem:\n{problem_text}\n\nWrite a complete, robust, step-by-step Python3 algorithm to solve this. DO NOT include any comments, docstrings, or conversational text. Only output the raw code.\nPython code solution:\n"
     inputs = tokenizer(prompt, return_tensors="pt")
     
     with torch.no_grad():