[Bug] Fixing the browser chat interface after my cli changes. Also GET support. (#315)

Author: orionr

chat_in_browser.py

@@ -30,10 +30,10 @@ def main():
             output += line.decode('utf-8').strip() + "\n"
         return render_template('chat.html', convo="Hello! What is your prompt?", disable_input=disable_input)
 
-    @app.route('/chat', methods=['POST'])
+    @app.route('/chat', methods=['GET', 'POST'])
     def chat():
         # Retrieve the HTTP POST request parameter value from 'request.form' dictionary
-        _prompt = request.form.get('prompt')
+        _prompt = request.form.get('prompt', '')
         proc.stdin.write((_prompt + "\n").encode('utf-8'))
         proc.stdin.flush()
 

torchchat.py

@@ -125,7 +125,8 @@
             else:
                 i += 1
 
-        args_plus_chat = ['"{}"'.format(s) for s in sys.argv[2:]] + ["\"--chat\""]
+        # Construct arguments for the flask app minus 'browser' command plus '--chat'
+        args_plus_chat = ['"{}"'.format(s) for s in sys.argv[1:] if s != "browser"] + ['"--chat"']
         formatted_args = ", ".join(args_plus_chat)
         command = ["flask", "--app", "chat_in_browser:create_app(" + formatted_args + ")", "run", "--port", f"{port}"]
         subprocess.run(command)