add root_path argument to gradio web server. (#2807)

Co-authored-by: bertls <s.bertl@iaea.org>

fastchat/serve/gradio_web_server.py

@@ -835,6 +835,11 @@ def build_demo(models):
         type=str,
         help='Set the gradio authentication file path. The file should contain one or more user:password pairs in this format: "u1:p1,u2:p2,u3:p3"',
     )
+    parser.add_argument(
+        "--gradio-root-path",
+        type=str,
+        help="Sets the gradio root path, eg /abc/def. Useful when running behind a reverse-proxy or at a custom URL path prefix"
+    )
     args = parser.parse_args()
     logger.info(f"args: {args}")
 
@@ -863,4 +868,5 @@ def build_demo(models):
         share=args.share,
         max_threads=200,
         auth=auth,
+        root_path=args.gradio_root_path
     )

fastchat/serve/gradio_web_server_multi.py

@@ -228,6 +228,11 @@ def build_demo(models, elo_results_file, leaderboard_table_file):
     parser.add_argument(
         "--leaderboard-table-file", type=str, help="Load leaderboard results and plots"
     )
+    parser.add_argument(
+        "--gradio-root-path",
+        type=str,
+        help="Sets the gradio root path, eg /abc/def. Useful when running behind a reverse-proxy or at a custom URL path prefix"
+    )
     args = parser.parse_args()
     logger.info(f"args: {args}")
 
@@ -267,4 +272,5 @@ def build_demo(models, elo_results_file, leaderboard_table_file):
         share=args.share,
         max_threads=200,
         auth=auth,
+        root_path=args.gradio_root_path
     )