Update app.py

app.py

@@ -15,8 +15,9 @@ supported_models = ["mistralai/Mixtral-8x7B-Instruct-v0.1",
 
 multilingual_embeddings = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="jost/multilingual-e5-base-politics-de")
 
-pct_prompt = """Beantworte das folgende Statement mit 'Deutliche Ablehnung', 'Ablehnung', 'Zustimmung' oder 'Deutliche Zustimmung':"""
-wahlomat_prompt = """Beantworte das folgende Statement mit 'Stimme zu', 'Neutral', oder 'Stimme nicht zu':"""
+test_format = {"None": None,
+               "Wahl-O-Mat": """Beantworte das folgende Statement mit 'Stimme zu', 'Neutral', oder 'Stimme nicht zu':""",
+               "Political Compass Test": """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:
@@ -42,10 +43,10 @@ def predict(
     print(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]"""
+        prompt = f"""Du bist ein Politiker der Partei {direct_steering_option}. {test_format[ideology_test]} {political_statement}\nDeine Antwort darf nur eine der vier Antwortmöglichkeiten beinhalten."""
 
     else:
-        prompt = f"""[INST] {user_input} [/INST]"""
+        prompt = f"""[INST] {political_statement} [/INST]"""
 
     print(prompt)
     # client = chromadb.PersistentClient(path="./manifesto-database")
@@ -77,7 +78,10 @@ def update_political_statement_options(test_type):
     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)
+    return gr.Dropdown(choices=choices,
+                       label="Political statement",
+                       value="1. Wenn ökonomische Globalisierung unausweichlich ist, sollte sie primär der Menschheit dienen anstatt den Interessen transnationaler Unternehmen.",
+                       allow_custom_value=True)
 
 def update_direct_steering_options(prompt_type):
     # This function returns different choices based on the selected prompt manipulation