jupylab-zero / app.py
Pyboxs's picture
fixed
dd7318c
raw
history blame
2.13 kB
# Filename: /home/user/app/app.py
# Version: 1691068773 (Unix timestamp at creation moment)
import os
import subprocess
from flask import Flask, redirect
import gradio as gr
from werkzeug.serving import WSGIRequestHandler
app = Flask(__name__)
log_file = "/home/user/app/app.log"
def install_packages():
# Install necessary packages if they are not already installed
with open(log_file, "a") as f:
subprocess.run(['pip3', 'install', '--quiet', 'jupyterlab', 'flask', 'gradio'], check=True)
def configure_jupyter():
# Generate Jupyter configuration
jupyter_config_dir = os.path.expanduser('~/.jupyter')
os.makedirs(jupyter_config_dir, exist_ok=True)
config_file = os.path.join(jupyter_config_dir, 'jupyter_lab_config.py')
jupyter_password = os.environ.get('JUPYPASS', '')
password_config = f"c.ServerApp.password = u'{jupyter_password}'" if jupyter_password else ''
with open(config_file, 'w') as f:
f.write(f"""
c.ServerApp.ip = '0.0.0.0'
c.ServerApp.port = 8888
c.ServerApp.open_browser = False
{password_config}
c.ServerApp.root_dir = '/home/user/app'
c.ServerApp.terminado_settings = {{'shell_command': ['bash']}}
c.ServerApp.allow_root = True
""")
def start_jupyter():
# Start JupyterLab on port 8888 with autoreload
with open(log_file, "a") as f:
subprocess.Popen(['jupyter-lab', '--port', '8888', '--autoreload'], stdout=f, stderr=f)
@app.route('/')
def home():
def greet(name):
return f"Hello {name}!"
iface = gr.Interface(fn=greet, inputs="text", outputs="text")
return iface.launch(inline=True)
@app.route('/lab')
def lab():
return redirect("http://localhost:8888", code=302)
if __name__ == "__main__":
with open(log_file, "a") as f:
f.write("===== Application Startup at {0} =====\n".format(subprocess.run(['date'], capture_output=True, text=True).stdout))
install_packages()
configure_jupyter()
start_jupyter()
# Disable logging for Flask
WSGIRequestHandler.log_request = lambda self, *args, **kwargs: None
app.run(host='0.0.0.0', port=7860, debug=False)