Update app.py

app.py

@@ -2,6 +2,7 @@ from chromadb.utils import embedding_functions
 import chromadb
 from openai import OpenAI
 import gradio as gr
+import json
 import time
 
 anyscale_base_url = "https://api.endpoints.anyscale.com/v1"
@@ -9,6 +10,13 @@ multilingual_embeddings = embedding_functions.SentenceTransformerEmbeddingFuncti
 
 pct_prompt = """Beantworte das folgende Statement mit 'Deutliche Ablehnung', 'Ablehnung', 'Zustimmung' oder 'Deutliche Zustimmung':"""
 
+def load_json_data(filepath):
+    with open(filepath, 'r', encoding='utf-8') as file:
+        return json.load(file)
+
+pct_data = load_json_data('data/pct.json')
+wahl_o_mat_data = load_json_data('data/wahl-o-mat.json')
+
 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)
@@ -43,12 +51,13 @@ 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
     if test_type == "Wahl-O-Mat":
-        choices = list(range(1, 39))  # For Wahl-O-Mat, 38 statements
+        choices = [statement["text"] for statement in wahl_o_mat_data["statements"]]
     else:
-        choices = list(range(1, 63))  # For Political Compass Test, 62 statements
+        choices = [question["text"] for question in pct_data["questions"]]
 
-    return gr.Dropdown(choices=choices, label="Political statement")
+    return gr.Dropdown(choices=choices, label="Political statement", allow_custom_value=False)
 
 def update_direct_steering_options(prompt_type):
     # This function returns different choices based on the selected prompt manipulation
@@ -81,7 +90,7 @@ def main():
 
             political_statement = gr.Dropdown(
                 label="Political Statement",
-                choices=list(range(1, 39))  # Default to "Wahl-O-Mat" options
+                choices=[]  # Default to "Wahl-O-Mat" options
             )
 
             # Link the dropdowns so that the political statement dropdown updates based on the selected ideology test