Update app.py

app.py

@@ -87,8 +87,8 @@ def filter_text(text):
   pattern = f"(.*?){re.escape(phrase)}(.*)"
   match = re.search(pattern, text)
  # match = re.search(pattern, text,flags=re.DOTALL)
-  filtered_text = match.group(2)
-  return filtered_text
+ # filtered_text = match.group(2)
+  return text
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096
@@ -111,7 +111,7 @@ def infer(
     "You are an AI assistant that rewrites image prompts to be more descriptive and detailed."
     )
     user_prompt_rewrite = (
-    "Rewrite this prompt to be more descriptive and detailed: "
+    "Rewrite this prompt to be more descriptive and detailed and only return the rewritten text: "
     )
     input_text = f"{system_prompt_rewrite} {user_prompt_rewrite} {prompt}"
     print("-- got prompt --")