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","math"] = Field(..., description="Choice of meta prompt strategy")
+    meta_prompt_choice: Literal["star","done","physics","morphosis", "verse", "phor","bolism","math","arpe"] = Field(..., description="Choice of meta prompt strategy")
 
 class RefinementOutput(BaseModel):
     query_analysis: Optional[str] = None
@@ -35,7 +35,9 @@ class PromptRefiner:
         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           
+            selected_meta_prompt = math_meta_prompt  
+        elif prompt_input.meta_prompt_choice == "arpe":
+            selected_meta_prompt = autoregressive_metaprompt
         else:
             selected_meta_prompt = advanced_meta_prompt
     
@@ -113,7 +115,7 @@ class GradioInterface:
             with gr.Accordion("Meta Prompt explanation", open=False):
                 gr.Markdown(explanation_markdown)
             with gr.Row():
-                meta_prompt_choice = gr.Radio(["star","done","physics","morphosis", "verse", "phor","bolism","math"], label="Choose Meta Prompt", value="star")
+                meta_prompt_choice = gr.Radio(["star","done","physics","morphosis", "verse", "phor","bolism","math","arpe"], label="Choose Meta Prompt", value="star")
                 refine_button = gr.Button("Refine Prompt")
             with gr.Row():
                 gr.Markdown("### Initial prompt analysis")
@@ -228,6 +230,8 @@ if __name__ == '__main__':
     new_meta_prompt = os.getenv('new_meta_prompt')   
     advanced_meta_prompt = os.getenv('advanced_meta_prompt')
     math_meta_prompt = os.getenv('metamath')
+    autoregressive_metaprompt = os.getenv('autoregressive_metaprompt')
+
 
     prompt_refiner = PromptRefiner(api_token)
     gradio_interface = GradioInterface(prompt_refiner)