code

main.py

@@ -1,7 +1,24 @@
 from fastapi import FastAPI
+import subprocess
+import time
+import os
+import signal
 
 app = FastAPI()
 
 @app.get("/")
-def read_root():
-    return {"message": "Hello, World!"}
+async def read_root():
+    return {"message": "Hello, World!"}
+
+if __name__ == "__main__":
+    # Start the server in a separate process
+    server_process = subprocess.Popen(['uvicorn', 'main:app', '--host', '0.0.0.0', '--port', '8080'])
+    
+    try:
+        print("Server started. Will shutdown in 10 seconds.")
+        time.sleep(10)  # Wait for 10 seconds
+    finally:
+        print("Shutting down server.")
+        # Gracefully shut down the server
+        server_process.send_signal(signal.SIGINT)
+        server_process.wait()

start_server.sh

@@ -1,16 +1,12 @@
 #!/bin/bash
 JUPYTER_TOKEN="${JUPYTER_TOKEN:=huggingface}"
-nohup code-server --bind-addr 0.0.0.0:8080 /data >/tmp/code.log 2>&1 &
-sleep 5s
-ps aux | grep 'code-server' | grep -v 'grep' | awk '{print $2}' | xargs kill -9
+code-server -h
 sed -i '/password/d' /home/user/.config/code-server/config.yaml
 echo "password: ${JUPYTER_TOKEN}" >>/home/user/.config/code-server/config.yaml
 cat /home/user/.config/code-server/config.yaml
 
 NOTEBOOK_DIR="/data"
 
-nohup uvicorn main:app --host 0.0.0.0 --port 8080 >/data/start.log 2>&1 &
+python main.py
 
-sleep 20s
-ps aux | grep 'uvicorn' | grep -v 'grep' | awk '{print $2}' | xargs kill -9
-code-server --bind-addr 0.0.0.0:8080 /data >/data/start.log 2>&1 &
+code-server --proxy-domain atonyxu-code.hf.space --bind-addr 0.0.0.0:8080 /data >/data/start.log 2>&1 &