From 00ec398183a4ae4a60f517faece81effa92f385d Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Tue, 16 Mar 2021 11:44:30 -0700
Subject: [PATCH 1/5] fix: typo in trainer

---
 mainscripts/Trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py
index 507bc9c..dfa6d9d 100644
--- a/mainscripts/Trainer.py
+++ b/mainscripts/Trainer.py
@@ -446,7 +446,7 @@ def main(**kwargs):
         while True:
             if not c2s.empty():
                 item = c2s.get()
-                op = input['op']
+                op = item['op']
                 if op == 'show':
                     is_waiting_preview = False
                     loss_history = item['loss_history'] if 'loss_history' in item.keys() else None

From 8087b35168cccb7b719790a2ead79c264d96ff7a Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Tue, 16 Mar 2021 11:53:58 -0700
Subject: [PATCH 2/5] Updates changelog

---
 CHANGELOG.md | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index e6222da..f0496fb 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -11,6 +11,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### In Progress
 - [Freezeable layers (encoder/decoder/etc.)](https://github.com/faceshiftlabs/DeepFaceLab/tree/feature/freezable-weights)
 
+## [1.1.4] - 2020-03-16
+### Fixed
+- Fixed bug when exiting from Web UI
+
 ## [1.1.3] - 2020-03-16
 ### Changed
 - Updated changelog with unreleased features, links to working branches
@@ -34,7 +38,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Reset stale master branch to [seranus/DeepFaceLab](https://github.com/seranus/DeepFaceLab), 
   21 commits ahead of [iperov/DeepFaceLab](https://github.com/iperov/DeepFaceLab) ([compare](https://github.com/iperov/DeepFaceLab/compare/4818183...seranus:3f5ae05))
 
-[Unreleased]: https://github.com/olivierlacan/keep-a-changelog/compare/v1.1.2...HEAD
+[Unreleased]: https://github.com/olivierlacan/keep-a-changelog/compare/v1.1.4...HEAD
+[1.1.4]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.1.3...v1.1.4
+[1.1.3]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.1.2...v1.1.3
 [1.1.2]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.1.1...v1.1.2
 [1.1.1]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.1.0...v1.1.1
 [1.1.0]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.0.0...v1.1.0

From 0e0e83688c8b359f9690c413b8dcd24644dac53f Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Tue, 16 Mar 2021 14:14:39 -0700
Subject: [PATCH 3/5] Close socket from client on exit

---
 flaskr/templates/index.html | 23 ++++++++++++-----------
 1 file changed, 12 insertions(+), 11 deletions(-)

diff --git a/flaskr/templates/index.html b/flaskr/templates/index.html
index d97326f..4ab78dd 100644
--- a/flaskr/templates/index.html
+++ b/flaskr/templates/index.html
@@ -13,12 +13,24 @@
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <script type="text/javascript">
       $(function() {
+        const socket = io.connect();
+        socket.on('preview', function(msg) {
+          console.log(msg);
+          $('img#preview').attr("src", "{{ url_for('preview_image') }}?q=" + new Date().getTime());
+        });
+
+        socket.on('loss', function(loss_string) {
+          console.log(loss_string);
+          $('div#loss').html(loss_string);
+        });
+
         function save() {
           $.post("{{ url_for('save') }}");
         }
 
         function exit() {
           $.post("{{ url_for('exit') }}");
+          socket.close();
         }
 
         function update() {
@@ -63,17 +75,6 @@
         $('button#zoom_next').click(zoom_next);
 
         $('img#preview').click(update);
-
-        const socket = io.connect();
-        socket.on('preview', function(msg) {
-          console.log(msg);
-          $('img#preview').attr("src", "{{ url_for('preview_image') }}?q=" + new Date().getTime());
-        });
-
-        socket.on('loss', function(loss_string) {
-          console.log(loss_string);
-          $('div#loss').html(loss_string);
-        });
       });
     </script>
 </head>

From 2324a7e2f4819966024ae06c52f585932806b2ef Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Tue, 16 Mar 2021 16:17:50 -0700
Subject: [PATCH 4/5] docs: updated changelog

---
 CHANGELOG.md | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index f0496fb..759e931 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -11,6 +11,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### In Progress
 - [Freezeable layers (encoder/decoder/etc.)](https://github.com/faceshiftlabs/DeepFaceLab/tree/feature/freezable-weights)
 
+## [1.1.5] - 2020-03-16
+### Fixed
+- Fixed unclosed websocket in Web UI client when exiting
+
 ## [1.1.4] - 2020-03-16
 ### Fixed
 - Fixed bug when exiting from Web UI

From 784a040b9190eccd6ad198fccc5e83867ba7c7bd Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Tue, 16 Mar 2021 16:18:05 -0700
Subject: [PATCH 5/5] docs: update changelog

---
 CHANGELOG.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 759e931..0322f6e 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -42,7 +42,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Reset stale master branch to [seranus/DeepFaceLab](https://github.com/seranus/DeepFaceLab), 
   21 commits ahead of [iperov/DeepFaceLab](https://github.com/iperov/DeepFaceLab) ([compare](https://github.com/iperov/DeepFaceLab/compare/4818183...seranus:3f5ae05))
 
-[Unreleased]: https://github.com/olivierlacan/keep-a-changelog/compare/v1.1.4...HEAD
+[Unreleased]: https://github.com/olivierlacan/keep-a-changelog/compare/v1.1.5...HEAD
+[1.1.5]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.1.4...v1.1.5
 [1.1.4]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.1.3...v1.1.4
 [1.1.3]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.1.2...v1.1.3
 [1.1.2]: https://github.com/faceshiftlabs/DeepFaceLab/compare/v1.1.1...v1.1.2