From 2fd85eec17652b5c40d023eb7530c9d9afccfe30 Mon Sep 17 00:00:00 2001
From: chongy076 <50088438+chongy076@users.noreply.github.com>
Date: Mon, 1 May 2023 10:17:03 +0800
Subject: [PATCH 1/2] Improve socket serving and prevent session timeout

---
 app.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/app.py b/app.py
index 2e8084eb..7408e360 100644
--- a/app.py
+++ b/app.py
@@ -40,7 +40,7 @@ from geventwebsocket.handler import WebSocketHandler
 from gevent.pywsgi import WSGIServer
 
 app = Flask("GPT4All-WebUI", static_url_path="/static", static_folder="static")
-socketio = SocketIO(app, async_mode='gevent')
+socketio = SocketIO(app, async_mode='gevent', ping_timeout=30, ping_interval=15)
 app.config['SECRET_KEY'] = 'secret!'
 # Set the logging level to WARNING or higher
 logging.getLogger('socketio').setLevel(logging.WARNING)
@@ -617,20 +617,20 @@ if __name__ == "__main__":
             # Handle the error here
             print("WebSocket error:", e)
             super().handle_error(environ, start_response, e)
-
-    # chong -add socket server
-    http_server = WSGIServer((config["host"], config["port"]), app, handler_class=CustomWebSocketHandler)
-    http_server = WSGIServer((config["host"], config["port"]), app, handler_class=WebSocketHandler)
     
     url = f'http://{config["host"]}:{config["port"]}'
     
     print(f"Please open your browser and go to {url} to view the ui")
+
+    # chong -add socket server    
+    app.config['debug'] = config["debug"]
+
     if config["debug"]:
-        socketio.run(app,debug=True,  host=config["host"], port=config["port"])
+        print("debug mode:true")    
     else:
-        socketio.run(app, host=config["host"], port=config["port"])
-
-
+        print("debug mode:false")
+        
+    http_server = WSGIServer((config["host"], config["port"]), app, handler_class=WebSocketHandler)
 
     # if config["debug"]:
     #     app.run(debug=True, host=config["host"], port=config["port"])

From 0d16b4d6d2e997b99d08203777c3ce64dbfde827 Mon Sep 17 00:00:00 2001
From: chongy076 <50088438+chongy076@users.noreply.github.com>
Date: Mon, 1 May 2023 10:20:05 +0800
Subject: [PATCH 2/2] remove extra spacings

---
 app.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/app.py b/app.py
index 7408e360..12d0011e 100644
--- a/app.py
+++ b/app.py
@@ -610,7 +610,6 @@ if __name__ == "__main__":
     # app.config['executor'] = executor
     bot = Gpt4AllWebUI(app, socketio, config, personality, config_file_path)
 
-
     # chong Define custom WebSocketHandler with error handling 
     class CustomWebSocketHandler(WebSocketHandler):
         def handle_error(self, environ, start_response, e):