Update app.py

app.py

@@ -51,11 +51,11 @@ def predict(api_key, user_input, model1, model2, prompt_manipulation, direct_ste
     return response1, response2
 
 def update_political_statement_options(test_type):
-    # Choose the right data based on the test type
+    # Append an index starting from 1 before each statement
     if test_type == "Wahl-O-Mat":
-        choices = [statement["text"] for statement in wahl_o_mat_data["statements"]]
-    else:
-        choices = [question["text"] for question in pct_data["questions"]]
+        choices = [f"{i+1}. {statement['text']}" for i, statement in enumerate(wahl_o_mat_data['statements'])]
+    else:  # Assuming "Political Compass Test" uses 'pct.json'
+        choices = [f"{i+1}. {question['text']}" for i, question in enumerate(pct_data['questions'])]
 
     return gr.Dropdown(choices=choices, label="Political statement", allow_custom_value=False)
 
@@ -88,9 +88,11 @@ def main():
                 filterable=False
             )
 
+            # Initialize 'political_statement' with default 'Wahl-O-Mat' values
+            political_statement_initial_choices = [f"{i+1}. {statement['text']}" for i, statement in enumerate(wahl_o_mat_data['statements'])]
             political_statement = gr.Dropdown(
                 label="Political Statement",
-                choices=[]  # Default to "Wahl-O-Mat" options
+                choices=political_statement_initial_choices,  # Set default to 'Wahl-O-Mat' statements
             )
 
             # Link the dropdowns so that the political statement dropdown updates based on the selected ideology test