Update app.py

app.py

@@ -32,7 +32,19 @@ def predict(api_key, user_input, model1, model2, prompt_manipulation=None, direc
 
     return response1, response2
 
-    
+def update_dropdown_options(selected_option):
+    if selected_option == "Impersonation (direct steering)":
+        return ["Option 1", "Option 2"]
+
+    elif selected_option == "Most similar RAG (indirect steering with related context)":
+        return ["Option 1", "Option 2"]
+
+    elif selected_option == "Random RAG (indirect steering with randomized context)":
+        return ["Option 1", "Option 2"]
+
+    else:
+        return []
+
 def main():
     description = "This is a simple interface to compare two model prodided by Anyscale. Please enter your API key and your message."
     with gr.Blocks() as demo:
@@ -51,20 +63,8 @@ def main():
         
         # Conditional dropdown - options revealed based on ‘Prompt Manipulation’ selection
         with gr.Row():
-            direct_steering_option = gr.Dropdown(
-                label="Direct Steering Options",
-                choices=["Option 1", "Option 2", "Option 3", "Option 4"],
-                visible=True  # Initially hidden
-            )
-        
-        # Making direct_steering_option visible based on prompt_manipulation choice
-        def show_direct_steering_options(prompt_choice):
-            return prompt_choice == "Impersonation (direct steering)"
-        
-        prompt_manipulation.change(fn=show_direct_steering_options, 
-                                   inputs=[prompt_manipulation], 
-                                   outputs=[direct_steering_option])
-
+            direct_steering_option = gr.Dropdown(label="Direct Steering Option")
+            prompt_manipulation.change(fn=update_dropdown_options, inputs=prompt_manipulation, outputs=direct_steering_option)
         
         with gr.Row():
             api_key_input = gr.Textbox(label="API Key", placeholder="Enter your API key here", show_label=True, type="password")