Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,8 @@ import numpy as np
|
|
| 8 |
import pandas as pd
|
| 9 |
from datetime import datetime
|
| 10 |
from typing import Dict, Any, List, Optional
|
|
|
|
|
|
|
| 11 |
|
| 12 |
# ----------------------------------------------------------------------
|
| 13 |
# Plotly for dashboards
|
|
@@ -359,6 +361,26 @@ def refresh_dashboard():
|
|
| 359 |
generate_action_timeline()
|
| 360 |
)
|
| 361 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 362 |
# ----------------------------------------------------------------------
|
| 363 |
# OSS capabilities (for status display)
|
| 364 |
# ----------------------------------------------------------------------
|
|
|
|
| 8 |
import pandas as pd
|
| 9 |
from datetime import datetime
|
| 10 |
from typing import Dict, Any, List, Optional
|
| 11 |
+
import threading
|
| 12 |
+
import psutil # added for memory monitoring
|
| 13 |
|
| 14 |
# ----------------------------------------------------------------------
|
| 15 |
# Plotly for dashboards
|
|
|
|
| 361 |
generate_action_timeline()
|
| 362 |
)
|
| 363 |
|
| 364 |
+
# ----------------------------------------------------------------------
|
| 365 |
+
# Memory monitoring (added for debugging)
|
| 366 |
+
# ----------------------------------------------------------------------
|
| 367 |
+
def log_memory_usage():
|
| 368 |
+
"""Periodically log memory usage to help diagnose timeouts."""
|
| 369 |
+
try:
|
| 370 |
+
process = psutil.Process(os.getpid())
|
| 371 |
+
mem_info = process.memory_info()
|
| 372 |
+
logger.info(f"Memory usage: RSS={mem_info.rss / 1e6:.1f} MB, VMS={mem_info.vms / 1e6:.1f} MB")
|
| 373 |
+
# Also log system memory if available
|
| 374 |
+
sys_mem = psutil.virtual_memory()
|
| 375 |
+
logger.info(f"System memory: {sys_mem.percent}% used, {sys_mem.available / 1e9:.1f} GB free")
|
| 376 |
+
except Exception as e:
|
| 377 |
+
logger.error(f"Failed to log memory: {e}")
|
| 378 |
+
# Schedule next check in 60 seconds
|
| 379 |
+
threading.Timer(60, log_memory_usage).start()
|
| 380 |
+
|
| 381 |
+
# Start memory logging (once)
|
| 382 |
+
log_memory_usage()
|
| 383 |
+
|
| 384 |
# ----------------------------------------------------------------------
|
| 385 |
# OSS capabilities (for status display)
|
| 386 |
# ----------------------------------------------------------------------
|