From d5f1086a186abc7fc8ef1742ee57b9fdcd70b6cf Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Tue, 24 Sep 2019 14:00:49 -0600
Subject: [PATCH] fixed mental issue?

---
 generator/ctrl/ctrl_generator.py | 6 ++++++
 story/utils.py                   | 2 ++
 2 files changed, 8 insertions(+)

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index dc9420c..71a98d0 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -168,6 +168,8 @@ class CTRLGenerator():
     def result_replace(self, result):
         print("\n\nBEFORE RESULT_REPLACE:")
         print(repr(result))
+
+        first_letter_capitalized = result[0].isupper()
         result = result.replace("#", "")
         result = first_to_second_person(result)
         result = remove_profanity(result)
@@ -175,6 +177,9 @@ class CTRLGenerator():
         print("\n\nAFTER RESULT_REPLACE:")
         print(repr(result))
 
+        if not first_letter_capitalized:
+            result[0] = result[0].lower()
+
         return result
 
     def generate(self, prompt, options=None):
@@ -285,4 +290,5 @@ class CTRLGenerator():
             result = tokens_generated_so_far[prompt_length:]
             first_token = False
 
+
         return self.result_replace(result)
diff --git a/story/utils.py b/story/utils.py
index 50514e0..c2fc231 100644
--- a/story/utils.py
+++ b/story/utils.py
@@ -61,6 +61,8 @@ def cut_trailing_sentence(text):
 
 def replace_outside_quotes(text, current_word, repl_word):
 
+    current_word = current_word.replace(".", "\.")
+
     reg_expr = re.compile(current_word + '(?=([^"]*"[^"]*")*[^"]*$)')
 
     output = reg_expr.sub(repl_word, text)