Update app.py

app.py

@@ -9,7 +9,7 @@ from typing import Optional, Literal
 
 class PromptInput(BaseModel):
     text: str = Field(..., description="The initial prompt text")
-    meta_prompt_choice: Literal["superstar","star","done","physics","morphosis", "verse", "phor","bolism","math"] = Field(..., description="Choice of meta prompt strategy")
+    meta_prompt_choice: Literal["superstar","star","done","physics","morphosis", "verse", "phor","bolism","math", "math_meta"] = Field(..., description="Choice of meta prompt strategy")
 
 class RefinementOutput(BaseModel):
     query_analysis: Optional[str] = None
@@ -39,6 +39,8 @@ class PromptRefiner:
             selected_meta_prompt = advanced_echo_prompt_refiner
         elif prompt_input.meta_prompt_choice == "math":
             selected_meta_prompt = math_meta_prompt
+        elif prompt_input.meta_prompt_choice == "math_meta":
+            selected_meta_prompt = math_meta
         else:
             selected_meta_prompt = advanced_meta_prompt
     
@@ -123,7 +125,7 @@ class GradioInterface:
             with gr.Accordion("Meta Prompt explanation", open=False):
                 gr.Markdown(explanation_markdown)
             with gr.Row():
-                meta_prompt_choice = gr.Radio(["superstar", "star", "done", "physics", "morphosis", "verse", "phor","bolism","math"], label="Choose Meta Prompt", value="morphosis")
+                meta_prompt_choice = gr.Radio(["superstar", "star", "done", "physics", "morphosis", "verse", "phor","bolism","math","math_meta"], label="Choose Meta Prompt", value="morphosis")
                 refine_button = gr.Button("Refine Prompt")
             with gr.Row():
                 gr.Markdown("### Initial prompt analysis")
@@ -241,6 +243,7 @@ if __name__ == '__main__':
     new_meta_prompt = os.getenv('new_meta_prompt')   
     advanced_meta_prompt = os.getenv('advanced_meta_prompt')
     math_meta_prompt = os.getenv('math_meta_prompt')
+    math_meta = os.getenv('math_meta')
 
 
     prompt_refiner = PromptRefiner(api_key)