Update app.py

app.py

@@ -3,12 +3,33 @@ import gradio as gr
 
 controller = SweetCommander()
 
-demo = gr.Interface(
-    fn = controller,
-    inputs = [
-        gr.Textbox(label="Name", placeholder="Enter your name"),
-        gr.Textbox(label="Input", placeholder="Enter your message")
-    ],
-    outputs = gr.Textbox(label="Result")
-)
-demo.launch()
+title = "BlueDice - Practice Space"
+
+# demo = gr.Interface(
+#     fn = controller,
+#     inputs = [
+#         gr.Textbox(label="Name", placeholder="Enter your name"),
+#         gr.Textbox(label="Input", placeholder="Enter your message")
+#     ],
+#     outputs = gr.Textbox(label="Result")
+# )
+# demo.launch()
+
+def predict(bot_name, bot_response, user_name, user_input, history = []):
+    history.append(f"{bot_name}: {bot_response}")
+    history.append(f"{user_name}: {user_input}")
+    response_generator = controller(user_name, history)
+    response_message = ""
+    for response_token in response_generator:
+        response_message += controller.tokenizer.decode(response_token, skip_special_tokens = True)
+        yield response_message, history
+    return response_message, history
+
+gr.Interface(
+    fn = predict,
+    title = title,
+    allow_flagging = "manual",
+    inputs = ["BotName", "Response", "UserName", "Input", "State"],
+    outputs = ["Response", "State"],
+    theme = 'gradio/seafoam'
+).launch()