Update app.py

app.py

@@ -5,11 +5,13 @@ import time
 status_file = "status.json"
 command_file = "command.json"
 
+HEARTBEAT_TIMEOUT = 20  # seconds; if no heartbeat within this interval, PC is offline
+
 # Initialize storage files
 with open(status_file, "w") as f:
     json.dump({}, f)
 with open(command_file, "w") as f:
-    json.dump({}, f)
+    json.dump({"command": ""}, f)
 
 # Register PC as online
 def register_pc(name, password):
@@ -24,7 +26,7 @@ def register_pc(name, password):
 def check_pc(name):
     with open(status_file, "r") as f:
         status = json.load(f)
-    if name in status and time.time() - status[name]["timestamp"] < 60:
+    if name in status and time.time() - status[name]["timestamp"] < HEARTBEAT_TIMEOUT:
         return "Online"
     return "Offline"
 
@@ -37,7 +39,7 @@ def send_command(name, password, command):
     if name in status and status[name]["password"] == password:
         with open(command_file, "w") as f:
             json.dump({"command": command}, f)
-        return f"Command '{command}' sent to {name}."
+        return f"Command sent to {name}."
     return "Authentication failed."
 
 # Get command (for PC side polling)
@@ -55,12 +57,12 @@ with gr.Blocks() as ui:
     check_btn.click(check_pc, inputs=name_input, outputs=check_output)
 
     with gr.Row():
-        cmd_input = gr.Textbox(label="Command")
+        cmd_input = gr.Textbox(label="Command", lines=5)  # multi-line textbox for larger scripts
         send_btn = gr.Button("Send Command")
-        send_output = gr.Textbox(label="Response")
+        send_output = gr.Textbox(label="Response", lines=2)
     send_btn.click(send_command, inputs=[name_input, pass_input, cmd_input], outputs=send_output)
     
-    # Hidden components to expose the get_command API endpoint.
+    # Hidden endpoint to allow PC polling for commands
     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")