Update app.py

app.py

@@ -57,7 +57,7 @@ def upload_output(name, password, output, sent_time):
         status = json.load(f)
     if name in status and status[name]["password"] == password:
         received_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
-        # Format with time sent, a divider, then time received, another divider, and the actual output.
+        # Format the output with "Time Sent", a divider, then "Time Received", another divider, and the actual output.
         combined_output = (
             f"Time Sent: {sent_time}\n"
             "--------------------\n"
@@ -77,6 +77,7 @@ def get_output():
     return data.get("output", "")
 
 with gr.Blocks() as ui:
+    # PC info and status
     with gr.Row():
         name_input = gr.Textbox(label="PC Name")
         pass_input = gr.Textbox(label="Password", type="password")
@@ -84,18 +85,25 @@ with gr.Blocks() as ui:
     check_output = gr.Textbox(label="Status")
     check_btn.click(check_pc, inputs=name_input, outputs=check_output)
 
+    # Command sending row
     with gr.Row():
         cmd_input = gr.Textbox(label="Command", lines=5)  # Multi-line for larger scripts
         send_btn = gr.Button("Send Command")
         send_output = gr.Textbox(label="Response", lines=2)
     send_btn.click(send_command, inputs=[name_input, pass_input, cmd_input], outputs=send_output)
     
+    # Visible output display row
+    with gr.Row():
+        get_output_btn = gr.Button("Get Command Output")
+        output_display = gr.Textbox(label="Command Output", lines=10)
+    get_output_btn.click(get_output, inputs=[], outputs=output_display)
+    
     # Hidden endpoints for PC polling and uploading output:
     hidden_get_cmd_btn = gr.Button("Hidden Get Command", visible=False)
     hidden_cmd_output = gr.Textbox(visible=False)
     hidden_get_cmd_btn.click(get_command, inputs=[], outputs=hidden_cmd_output, api_name="/get_command")
     
-    # Hidden components for uploading output: include an extra textbox for sent_time
+    # Hidden components for uploading output: includes a textbox for sent_time.
     output_input = gr.Textbox(label="Output", visible=False)
     sent_time_input = gr.Textbox(label="Sent Time", visible=False)
     hidden_upload_btn = gr.Button("Hidden Upload Output", visible=False)
@@ -106,10 +114,5 @@ with gr.Blocks() as ui:
         outputs=hidden_upload_output, 
         api_name="/upload_output"
     )
-    
-    # Hidden endpoint to retrieve output
-    hidden_get_output_btn = gr.Button("Hidden Get Output", visible=False)
-    hidden_get_output = gr.Textbox(visible=False)
-    hidden_get_output_btn.click(get_output, inputs=[], outputs=hidden_get_output, api_name="/get_output")
 
 ui.launch(share=True)