Fix the --port arg for browser mode (#447)

* fix port for browser command

* fix port

Author: Olivia-liu

README.md

@@ -146,10 +146,10 @@ Run a chatbot in your browser that’s supported by the model you specify in the
 **Examples**
 
 ```
-python3 torchchat.py browser stories15M --temperature 0 --num-samples 10
+python3 torchchat.py browser stories15M --temperature 0 --num-samples 100
 ```
 
-*Running on http://127.0.0.1:5000* should be printed out on the terminal. Click the link or go to [http://127.0.0.1:5000](http://127.0.0.1:5000) on your browser to start interacting with it.
+*Running on http://127.0.0.1:5000* should be printed out on the terminal. Click the link or go to [http://127.0.0.1:5000](http://127.0.0.1:5000) on your browser to start interacting with it. If port 5000 has already been taken, run the command again with `--port`, e.g. `--port 5001`.
 
 Enter some text in the input box, then hit the enter key or click the “SEND” button. After a second or two, the text you entered together with the generated text will be displayed. Repeat to have a conversation.
 

cli.py

@@ -36,10 +36,6 @@ def add_arguments_for_chat(parser):
 def add_arguments_for_browser(parser):
     # Only browser specific options should be here
     _add_arguments_common(parser)
-    parser.add_argument(
-        "--port", type=int, default=5000, help="Port for the web server in browser mode"
-    )
-    _add_arguments_common(parser)
 
 
 def add_arguments_for_download(parser):
@@ -294,6 +290,12 @@ def add_arguments(parser):
         default=".model-artifacts",
         help="The directory to store downloaded model artifacts",
     )
+    parser.add_argument(
+        "--port",
+        type=int,
+        default=5000,
+        help="Port for the web server in browser mode",
+    )
 
 
 def arg_init(args):

torchchat.py

@@ -125,18 +125,15 @@
         check_args(args, "browser")
 
         # Look for port from cmd args. Default to 5000 if not found.
-        # The port args will be passed directly to the Flask app.
         port = 5000
         i = 2
         while i < len(sys.argv):
-            # Check if the current argument is '--port'
             if sys.argv[i] == "--port":
-                # Check if there's a value immediately following '--port'
                 if i + 1 < len(sys.argv):
                     # Extract the value and remove '--port' and the value from sys.argv
                     port = sys.argv[i + 1]
-                    del sys.argv[i : i + 2]  # Delete '--port' and the value
-                    break  # Exit loop since port is found
+                    del sys.argv[i : i + 2]
+                    break
             else:
                 i += 1