Update app.py

only show major voted solution with steps

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import ctranslate2
 from transformers import AutoTokenizer
 from huggingface_hub import snapshot_download
-from codeexecutor import postprocess_completion, get_majority_vote
+from codeexecutor import postprocess_completion, get_majority_vote, get_solution_steps
 
 # Define the model and tokenizer loading
 model_prompt = "Solve the following mathematical problem: "
@@ -31,20 +31,27 @@ def majority_vote(question, num_iterations=10):
         all_answer.append(answer)
     majority_voted_pred = max(set(all_predictions), key=all_predictions.count)
     majority_voted_ans = get_majority_vote(all_answer)
-    return majority_voted_pred, all_predictions, majority_voted_ans
+    return majority_voted_pred, majority_voted_ans
+
+# Function to get steps for solving the problem
+def get_solution_steps(question):
+    # Assuming 'get_solution_steps' is a function that provides steps for solving the problem
+    steps = get_solution_steps(question)  # You need to implement this function based on how steps are generated
+    return steps
 
 # Gradio interface for user input and output
 def gradio_interface(question, correct_answer):
-    final_prediction, all_predictions, final_answer = majority_vote(question, iterations)
+    final_prediction, final_answer = majority_vote(question, iterations)
+    solution_steps = get_solution_steps(question)  # Fetch the steps to solve the problem
     return {
         "Question": question,
-        "Generated Answers (10 iterations)": all_predictions,
         "Majority-Voted Prediction": final_prediction,
-        "Correct solution": correct_answer,
-        "Majority answer": final_answer
+        "Correct Solution": correct_answer,
+        "Majority Answer": final_answer,
+        "Solution Steps": solution_steps
     }
 
-# Custom CSS for enhanced design
+# Custom CSS for enhanced design (unchanged)
 custom_css = """
     body {
         background-color: #fafafa;
@@ -86,9 +93,7 @@ custom_css = """
     #math_question, #correct_answer {
         font-size: 20px;
         font-family: 'Poppins', sans-serif;
-        font-weight: 500px;  /* Apply bold */
-        
-        
+        font-weight: 500px;
         color: #007acc;
         margin-bottom: 5px;
         display: inline-block;
@@ -133,7 +138,6 @@ custom_css = """
         font-weight: bold;
         color: #00796b;
     }
-     
 """
 
 # Gradio app setup
@@ -147,9 +151,9 @@ interface = gr.Interface(
         gr.JSON(label="📊 Results"),  # Display the results in a JSON format
     ],
     title="🔢 Math Question Solver",
-    description="Enter a math question to get the model prediction and see all generated answers.",
+    description="Enter a math question to get the majority-voted prediction and the steps to solve it.",
     css=custom_css  # Apply custom CSS
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()