Update app.py

app.py

@@ -1,48 +1,49 @@
-import os
+# run_jupyterlab.py
 import subprocess
-import time
-import gradio as gr
-
-def launch_jupyter():
-    # Only install nodejs + localtunnel once
-    os.system("apt-get update && apt-get install -y nodejs")  # npm comes bundled
-    os.system("npm install -g localtunnel")
-    os.system("apt install curl -y")
-    os.system("curl -sSf https://sshx.io/get | sh")
-    os.system("sshx")
-    token = "letmein123"
-
-    # Launch JupyterLab with authentication and allow remote access
-    subprocess.Popen([
-        "jupyter", "lab",
-        "--ip=127.0.0.1",
-        "--port=6600",
-        "--no-browser",
-        f"--ServerApp.token={token}",
-        "--ServerApp.allow_origin='*'",
-        "--ServerApp.allow_remote_access=True"
-    ])
-
-    time.sleep(5)  # Give time for Jupyter to spin up
-
-    # Launch localtunnel
-    proc = subprocess.Popen(
-        ["lt", "--port", "6600"],
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT,
-        text=True
-    )
-
-    for line in proc.stdout:
-        if "your url is:" in line:
-            url = line.strip().split("your url is:")[1].strip()
-            return f"""
-                <h3>✅ JupyterLab Running</h3>
-                <p>Public URL: <a href="{url}?token={token}" target="_blank">{url}?token={token}</a></p>
-                <iframe src="{url}?token={token}" width="100%" height="600px" style="border: none;"></iframe>
-            """
-
-    return "<p>❌ Failed to start LocalTunnel or retrieve its URL</p>"
-
-demo = gr.Interface(fn=launch_jupyter, inputs=[], outputs=gr.HTML())
-demo.launch()
+import sys
+import os
+import signal # For handling termination signals
+
+def launch_jupyterlab():
+    """Launches JupyterLab programmatically on port 7860 and waits for it to finish."""
+    print("Attempting to launch JupyterLab on port 7860...")
+
+    command = [
+        sys.executable,
+        "-m",
+        "jupyterlab",
+        "--no-browser", # Remove this line if you want it to open automatically in your browser
+        "--port",
+        "7860",
+        "--ip=0.0.0.0" # IMPORTANT: Make sure it listens on all interfaces inside the container
+    ]
+
+    # Optional: Uncomment and modify if you want to specify a working directory
+    # command.extend(["--notebook-dir", "/path/to/your/notebooks"])
+
+    try:
+        print(f"Executing command: {' '.join(command)}")
+        # Use subprocess.Popen to run the command
+        process = subprocess.Popen(command)
+
+        print("\nJupyterLab launched! Check your terminal for the URL.")
+        print("You should see a URL like 'http://localhost:7860/lab?token=...'")
+        print("Copy and paste this URL into your web browser to access JupyterLab.")
+        print("Keep this terminal window open (or container running) while you are using JupyterLab.")
+
+        # --- IMPORTANT CHANGE HERE ---
+        # Wait for the JupyterLab process to terminate.
+        # This keeps the Python script (and thus the container's main process) alive.
+        print("\nWaiting for JupyterLab process to terminate...")
+        process.wait() 
+        print("JupyterLab process terminated.")
+
+    except FileNotFoundError:
+        print("Error: 'jupyterlab' command not found.")
+        print("Please ensure JupyterLab is installed. You can install it with:")
+        print("pip install jupyterlab")
+    except Exception as e:
+        print(f"An error occurred while launching JupyterLab: {e}")
+
+if __name__ == "__main__":
+    launch_jupyterlab()