Update app.py

app.py

@@ -32,6 +32,15 @@ def predict(api_key, user_input, model1, model2, prompt_manipulation=None, direc
 
     return response1, response2
 
+def update_direct_steering_options(prompt_type):
+    # This function returns different choices based on the selected prompt manipulation
+    options = {
+        "None": [],
+        "Impersonation (direct steering)": ["Option 1", "Option 2", "Option 3"],
+        "Most similar RAG (indirect steering with related context)": ["Option A", "Option B", "Option C"],
+        "Random RAG (indirect steering with randomized context)": ["Random 1", "Random 2", "Random 3"]
+    }
+    return options.get(prompt_type, [])
 
 def main():
     description = "This is a simple interface to compare two model prodided by Anyscale. Please enter your API key and your message."
@@ -50,7 +59,10 @@ def main():
                 value="None", # default value
             )
 
-            direct_steering_option = gr.Dropdown(label="Direct Steering Option")
+            direct_steering_option = gr.Dropdown(label="Select party/ideology")
+
+            # Link the dropdowns so that the option dropdown updates based on the selected prompt manipulation
+            prompt_manipulation.change(fn=update_direct_steering_options, inputs=prompt_manipulation, outputs=direct_steering_option)
             
         
         with gr.Row():