Update app.py

app.py

@@ -56,15 +56,22 @@ class Story(Document):
 
 # Utility function for prompts
 def make_prompt(entry):
-    return  f"### Human, Don't answer inappropriate messages:{entry} ### Assistant:"
+    return  f"### Human, Don't answer inappropriate messages, make it around 500 words. Don't use ;:{entry} ### Assistant:"
     # f"TELL A STORY, RELATE TO COMPUTER SCIENCE, INCLUDE ASSESMENTS. MAKE IT REALISTIC AND AROUND 500 WORDS, END THE STORY WITH "THE END.": {entry}"
 
 
 def process_text(text):
     parts = text.split('[')
-    clean_parts = [part.split(']')[1] if ']' in part else part for part in parts]
-    text = re.sub(r'assessment;', '', text)
-    return ''.join(clean_parts)
+    clean_parts = []
+    for part in parts:
+        sub_parts = part.split(']')
+        if len(sub_parts) > 1:
+            clean_parts.append(sub_parts[1])
+        else:
+            clean_parts.append(sub_parts[0])
+        cleaned_text = ''.join(clean_parts)
+        cleaned_text = re.sub(r'assessment;', '', cleaned_text)
+    return cleaned_text
     
 def contains_profanity(text, profanity_set):
     words = text.split()