hufdjh'

app.py

@@ -70,11 +70,11 @@ def majority_vote_with_steps(question, num_iterations=10):
     majority_voted_ans = get_majority_vote(all_answers)
     if sucess:
             print(type_check(majority_voted_ans))
-            if type_check(expression) == "Polynomial":
-                plotfile = draw_polynomial_plot(expression) 
+            if type_check(majority_voted_ans) == "Polynomial":
+                plotfile = draw_polynomial_plot(majority_voted_ans) 
     else:
-        if os.path.exists("thankyou.png"):
-            plotfile = "thankyou.png"
+        if os.path.exists("polynomial_plot.png"):
+            plotfile = "polynomial_plot.png"
         else:
             plotfile = None
 

temp3.py

@@ -2,6 +2,9 @@ import gradio as gr
 from codeexecutor import get_majority_vote, type_check, postprocess_completion, draw_polynomial_plot
 import re
 iterations=4
+import base64
+from io import BytesIO
+# from PIL import Image
 
 # Mock function for generating predictions
 def get_prediction(question):
@@ -58,7 +61,7 @@ def majority_vote_with_steps(question, num_iterations=10):
         if type_check(expression) == "Polynomial":
             plotfile = draw_polynomial_plot(expression)
     else:
-        plotfile = None
+        plotfile = "polynomial_plot.png"
 
     # Find the steps corresponding to the majority voted answer
     for i, ans in enumerate(all_answers):
@@ -78,15 +81,27 @@ def chat_interface(history, question):
     
     # Convert the plot image to base64 for embedding in chat (if plot exists)
     if plotfile:
-        history.append(("MathBot", f"Answer: \n{steps_solution}\n![Polynomial Plot]({plotfile})"))
+        history.append(("what is the sum of polynomial 2x+3 and 3x?", f"Answer: \n{steps_solution}"))
+        
+        with open(plotfile, "rb") as image_file:
+            image_bytes = image_file.read()
+            base64_image = base64.b64encode(image_bytes).decode("utf-8")
+            image_data = f'<img src="data:image/png;base64,{base64_image}" width="300"/>'
+            history.append(("", image_data)) 
     else:
         history.append(("MathBot", f"Answer: \n{steps_solution}"))
     
-    return history # Return None for the image output as it's embedded in the chat
+    return history 
+
+css = """
+#math-question-textbox {
+    font-size: 24px !important;
+}
+"""
 
 # Gradio app setup using Blocks
-with gr.Blocks() as interface:
-    chatbot = gr.Chatbot(label="Chat with MathBot", elem_id="chat_history")
+with gr.Blocks(css=css) as interface:
+    chatbot = gr.Chatbot(label="Chat with MathBot", elem_id="chat_history",height="70vh")
     math_question = gr.Textbox(label="Your Question", placeholder="Ask a math question...", elem_id="math_question")
     
     math_question.submit(chat_interface, inputs=[chatbot, math_question], outputs=[chatbot])