Update app.py

app.py

@@ -71,14 +71,25 @@ def majority_vote_with_steps(question, num_iterations=10):
 
     return majority_voted_ans, steps_solution
 
+# Correct solution for the problem (programmatically or hardcoded)
+def fetch_correct_solution(question):
+    # You can define or fetch the correct solution here
+    # For example, for demonstration, I'll hardcode it
+    correct_solution_map = {
+        "What is 2 + 2?": "4",
+        "Solve for x in 2x + 3 = 7.": "x = 2",
+    }
+    return correct_solution_map.get(question, "Correct solution not available.")
+
 # Gradio interface for user input and output
-def gradio_interface(question, correct_answer):
+def gradio_interface(question):
     final_answer, steps_solution = majority_vote_with_steps(question, iterations)
+    correct_solution = fetch_correct_solution(question)
     return {
         "Question": question,
         "Majority-Voted Answer": final_answer,
         "Steps to Solve": steps_solution,
-        "Correct Solution": correct_answer
+        "Correct Solution": correct_solution  # Include the correct solution in the output
     }
 
 # Custom CSS for enhanced design (unchanged)
@@ -120,7 +131,7 @@ custom_css = """
         background-color: #f1f8ff;
         margin-bottom: 15px;
     }
-    #math_question, #correct_answer {
+    #math_question {
         font-size: 20px;
         font-family: 'Poppins', sans-serif;
         font-weight: 500px;
@@ -175,14 +186,12 @@ interface = gr.Interface(
     fn=gradio_interface,
     inputs=[
         gr.Textbox(label="🧠 Math Question", placeholder="Enter your math question here...", elem_id="math_question"),
-        
     ],
     outputs=[
         gr.JSON(label="📊 Results"),  # Display the results in a JSON format
-       
     ],
     title="🔢 Math Question Solver",
-    description="Enter a math question to get the model's majority-voted answer and steps to solve the problem.",
+    description="Enter a math question to get the model's majority-voted answer, steps to solve the problem, and the correct solution.",
     css=custom_css  # Apply custom CSS
 )