The best way to Deploy Open WebUI with Safe OpenAI API Integration, Public Tunneling, and Browser-Based mostly Chat Entry

import os
import re
import time
import json
import shutil
import sign
import secrets and techniques
import subprocess
import urllib.request
from getpass import getpass
from pathlib import Path


print("Putting in Open WebUI and helper packages...")
subprocess.check_call([
   "python", "-m", "pip", "install", "-q",
   "open-webui",
   "requests",
   "nest_asyncio"
])


print("nEnter your OpenAI API key securely.")
openai_api_key = getpass("OpenAI API Key: ").strip()


if not openai_api_key:
   increase ValueError("OpenAI API key can't be empty.")


default_model = enter("Default mannequin to make use of inside Open WebUI [gpt-4o-mini]: ").strip()
if not default_model:
   default_model = "gpt-4o-mini"

os.environ["ENABLE_OPENAI_API"] = "True"
os.environ["OPENAI_API_KEY"] = openai_api_key
os.environ["OPENAI_API_BASE_URL"] = "https://api.openai.com/v1"
os.environ["WEBUI_SECRET_KEY"] = secrets and techniques.token_hex(32)
os.environ["WEBUI_NAME"] = "Open WebUI on Colab"
os.environ["DEFAULT_MODELS"] = default_model


data_dir = Path("/content material/open-webui-data")
data_dir.mkdir(mother and father=True, exist_ok=True)
os.environ["DATA_DIR"] = str(data_dir)

cloudflared_path = Path("/content material/cloudflared")
if not cloudflared_path.exists():
   print("nDownloading cloudflared...")
   url = "https://github.com/cloudflare/cloudflared/releases/newest/obtain/cloudflared-linux-amd64"
   urllib.request.urlretrieve(url, cloudflared_path)
   cloudflared_path.chmod(0o755)


print("nStarting Open WebUI server...")


server_log = open("/content material/open-webui-server.log", "w")
server_proc = subprocess.Popen(
   ["open-webui", "serve"],
   stdout=server_log,
   stderr=subprocess.STDOUT,
   env=os.environ.copy()
)

local_url = "http://127.0.0.1:8080"
prepared = False
for _ in vary(120):
   attempt:
       import requests
       r = requests.get(local_url, timeout=2)
       if r.status_code < 500:
           prepared = True
           break
   besides Exception:
       cross
   time.sleep(2)


if not prepared:
   server_log.shut()
   with open("/content material/open-webui-server.log", "r") as f:
       logs = f.learn()[-4000:]
   increase RuntimeError(
       "Open WebUI didn't begin efficiently.nn"
       "Current logs:n"
       f"{logs}"
   )


print("Open WebUI is working regionally at:", local_url)


print("nCreating public tunnel...")


tunnel_proc = subprocess.Popen(
   [str(cloudflared_path), "tunnel", "--url", local_url, "--no-autoupdate"],
   stdout=subprocess.PIPE,
   stderr=subprocess.STDOUT,
   textual content=True
)

public_url = None
start_time = time.time()
whereas time.time() - start_time < 90:
   line = tunnel_proc.stdout.readline()
   if not line:
       time.sleep(1)
       proceed
   match = re.search(r"https://[-a-zA-Z0-9]+.trycloudflare.com", line)
   if match:
       public_url = match.group(0)
       break


if not public_url:
   with open("/content material/open-webui-server.log", "r") as f:
       server_logs = f.learn()[-3000:]
   increase RuntimeError(
       "Tunnel began however no public URL was captured.nn"
       "Open WebUI server logs:n"
       f"{server_logs}"
   )


print("n" + "=" * 80)
print("Open WebUI is prepared.")
print("Public URL:", public_url)
print("Native URL :", local_url)
print("=" * 80)


print("nWhat to do subsequent:")
print("1. Open the Public URL.")
print("2. Create your admin account the primary time you open it.")
print("3. Go to the mannequin selector and select:", default_model)
print("4. Begin chatting with OpenAI by means of Open WebUI.")


print("nUseful notes:")
print("- Your OpenAI API key was handed by means of surroundings variables.")
print("- Information persists just for the present Colab runtime except you mount Drive.")
print("- If the tunnel stops, rerun the cell.")


def tail_open_webui_logs(traces=80):
   log_path = "/content material/open-webui-server.log"
   if not os.path.exists(log_path):
       print("No server log discovered.")
       return
   with open(log_path, "r") as f:
       content material = f.readlines()
   print("".be part of(content material[-lines:]))


def stop_open_webui():
   world server_proc, tunnel_proc, server_log
   for proc in [tunnel_proc, server_proc]:
       attempt:
           if proc and proc.ballot() is None:
               proc.terminate()
       besides Exception:
           cross
   attempt:
       server_log.shut()
   besides Exception:
       cross
   print("Stopped Open WebUI and tunnel.")


print("nHelpers accessible:")
print("- tail_open_webui_logs()")
print("- stop_open_webui()")