From 57653af14f9c7c6ad660f93b2384623480695765 Mon Sep 17 00:00:00 2001 From: Abubakar Abid Date: Mon, 6 Jul 2020 18:40:29 -0500 Subject: [PATCH] networking passed in sys out --- gradio/interface.py | 2 +- gradio/networking.py | 12 ++++++++---- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/gradio/interface.py b/gradio/interface.py index 3c59a26fe0..4d5134522b 100644 --- a/gradio/interface.py +++ b/gradio/interface.py @@ -268,7 +268,7 @@ class Interface: if share: try: share_url = networking.setup_tunnel(server_port) - print("External URL:", share_url) + print("Running on External URL:", share_url) except RuntimeError: share_url = None if self.verbose: diff --git a/gradio/networking.py b/gradio/networking.py index bafe9fbda6..827d0f9065 100644 --- a/gradio/networking.py +++ b/gradio/networking.py @@ -14,7 +14,7 @@ from gradio.tunneling import create_tunnel import urllib.request from shutil import copyfile import requests -import os +import sys INITIAL_PORT_VALUE = ( @@ -114,9 +114,14 @@ def get_first_available_port(initial, final): ) -def serve_files_in_background(interface, port, directory_to_serve=None, server_name=LOCALHOST_NAME): +def serve_files_in_background(interface, port, directory_to_serve=None, server_name=LOCALHOST_NAME, + stdout=None): class HTTPHandler(SimpleHTTPRequestHandler): """This handler uses server.base_path instead of always using os.getcwd()""" + def __init__(self, *args): + if stdout is not None: + sys.stdout = stdout + super().__init__(*args) def _set_headers(self): self.send_response(200) @@ -134,7 +139,6 @@ def serve_files_in_background(interface, port, directory_to_serve=None, server_n def do_POST(self): # Read body of the request. - if self.path == "/api/predict/": # Make the prediction. self._set_headers() @@ -216,7 +220,7 @@ def start_simple_server(interface, directory_to_serve=None, server_name=None): INITIAL_PORT_VALUE, INITIAL_PORT_VALUE + TRY_NUM_PORTS ) httpd = serve_files_in_background( - interface, port, directory_to_serve, server_name) + interface, port, directory_to_serve, server_name, stdout=sys.stdout) return port, httpd