Update app.py

app.py

@@ -59,8 +59,11 @@ def responsevote(
     # Split the sentence into words
     words = lowercase_sentence.split()
     
-    words = ['partei' if word == 'afd' else word for word in words]
-        
+    words_to_replace = ['afd', 'npd']
+    
+    # Substitute "afd" and "npd" with "partei"
+    words = ['partei' if word in words_to_replace else word for word in words]
+            
     # Join the words back into a sentence
     prompt_sentence = ' '.join(words)
     formatted_prompt = format_promptc(system+"\n"+prompt_sentence)
@@ -97,9 +100,9 @@ def responsecritical(
     # Split the sentence into words
     words = lowercase_sentence.split()
     
-    # Check if "afd" is in the words list and remove it
-    if "afd" in words:
-        words.remove("afd")
+    words_to_remove = ['afd', 'npd']
+    words = [word for word in words if word not in words_to_remove]
+    
         
     # Join the words back into a sentence
     prompt_sentence = ' '.join(words)