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"] = Field(..., description="Choice of meta prompt strategy")
+    meta_prompt_choice: Literal["superstar","star","done","physics","morphosis", "verse", "phor","bolism","math] = Field(..., description="Choice of meta prompt strategy")
 
 class RefinementOutput(BaseModel):
     query_analysis: Optional[str] = None
@@ -36,7 +36,9 @@ class PromptRefiner:
         elif prompt_input.meta_prompt_choice == "star":
             selected_meta_prompt = echo_prompt_refiner
         elif prompt_input.meta_prompt_choice == "superstar":
-            selected_meta_prompt = advanced_echo_prompt_refiner            
+            selected_meta_prompt = advanced_echo_prompt_refiner
+        elif prompt_input.meta_prompt_choice == "math":
+            selected_meta_prompt = math_meta_prompt
         else:
             selected_meta_prompt = advanced_meta_prompt
     
@@ -237,6 +239,8 @@ if __name__ == '__main__':
     original_meta_prompt = os.getenv('original_meta_prompt')    
     new_meta_prompt = os.getenv('new_meta_prompt')   
     advanced_meta_prompt = os.getenv('advanced_meta_prompt')
+    math_meta_prompt = os.getenv('metamath')
+
 
     prompt_refiner = PromptRefiner(api_key)
     gradio_interface = GradioInterface(prompt_refiner)