Update app.py

app.py

@@ -1,10 +1,11 @@
-from src.utils.hf_packager import HFSpacePackager
 from src.core.cognitive_engine import CognitiveEngine
+from src.utils.hf_packager import HFSpacePackager
 import gradio as gr
 import time
 import os
+import psutil
 
-# Pre-load models during space initialization
+# Initialize components
 cognitive_engine = CognitiveEngine()
 packager = HFSpacePackager()
 
@@ -27,12 +28,10 @@ def run_cycle(progress=gr.Progress()):
         return f"⚠️ Critical error: {str(e)}"
 
 def get_resource_usage():
-    # Simplified resource monitoring
-    import psutil
     return {
-        "CPU": f"{psutil.cpu_percent()}%",
-        "Memory": f"{psutil.virtual_memory().percent}%",
-        "Disk": f"{psutil.disk_usage('/').percent}%"
+        "cpu": f"{psutil.cpu_percent()}%",
+        "memory": f"{psutil.virtual_memory().percent}%",
+        "disk": f"{psutil.disk_usage('/').percent}%"
     }
 
 with gr.Blocks(css="static/style.css", theme=gr.themes.Soft()) as demo:
@@ -42,13 +41,12 @@ with gr.Blocks(css="static/style.css", theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=1):
             start_btn = gr.Button("🚀 Start Improvement Cycle", variant="primary")
             status = gr.Textbox(label="Status", interactive=False)
-            hf_token = gr.Textbox(label="HF Token (for private snapshots)", type="password")
             
             gr.Markdown("### System Resources")
             resource_display = gr.JSON(label="Current Usage")
             
             demo.load(
-                lambda: None,
+                get_resource_usage,
                 inputs=[],
                 outputs=resource_display,
                 every=5