Update app.py

app.py

@@ -33,7 +33,7 @@ def predict(input, msg_history=STARTING_PROMPT):
   return [response, msg_history]
   
 
-def prompt_builder_predict(questionType=None, difficulty=0, topic=None, prerequisites=None, caseStudy=False, additionalPrompt=None, msg_history=STARTING_PROMPT):
+def prompt_builder_predict(questionType=None, difficulty=0, topic=None, prerequisites=None, caseStudy=False, additionalPrompt=None, msg_history=STARTING_PROMPT, latex=False):
 
   level = ['Very Easy', 'Easy', 'Medium', 'Difficult', 'Extremely Difficult']
   prompt = 'randomly generatate a math question '
@@ -47,6 +47,8 @@ def prompt_builder_predict(questionType=None, difficulty=0, topic=None, prerequi
     prompt = prompt + f"This question will require to use the following methods to solve: {' and '.join(prerequisites)}. "
   if caseStudy:
     prompt = prompt + 'This question must be in the form of case study where it tries to test the application of the topic in the real life scenario. '
+  if latex:
+    prompt = prompt + 'Convert all mathematical equation parts of the question to LaTeX format. '      
   if additionalPrompt:
     prompt = prompt + f"In addition, {additionalPrompt}."
   
@@ -70,16 +72,18 @@ with gr.Blocks() as demo:
     topic = gr.Dropdown(["Simultaneous Equation", "Linear Equation",  "Derivatives", "Integrals",  "Optimization"], value='Simultaneous Equation',  label="Main Testing Topic")
     prerequisites = gr.Dropdown(["Elimination", "Subsitution", "Linear Equation", "Algebra",  "Geometry", "Trigonometry", "Logarithms", "Power Rule", "Sum Rule", 'Difference Rule', "Product Rule", "Quotient Rule", 'Reciprocal Rule', "Chain Rule", "Implicit Differentiation", "Logarithmic Differentiation"], multiselect=True, interactive=True, label="Prerequisite Topics")
 
-  caseStudy = gr.Checkbox(label="Case Study", info="Does this question test the application of theory in real life scenarios?")
+  with gr.Row():
+    caseStudy = gr.Checkbox(label="Case Study", info="Does this question test the application of theory in real life scenarios?")
+    latex = gr.Checkbox(label="LaTeX", info="Display all equations in LaTex format?")
+  
   additionalInfo = gr.Textbox(label="Additional information (prompt)", placeholder="Give a scenario where Jim and John are working in a garden....") 
 
-
   gen_btn = gr.Button("Generate A New Question")
 
   with gr.Row():
     question = gr.TextArea(label="Generated Question")
 
-  gen_btn.click(fn=prompt_builder_predict, inputs = [questionType, difficulty, topic, prerequisites, caseStudy, additionalInfo, msg_history], outputs= [question, msg_history])
+  gen_btn.click(fn=prompt_builder_predict, inputs = [questionType, difficulty, topic, prerequisites, caseStudy, additionalInfo, msg_history, latex], outputs= [question, msg_history])
   
   with gr.Row():
     prompt = gr.Textbox(label='Additional Prompt', info='Not satified with the result? Enter instructions to modify the question.', placeholder='Include the case study of....', visible=False)