added all features

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 import ctranslate2
 from transformers import AutoTokenizer
 from huggingface_hub import snapshot_download
-from codeexecutor import get_majority_vote
+from codeexecutor import get_majority_vote,type_check
+import codeexecutor
+
 import re
 import os
 # Define the model and tokenizer loading
@@ -41,7 +43,7 @@ def parse_prediction(prediction):
     if answer is None:
         # If no "Answer:" found, assume last line is the answer
         answer = lines[-1].strip()
-        steps = lines[:-1]
+        steps = lines
     steps_text = '\n'.join(steps).strip()
     return answer, steps_text
 
@@ -53,10 +55,16 @@ def majority_vote_with_steps(question, num_iterations=10):
 
     for _ in range(num_iterations):
         prediction = get_prediction(question)
-        answer, steps = parse_prediction(prediction)
-        all_predictions.append(prediction)
-        all_answers.append(answer)
-        steps_list.append(steps)
+        answer,sucess= postprocess_completion(prediction, return_status=True, last_code_block=True)
+        if sucess:
+            all_predictions.append(prediction)
+            all_answers.append(answer)
+            steps_list.append(steps)
+        else:
+            answer, steps = parse_prediction(prediction)
+            all_predictions.append(prediction)
+            all_answers.append(answer)
+            steps_list.append(steps)
 
     # Get the majority voted answer
     majority_voted_ans = get_majority_vote(all_answers)
@@ -65,13 +73,14 @@ def majority_vote_with_steps(question, num_iterations=10):
     for i, ans in enumerate(all_answers):
         if ans == majority_voted_ans:
             steps_solution = steps_list[i]
+            answer=parse_prediction(steps_solution)
             break
     else:
+        answer=majority_voted_ans
         steps_solution = "No steps found"
 
-    return majority_voted_ans, steps_solution
+    return answer, steps_solution
 
-# Gradio interface for user input and output
 def gradio_interface(question, correct_answer):
     final_answer, steps_solution = majority_vote_with_steps(question, iterations)
     return question, final_answer, steps_solution, correct_answer
@@ -166,7 +175,7 @@ custom_css = """
 """
 
 # Define the directory path
-flagging_dir = "./flagged_data"
+flagging_dir = "./flagged_data" 
 
 # Create the directory if it doesn't exist
 if not os.path.exists(flagging_dir):
@@ -179,9 +188,9 @@ interface = gr.Interface(
         gr.Textbox(label="🧠 Math Question", placeholder="Enter your math question here...", elem_id="math_question"),
     ],
     outputs=[
-        gr.Textbox(label="Majority-Voted Answer", interactive=False),  # Non-editable
-        gr.Textbox(label="Steps to Solve", interactive=False),  # Non-editable
-        gr.Textbox(label="✅ Correct Solution", interactive=True),  # Editable textbox for correct solution
+        gr.Textbox(label="Question", interactive=False),  # Non-editable
+        gr.Textbox(label="Answer", interactive=False),  # Non-editable
+        gr.Textbox(label="Solution", interactive=True),  # Editable textbox for correct solution
     ],
     title="🔢 Math Question Solver",
     description="Enter a math question to get the model's majority-voted answer and steps to solve the problem.",

codeexecutor.py

@@ -97,7 +97,7 @@ def execute_completion(executor, completion, return_status, last_code_block):
     success = successes[-1]
     if return_status:
         return output, success
-    return output
+    return output ,False
 
 
 def postprocess_completion(text, return_status, last_code_block):
@@ -113,3 +113,24 @@ def get_majority_vote(answers):
     c = Counter(answers)
     value, _ = c.most_common()[0]
     return value
+
+
+def type_check(self,expr_str):
+       
+       
+        expr = sp.sympify(expr_str)
+        
+        # Check if the expression is a real number
+        if expr.is_real:
+            return "Real"
+
+        # Check if the expression is a complex number
+        if expr.is_complex:
+            return "Complex"
+
+        # Check if the expression is a polynomial
+        if expr.is_polynomial():
+            return "Polynomial"
+
+        # Otherwise, classify as other
+        return "Other"

temp.py

@@ -3,6 +3,7 @@ import ctranslate2
 from transformers import AutoTokenizer
 from huggingface_hub import snapshot_download
 from codeexecutor import get_majority_vote
+import codeexecutor
 import re
 
 # Define the model and tokenizer loading
@@ -11,6 +12,9 @@ tokenizer = AutoTokenizer.from_pretrained("AI-MO/NuminaMath-7B-TIR")
 model_path = snapshot_download(repo_id="Makima57/deepseek-math-Numina")
 generator = ctranslate2.Generator(model_path, device="cpu", compute_type="int8")
 iterations = 10
+executor = PythonREPL()
+
+
 
 # Function to generate predictions using the model
 def get_prediction(question):