Update app.py

app.py

@@ -22,13 +22,37 @@ def invoke(openai_api_key, task):
         del os.environ["OPENAI_API_KEY"]
         return result
 
+def clear():
+    return ""
+
 gr.close_all()
 
-demo = gr.Interface(fn = invoke, 
-                    inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1),
-                              gr.Textbox(label = "Task", value = os.environ['INPUT'])],
-                    outputs = [gr.Markdown(label = "Output", value = os.environ["OUTPUT"], line_breaks = True, sanitize_html = False)],
-                    title = "Multi-Agent AI: Coding",
-                    description = os.environ["DESCRIPTION"])
+with gr.Blocks() as assistant:
+    gr.Markdown("## Multi-Agent AI: Coding")
+    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)
+                task = gr.Textbox(label = "Task", value=os.environ["INPUT"], lines = 1)            
+            with gr.Row():
+                clear_btn = gr.ClearButton(
+                    components=[openai_api_key, task]
+                )
+                submit_btn = gr.Button("Submit", variant="primary")
+        with gr.Column(scale=3):
+            output = gr.Markdown(label = "Output", value=os.environ["OUTPUT"], line_breaks = True, sanitize_html = False)
+
+    clear_btn.click(
+        fn=clear,
+        outputs=output
+    )
+    
+    submit_btn.click(
+        fn=invoke,
+        inputs=[openai_api_key, task],
+        outputs=article
+    )
 
-demo.launch()
+assistant.launch()