Update app.py

app.py

@@ -64,24 +64,31 @@ def login_required(f):
 
 def run_playit_claim_mode():
     """
-    Executes 'playit claim' to generate a claim code, required when the key is missing.
+    Executes 'playit agent claim generate' to get a claim code.
     """
     try:
-        # Run 'playit claim' synchronously to get the output
+        # --- FIX: Changed the command from ['playit', 'claim'] to ['playit', 'agent', 'claim', 'generate'] ---
         result = subprocess.run(
-            ['playit', 'claim'],
+            ['playit', 'agent', 'claim', 'generate'],
             capture_output=True,
             text=True,
             timeout=10 
         )
 
         if result.returncode == 0:
-            # The claim code is usually the last word printed by the command
-            claim_code = result.stdout.strip().split()[-1]
-            # Call the display function with the claim code
+            # Assuming the claim code is still the last word (e.g., "claim.playit.gg/CODE")
+            claim_output = result.stdout.strip()
+            # Look for the URL, which contains the code
+            claim_code = next((line.split('/')[-1] for line in claim_output.splitlines() if 'playit.gg/claim/' in line), 'CODE NOT FOUND')
+            
+            # If the code is still not reliably found, fall back to the last word if it looks like a code
+            if claim_code == 'CODE NOT FOUND':
+                 # This is a safer guess for a code
+                 claim_code = claim_output.split()[-1]
+            
+            server_manager.log(f"PlayIt Claim code generated: {claim_code}", "success")
             start_tunneling(claim_code=claim_code)
         else:
-            # --- MODIFIED PART: PRINTING THE ACTUAL ERROR ---
             error_details = result.stderr.strip() or "No detailed error output."
             print("=" * 50)
             print(f"PlayIt Claim FAILED. Details:")
@@ -89,7 +96,6 @@ def run_playit_claim_mode():
             print("=" * 50)
             server_manager.log(f"PlayIt claim failed: {error_details}", "error")
             start_tunneling(error_message="PlayIt 'claim' failed (See logs for details).")
-            # -----------------------------------------------
 
     except Exception as e:
         server_manager.log(f"Error during PlayIt claim: {e}", "error")