Spaces:
Running
Running
File size: 3,654 Bytes
e8c0a63 ef78f90 e8c0a63 ef78f90 e8c0a63 ef78f90 e8c0a63 ef78f90 e8c0a63 ef78f90 e8c0a63 ef78f90 e8c0a63 ef78f90 e8c0a63 ef78f90 e8c0a63 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import gradio as gr
from codeexecutor import get_majority_vote, type_check, postprocess_completion, draw_polynomial_plot
import re
iterations = 4
# Function to generate mock predictions (as the model isn't loaded)
def get_prediction(question):
return "Solve the following mathematical problem: what is the sum of polynomial 2x+3 and 3x?\n### Solution: To solve the problem of summing the polynomials \\(2x + 3\\) and \\(3x\\), we can follow these steps:\n\n1. Define the polynomials.\n2. Sum the polynomials.\n3. Simplify the resulting polynomial expression.\n\nThe sum of the polynomials \\(2x + 3\\) and \\(3x\\) is \\(\\boxed{5x + 3}\\).\n"
# Function to parse the prediction to extract the answer and steps
def parse_prediction(prediction):
lines = prediction.strip().split('\n')
answer = None
steps = []
for line in lines:
match = re.match(r'^\s*(?:Answer|answer)\s*[:=]\s*(.*)', line)
if match:
answer = match.group(1).strip()
else:
steps.append(line)
if answer is None:
answer = lines[-1].strip()
steps = lines
steps_text = '\n'.join(steps).strip()
return answer, steps_text
# Function to extract boxed answers
def extract_boxed_answer(text):
match = re.search(r'\\boxed\{(.*?)\}', text)
if match:
return match.group(1)
return None
# Function to perform majority voting and get steps
def majority_vote_with_steps(question, num_iterations=10):
all_predictions = []
all_answers = []
steps_list = []
for _ in range(num_iterations):
prediction = get_prediction(question)
answer, success = postprocess_completion(prediction, return_status=True, last_code_block=True)
if success:
all_predictions.append(prediction)
all_answers.append(answer)
steps_list.append(prediction)
else:
answer, steps = parse_prediction(prediction)
all_predictions.append(prediction)
all_answers.append(answer)
steps_list.append(steps)
if success:
majority_voted_ans = get_majority_vote(all_answers)
expression = majority_voted_ans
if type_check(expression) == "Polynomial":
plotfile = draw_polynomial_plot(expression)
else:
plotfile = None
# Find the steps corresponding to the majority voted answer
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 answer, steps_solution, plotfile
# Function to handle chat-like interaction
def chat_interface(history, question):
final_answer, steps_solution, plotfile = majority_vote_with_steps(question, iterations)
history.append(("User", question))
history.append(("MathBot", f"Answer: {final_answer}\nSteps:\n{steps_solution}"))
return history, plotfile
# Gradio app setup using Blocks for layout management
with gr.Blocks() as interface:
with gr.Column():
chat_history = gr.Chatbot(label="Chat with MathBot", elem_id="chat_history")
math_question = gr.Textbox(label="Your Question", placeholder="Ask a math question...", elem_id="math_question")
chatbot_output = gr.Chatbot(label="Chat History")
polynomial_plot = gr.Image(label="Polynomial Plot")
math_question.submit(chat_interface, inputs=[chat_history, math_question], outputs=[chatbot_output, polynomial_plot])
if __name__ == "__main__":
interface.launch()
|