Update app.py

app.py

@@ -347,6 +347,7 @@ def expand_prompt(prompt):
             "Rephrase this scene to have more elaborate details: "
         )
         input_text = f"{system_prompt_rewrite} {user_prompt_rewrite} {prompt}"
+        
         print("-- got prompt --")
         # Encode the input text and include the attention mask
         encoded_inputs = txt_tokenizer(input_text, return_tensors="pt", return_attention_mask=True).to("cuda:0")
@@ -366,11 +367,12 @@ def expand_prompt(prompt):
         )
         enhanced_prompt = txt_tokenizer.decode(outputs[0], skip_special_tokens=True)
         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(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 --')
+        enhanced_prompt = enhanced_prompt[len(input_text):]
         print(enhanced_prompt)
         '''
         input_text_2 = f"{system_prompt_rewrite} {user_prompt_rewrite_2} {enhanced_prompt}"