Update app.py

app.py

@@ -305,16 +305,18 @@ def expand_prompt(prompt):
             do_sample=True,
         )
         enhanced_prompt = txt_tokenizer.decode(outputs[0], skip_special_tokens=True)
-        print('-- generated prompt 1 --')
-        print(enhanced_prompt)
+        #print('-- generated prompt 1 --')
+        #print(enhanced_prompt)
         #enhanced_prompt = filter_text(enhanced_prompt,prompt)
         #enhanced_prompt = filter_text(enhanced_prompt,user_prompt_rewrite)
         #enhanced_prompt = filter_text(enhanced_prompt,system_prompt_rewrite)
         print('-- filtered prompt --')
+    
         #target_phrase = "The rewritten prompt:"
         #pattern = r".*(?=" + re.escape(target_phrase) + r")"
         #enhanced_prompt = re.sub(pattern, "", enhanced_prompt)
         #filter_text(enhanced_prompt,target_phrase)
+        enhanced_prompt = enhanced_prompt[len(input_text):]
         print(enhanced_prompt)
         '''
         input_text_2 = f"{system_prompt_rewrite} {user_prompt_rewrite_2} {enhanced_prompt}"