Update app.py

app.py

@@ -18,13 +18,37 @@ def invoke(openai_api_key, num_moves = 10):
         del os.environ["OPENAI_API_KEY"]
         return result
 
-gr.close_all()
+def clear():
+    return ""
 
-demo = gr.Interface(fn = invoke, 
-                    inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1),
-                              gr.Number(label = "Number of Moves", value = 10, minimum = 1, maximum = 50)],
-                    outputs = [gr.Markdown(label = "Game", value=os.environ["OUTPUT"], line_breaks=True, sanitize_html=False)],
-                    title = "Multi-Agent AI: Chess",
-                    description = os.environ["DESCRIPTION"])
+gr.close_all()
 
-demo.launch()
+with gr.Blocks() as assistant:
+    gr.Markdown("## Multi-Agent AI: Chess")
+    gr.Markdown(os.environ.get("DESCRIPTION"))
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            with gr.Row():
+                openai_api_key = gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1)
+                num_moves = gr.Number(label = "Number of Moves", value = 10, minimum = 1, maximum = 50)
+            with gr.Row():
+                clear_btn = gr.ClearButton(
+                    components=[openai_api_key, num_moves]
+                )
+                submit_btn = gr.Button("Submit", variant="primary")
+        with gr.Column(scale=3):
+            game = gr.Markdown(label = "Game", value=os.environ["OUTPUT"], line_breaks=True, sanitize_html=False)
+
+    clear_btn.click(
+        fn=clear,
+        outputs=game
+    )
+    
+    submit_btn.click(
+        fn=invoke,
+        inputs=[openai_api_key, num_moves],
+        outputs=game
+    )
+
+assistant.launch()