Update app.py

app.py

@@ -196,7 +196,8 @@ def filter_text(text,phraseC):
         if match:
           filtered_text = match.group(2)
           for phrase in phraseC:
-              filtered_text = re.sub(phrase, "", filtered_text, flags=re.DOTALL)
+              for wrd in phrase
+                  filtered_text = re.sub(wrd, "", filtered_text, flags=re.DOTALL)
           return filtered_text
         else:
           return filtered_text
@@ -300,6 +301,7 @@ def expand_prompt(prompt):
             do_sample=True,
         )
         enhanced_prompt = txt_tokenizer.decode(outputs[0], skip_special_tokens=True)
+        enhanced_prompt = filter_text(enhanced_prompt,prompt)
         input_text_2 = f"{system_prompt_rewrite} {user_prompt_rewrite_2} {enhanced_prompt}"
         encoded_inputs_2 = txt_tokenizer(input_text_2, return_tensors="pt", return_attention_mask=True).to("cuda:0")
         input_ids_2 = encoded_inputs_2["input_ids"].to("cuda:0")
@@ -315,7 +317,6 @@ def expand_prompt(prompt):
         # Use the encoded tensor 'text_inputs' here
         enhanced_prompt_2 = txt_tokenizer.decode(outputs_2[0], skip_special_tokens=True)
         print('-- generated prompt --')
-        enhanced_prompt = filter_text(enhanced_prompt,prompt)
         enhanced_prompt_2 = filter_text(enhanced_prompt_2,prompt)
         print('-- filtered prompt --')
         print(enhanced_prompt)