Update app.py

app.py

@@ -99,10 +99,16 @@ def generate(
     t.start()
 
     outputs = []
+    end_phrase = "the end."
     for text in streamer:
         processed_text = process_text(text)
         outputs.append(processed_text)
-        yield "".join(outputs)
+        current_output = "".join(outputs)
+        yield current_output
+        # Check if 'the end.' is in the current output, case-insensitive
+        if end_phrase in current_output.lower():
+            break  # Stop generating further if 'the end.' is found
+
     final_story = "".join(outputs)
     try:
         saved_story = Story(message=message, content=final_story).save()