mathprompter improvements

app.py

@@ -27,15 +27,15 @@ def run(token, question, method):
                 gr.Code.update(value=generated_code, interactive=True), gr.Button.update(visible=True))
 
     elif method == "MathPrompter":
-        exp_op, code_op, generated_code = solve_mp(question, token)
-
+        exp_op, code_op, generated_code, generated_exp = solve_mp(question, token)
+        display_value = generated_exp + "\n\n" + generated_code
         if code_op is not None:
             ans = f"Answer from Expression execution: {exp_op} \nAnswer from Code execution: {code_op}  "
-            return ans, gr.Code.update(value=generated_code, interactive=True), gr.Button.update(visible=True)
+            return ans, gr.Code.update(value=display_value, interactive=True), gr.Button.update(visible=True)
         else:
             return (
                 "Code execution failed, please review it from below and re-run it or try-asking again with more details",
-                gr.Code.update(value=generated_code, interactive=True), gr.Button.update(visible=True))
+                gr.Code.update(value=display_value, interactive=True), gr.Button.update(visible=True))
 
     else:
         raise gr.Error("Please select the evaluating strategy from dropdown")

mathprompter.py

@@ -36,7 +36,6 @@ def generate_python_code(question, equation, param, token):
 
 def run(question, random_candidates, hps, token):
     question, var_map = generate_algebric_template(question)
-    print(question)
 
     # generating the random candidates for arguments
     random_mapping = pd.DataFrame(columns=list(var_map.keys()))
@@ -105,7 +104,6 @@ def solve_mp(question, token):
     (expression, code), var_map = run(question, 5, hps, token)
     exp_op = None
     code_op = None
-
     try:
         # expression output
         for k, v in var_map.items():
@@ -115,13 +113,11 @@ def solve_mp(question, token):
 
         if "input(" in expression:
             raise Exception
-
         exp_op = run_code(expression)
     except:
         print("expression cannot be executed", expression)
     try:
         # code output
-        print(code)
         parameters = code.split("\n")[0].split("def solution")[-1][1:-2].split(",")
         if '' in parameters:
             parameters.remove('')
@@ -137,6 +133,6 @@ def solve_mp(question, token):
             raise Exception
         code_op = run_code(code)
     except:
-        return None, None, code
+        return None, None, code, expression
 
-    return exp_op, code_op, code
+    return exp_op, code_op, code, expression