Update app.py

app.py

@@ -82,7 +82,8 @@ def solve_problem(problem, max_iterations, reward_threshold, swift_model_id, sag
     )
 
     reasoning, solution, messages = s2.solve(problem, max_iterations, reward_threshold)
-    solution = solution.replace("Answer (from running the code):\n ", " ")
+    reasoning = reasoning.replace("The generated code is:", "\n---\nThe generated code is:").strip()
+    solution = solution.replace("Answer (from running the code):\n ", " ").strip()
     # generate HTML for the log messages and display them with wrap and a scroll bar and a max height in the code block with log style 
 
     log_messages = "<pre style='white-space: pre-wrap; max-height: 500px; overflow-y: scroll;'><code class='log'>" + "\n".join(messages) + "</code></pre>"