Skip to content

Commit

Permalink
feat(app): Support providing the server_name parameter
Browse files Browse the repository at this point in the history
  • Loading branch information
GuangChen2333 committed Jan 13, 2025
1 parent 9f7b7c1 commit 13e5f95
Showing 1 changed file with 17 additions and 11 deletions.
28 changes: 17 additions & 11 deletions app.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,9 @@
import os
import logging

from typing import Any

DEFAULT_SERVER_NAME = "127.0.0.1"
DEFAULT_PORT = 6969
MAX_PORT_ATTEMPTS = 10

Expand Down Expand Up @@ -62,7 +65,7 @@

# Define Gradio interface
with gr.Blocks(
theme=my_applio, title="Applio", css="footer{display:none !important}"
theme=my_applio, title="Applio", css="footer{display:none !important}"
) as Applio:
gr.Markdown("# Applio")
gr.Markdown(
Expand Down Expand Up @@ -111,28 +114,31 @@
)


def launch_gradio(port):
def launch_gradio(server_name: str, server_port: int):
Applio.launch(
favicon_path="assets/ICON.ico",
share="--share" in sys.argv,
inbrowser="--open" in sys.argv,
server_port=port,
server_name=server_name,
server_port=server_port,
)


def get_port_from_args():
if "--port" in sys.argv:
port_index = sys.argv.index("--port") + 1
if port_index < len(sys.argv):
return int(sys.argv[port_index])
return DEFAULT_PORT
def get_value_from_args(key: str, default: Any = None) -> Any:
if key in sys.argv:
index = sys.argv.index(key) + 1
if index < len(sys.argv):
return sys.argv[index]
return default


if __name__ == "__main__":
port = get_port_from_args()
port = int(get_value_from_args("--port", DEFAULT_PORT))
server = get_value_from_args("--server-name", DEFAULT_SERVER_NAME)

for _ in range(MAX_PORT_ATTEMPTS):
try:
launch_gradio(port)
launch_gradio(server, port)
break
except OSError:
print(
Expand Down

0 comments on commit 13e5f95

Please sign in to comment.