code

.gitignore

@@ -0,0 +1,3 @@
+./.vscode
+./.idea
+./__pycache__

main.py

@@ -3,17 +3,27 @@ import subprocess
 import time
 import os
 import signal
+import sys
 
 app = FastAPI()
 
+
 @app.get("/")
 async def read_root():
     return {"message": "Hello, World!"}
 
 if __name__ == "__main__":
+    # Add the current directory to the Python path so that 'main' module can be found
+    script_dir = os.path.dirname(os.path.abspath(__file__))
+    if script_dir not in sys.path:
+        sys.path.insert(0, script_dir)
+
     # Start the server in a separate process
-    server_process = subprocess.Popen(['uvicorn', 'main:app', '--host', '0.0.0.0', '--port', '8080'])
-    
+    server_process = subprocess.Popen(
+        ['uvicorn', 'main:app', '--host', '0.0.0.0', '--port', '8080'],
+        cwd=script_dir  # Set the working directory for the subprocess
+    )
+
     try:
         print("Server started. Will shutdown in 10 seconds.")
         time.sleep(10)  # Wait for 10 seconds
@@ -21,4 +31,4 @@ if __name__ == "__main__":
         print("Shutting down server.")
         # Gracefully shut down the server
         server_process.send_signal(signal.SIGINT)
-        server_process.wait()
+        server_process.wait()

start_server.sh

@@ -6,4 +6,5 @@ echo "password: ${JUPYTER_TOKEN}" >>/home/user/.config/code-server/config.yaml
 cat /home/user/.config/code-server/config.yaml
 
 NOTEBOOK_DIR="/data"
+python /home/user/app/main.py
 code-server --proxy-domain atonyxu-code.hf.space --bind-addr 0.0.0.0:8080 /data >/data/start.log 2>&1 &