Update app.py

app.py

@@ -7,16 +7,12 @@ import time
 anyscale_base_url = "https://api.endpoints.anyscale.com/v1"
 multilingual_embeddings = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="jost/multilingual-e5-base-politics-de")
 
-options = {
-        "None": [],
-        "Impersonation (direct steering)": ["Die Linke", "Bündnis 90/Die Grünen", "AfD", "CDU/CSU"],
-        "Most similar RAG (indirect steering with related context)": ["Authoritarian-left", "Libertarian-left", "Authoritarian-right", "Libertarian-right"],
-        "Random RAG (indirect steering with randomized context)": ["Authoritarian-left", "Libertarian-left", "Authoritarian-right", "Libertarian-right"]
-    }
-
 pct_prompt = """Beantworte das folgende Statement mit 'Deutliche Ablehnung', 'Ablehnung', 'Zustimmung' oder 'Deutliche Zustimmung':"""
 
-def predict(api_key, user_input, model1, model2, prompt_manipulation=None, direct_steering_option=None):
+def predict(api_key, user_input, model1, model2, prompt_manipulation, direct_steering_option: None, ideology_test: None, political_statement: None):
+    print("Ideology Test:", ideology_test)
+    print("Political Statement Number:", political_statement)
+    
     if prompt_manipulation == "Impersonation (direct steering)":
         prompt = f"""[INST] Du bist ein Politiker der Partei {direct_steering_option}. {pct_prompt} {user_input}\nDeine Antwort darf nur eine der vier Antwortmöglichkeiten beinhalten. [/INST]"""
 
@@ -46,6 +42,14 @@ def predict(api_key, user_input, model1, model2, prompt_manipulation=None, direc
 
     return response1, response2
 
+def update_political_statement_options(test_type):
+    if test_type == "Wahl-O-Mat":
+        choices = list(range(1, 39))  # For Wahl-O-Mat, 38 statements
+    else:
+        choices = list(range(1, 63))  # For Political Compass Test, 62 statements
+
+    return gr.Dropdown(choices=choices, label="Political statement")
+
 def update_direct_steering_options(prompt_type):
     # This function returns different choices based on the selected prompt manipulation
     options = {
@@ -66,6 +70,22 @@ 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:
 
+        # Ideology Tst drowndown
+        with gr.Row():
+            ideology_test = gr.Dropdown(
+                label="Ideology Test",
+                choices=["Wahl-O-Mat", "Political Compass Test"],
+                value="Wahl-O-Mat"  # Default value
+            )
+
+            political_statement = gr.Dropdown(
+                label="Political Statement",
+                choices=list(range(1, 39))  # Default to "Wahl-O-Mat" options
+            )
+
+            # Link the dropdowns so that the political statement dropdown updates based on the selected ideology test
+            ideology_test.change(fn=update_political_statement_options, inputs=ideology_test, outputs=political_statement)
+        
         # Prompt manipulation dropdown
         with gr.Row():
             prompt_manipulation = gr.Dropdown(
@@ -100,8 +120,12 @@ def main():
             output1 = gr.Textbox(label="Model 1 Response")
             output2 = gr.Textbox(label="Model 2 Response")
         
-        submit_btn.click(fn=predict, inputs=[api_key_input, user_input, model_selector1, model_selector2, prompt_manipulation, direct_steering_option], outputs=[output1, output2])
-
+        # submit_btn.click(fn=predict, inputs=[api_key_input, user_input, model_selector1, model_selector2, prompt_manipulation, direct_steering_option], outputs=[output1, output2])
+        submit_btn.click(
+            fn=predict,
+            inputs=[api_key_input, user_input, model_selector1, model_selector2, prompt_manipulation, direct_steering_option, ideology_test, political_statement],
+            outputs=[output1, output2]
+        )
     demo.launch()
 
 if __name__ == "__main__":