Update app.py

app.py

@@ -61,34 +61,39 @@ def login_required(f):
 # ====================================================================
 # === START PLAYIT.GG AGENT FUNCTIONS (CLAIM/SECRET KEY LOGIC) ===
 # ====================================================================
-
 def run_playit_claim_mode():
-    """
-    Executes 'playit agent claim generate' to get a claim code.
-    """
     try:
-        # --- FIX: Changed the command from ['playit', 'claim'] to ['playit', 'agent', 'claim', 'generate'] ---
+        # --- FIX: Reverting command to the simplest form: ['playit', 'claim'] ---
         result = subprocess.run(
-            ['playit', 'agent', 'claim', 'generate'],
+            ['playit', 'claim'],
             capture_output=True,
             text=True,
             timeout=10 
         )
 
         if result.returncode == 0:
-            # 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
+            # Robustly search the output for the actual code (the part after the last /)
+            # This handles cases where it prints a full URL like claim.playit.gg/CODE
+            claim_code = 'CODE NOT FOUND'
+            for line in claim_output.splitlines():
+                if 'playit.gg/claim/' in line:
+                    claim_code = line.split('/')[-1].strip()
+                    break
+            
+            # Fallback: if no URL was found, try the very last word of the output, which is common.
+            if claim_code == 'CODE NOT FOUND' and claim_output:
                  claim_code = claim_output.split()[-1]
             
+            if claim_code == 'CODE NOT FOUND':
+                 raise ValueError("Could not find a valid claim code in playit output.")
+
+
             server_manager.log(f"PlayIt Claim code generated: {claim_code}", "success")
             start_tunneling(claim_code=claim_code)
         else:
+            # Command failed, print detailed error
             error_details = result.stderr.strip() or "No detailed error output."
             print("=" * 50)
             print(f"PlayIt Claim FAILED. Details:")
@@ -99,7 +104,7 @@ def run_playit_claim_mode():
 
     except Exception as e:
         server_manager.log(f"Error during PlayIt claim: {e}", "error")
-        start_tunneling(error_message="PlayIt claim encountered an unexpected error.")
+        start_tunneling(error_message=f"PlayIt claim encountered an unexpected error: {type(e).__name__}.")
 
 
 def run_playit_secret_mode():