Update app.py

app.py

@@ -1176,6 +1176,7 @@ kernel = ZeroEngine()
 session_id = "turtle170"
 
 with gr.Blocks(title="ZeroEngine V0.2", css=CUSTOM_CSS) as demo:
+    gr.LoginButton()
     # Header with Token Display
     with gr.Row():
         with gr.Column(scale=8):
@@ -1287,15 +1288,16 @@ with gr.Blocks(title="ZeroEngine V0.2", css=CUSTOM_CSS) as demo:
             )
 
     # --- UI LOGIC ---
-    def update_stats():
+    def update_stats(profile: gr.OAuthProfile | None):
         try:
-            m = ResourceMonitor.get_metrics()
-            balance = token_manager.get_balance(session_id)
-            return f"{m['ram_used_gb']}/{m['ram_total_gb']} GB", f"{m['cpu_usage_pct']}%", f"{balance}"
-        except Exception as e:
-            logger.error(f"Stats update error: {e}")
-            return "Error", "Error", "0.00"
-            return "Error", "Error"
+        m = ResourceMonitor.get_metrics()
+        current_user = profile.username if profile else "Guest"
+        balance = token_manager.get_balance(current_user)
+        
+        return f"{m['ram_used_gb']}/{m['ram_total_gb']} GB", f"{m['cpu_usage_pct']}%", f"{balance}"
+    except Exception as e:
+        logger.error(f"Stats update error: {e}")
+        return "Error", "Error", "0.00"
 
     def on_scan(repo):
         try: