tunnel cleanup
frdel committed
May 20, 2025 at 11:11 UTC
5ed43656a52c11aa8ebc328aa29892356cef53f5
2 files changed
+4
-29
requirements.txt
+1
@@ -25,6 +25,7 @@ paramiko==3.5.0
25
playwright==1.52.0
26
pypdf==4.3.1
27
python-dotenv==1.0.1
28
+pytz==2024.2
29
sentence-transformers==3.0.1
30
tiktoken==0.8.0
31
unstructured==0.15.13
run_tunnel.py
+3
-29
@@ -1,10 +1,6 @@
1
-from functools import wraps
2
-import signal
1
import threading
4
-import sys
5
-from flask import Flask, request, Response
6
-from python.helpers.files import get_abs_path
7
-from python.helpers import persist_chat, runtime, dotenv, process
2
+from flask import Flask, request
3
+from python.helpers import runtime, dotenv, process
4
from python.helpers.print_style import PrintStyle
5
6
from python.api.tunnel import Tunnel
@@ -47,32 +43,10 @@ def run():
43
request_handler=NoRequestLoggingWSGIRequestHandler,
44
threaded=True,
45
)
50
-
51
- printer = PrintStyle()
52
-
53
- # def signal_handler(sig=None, frame=None):
54
- # nonlocal tunnel, server, printer
55
- # with lock:
56
- # printer.print("Caught signal, stopping tunnel server...")
57
- # if server:
58
- # server.shutdown()
59
- # process.stop_server()
60
- # if tunnel:
61
- # tunnel.stop()
62
- # tunnel = None
63
- # printer.print("Tunnel server stopped")
64
- # sys.exit(0)
65
-
66
- # signal.signal(signal.SIGINT, signal_handler)
67
- # signal.signal(signal.SIGTERM, signal_handler)
68
-
46
+
47
process.set_server(server)
48
# server.log_startup()
49
server.serve_forever()
72
- # Run Flask app
73
- # app.run(
74
- # request_handler=NoRequestLoggingWSGIRequestHandler, port=port, host=host
75
- # )
50
finally:
51
# Clean up tunnel if it was started
52
if tunnel: