Update app.py

app.py

@@ -51,13 +51,22 @@ def get_command():
         cmd_data = json.load(f)
     return cmd_data.get("command", "")
 
-# Upload command output from PC
-def upload_output(name, password, output):
+# Upload command output from PC with timestamps
+def upload_output(name, password, output, sent_time):
     with open(status_file, "r") as f:
         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.
+        combined_output = (
+            f"Time Sent: {sent_time}\n"
+            "--------------------\n"
+            f"Time Received: {received_time}\n"
+            "--------------------\n"
+            f"{output}"
+        )
         with open(output_file, "w") as f:
-            json.dump({"output": output}, f)
+            json.dump({"output": combined_output}, f)
         return "Output uploaded successfully."
     return "Authentication failed."
 
@@ -86,11 +95,19 @@ with gr.Blocks() as ui:
     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
     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)
     hidden_upload_output = gr.Textbox(visible=False)
-    hidden_upload_btn.click(upload_output, inputs=[name_input, pass_input, output_input], outputs=hidden_upload_output, api_name="/upload_output")
+    hidden_upload_btn.click(
+        upload_output, 
+        inputs=[name_input, pass_input, output_input, sent_time_input],
+        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")