Merge pull request #1573 from ReactionMechanismGenerator/quotegen

RMG-NLP
ReactionMechanismGenerator · Apr 8, 2019 · a5ab75c · a5ab75c
2 parents 45b8a13 + 1e17830
commit a5ab75c
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/environment_linux.yml b/environment_linux.yml
@@ -44,3 +44,4 @@ dependencies:
   - mpmath
   - dde
   - pyyaml
+  - textgenrnn
diff --git a/environment_mac.yml b/environment_mac.yml
@@ -43,3 +43,4 @@ dependencies:
   - mpmath
   - dde
   - pyyaml
+  - textgenrnn
diff --git a/environment_windows.yml b/environment_windows.yml
@@ -42,3 +42,4 @@ dependencies:
   - mpmath
   - dde
   - pyyaml
+  - textgenrnn
diff --git a/rmgpy/rmg/main.py b/rmgpy/rmg/main.py
@@ -1363,6 +1363,21 @@ def finish(self):
         """
         Complete the model generation.
         """
+        # Print neural network-generated quote
+        import datetime
+        import textwrap
+        try:
+            from textgenrnn.quotes import get_quote
+        except ImportError:
+            pass
+        else:
+            quote = '"' + get_quote() + '"'
+            logging.info('')
+            logging.info(textwrap.fill(quote, subsequent_indent=' '))
+            logging.info('             ---Quote-generating neural network, {}'.format(
+                datetime.datetime.now().strftime("%B %Y")
+            ))
+
         # Log end timestamp
         logging.info('')
         logging.info('RMG execution terminated at ' + time.asctime())