Update app.py

app.py

@@ -8,7 +8,7 @@ from typing import Optional, Literal
 
 class PromptInput(BaseModel):
     text: str = Field(..., description="The initial prompt text")
-    meta_prompt_choice: Literal["star","done","physics","morphosis", "verse", "phor","bolism"] = Field(..., description="Choice of meta prompt strategy")
+    meta_prompt_choice: Literal["star","done","physics","morphosis", "verse", "phor","bolism","math"] = Field(..., description="Choice of meta prompt strategy")
 
 class RefinementOutput(BaseModel):
     query_analysis: Optional[str] = None
@@ -34,6 +34,8 @@ class PromptRefiner:
             selected_meta_prompt = metadone
         elif prompt_input.meta_prompt_choice == "star":
             selected_meta_prompt = echo_prompt_refiner
+        elif prompt_input.meta_prompt_choice == "math":
+            selected_meta_prompt = math_meta_prompt           
         else:
             selected_meta_prompt = advanced_meta_prompt
     
@@ -225,6 +227,7 @@ 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_token)
     gradio_interface = GradioInterface(prompt_refiner)