added notebook agent in on a thread

notebook/agent.py

@@ -0,0 +1,43 @@
+import threading
+
+from flask import Flask, jsonify, request
+from werkzeug.serving import make_server
+
+app = Flask(__name__)
+
+@app.route("/chat", methods=["POST"])
+def chat():
+    data = request.get_json()
+    print(f"NOTEBOOK AGENT SERVER SIDECAR SUCCESSFULLY RECEIVED CHAT PAYLOAD: {data}")  # noqa: T201
+    return jsonify(data)
+
+
+class AgentServer:
+    def __init__(self):
+        self._server = make_server("0.0.0.0", 4242, app, threaded=True)
+        self._thread = threading.Thread(target=self._server.serve_forever)
+
+    def start(self):
+        print("Starting agent server...")  # noqa: T201
+        self._thread.start()
+
+    def stop(self):
+        print("Stopping agent server...")  # noqa: T201
+        self._server.shutdown()
+        self._thread.join()
+
+
+_server_instance = None
+
+
+def run_agent_server():
+    global _server_instance
+    if _server_instance is None:
+        _server_instance = AgentServer()
+        _server_instance.start()
+
+
+def stop_agent_server():
+    global _server_instance
+    if _server_instance:
+        _server_instance.stop()

notebook/app.py

@@ -375,8 +375,15 @@ class NotebookLauncher(ServerApp):
 
     def start(self) -> None:
         """Start the app."""
+        from .agent import run_agent_server
+        run_agent_server()
         super().start()
 
+    def stop(self) -> None:
+        from .agent import stop_agent_server
+        stop_agent_server()
+        super().stop()
+
 
 class ExecApp(Application):
     """Execute a command."""

notebook/server.py

@@ -1,7 +0,0 @@
-from flask import Flask, jsonify
-
-app = Flask(__name__)
-
-@app.route('/')
-def index():
-    return jsonify({"message": "Thesis Notebook Server is running."})