Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,3 +1,42 @@
+import shutil
+import os
+import streamlit as st
+
+# ---------------------------
+# 🧹 One-time cache cleaner (prevents 50 GB overflow)
+# ---------------------------
+def clean_cache():
+    folders = [
+        "/root/.cache/huggingface",
+        "/root/.cache/transformers",
+        "/root/.cache/torch",
+        "/tmp/hf_cache",
+    ]
+    total_deleted = 0
+    for folder in folders:
+        if os.path.exists(folder):
+            # estimate size before deleting
+            size = sum(
+                os.path.getsize(os.path.join(dp, f)) for dp, _, files in os.walk(folder) for f in files
+            ) / (1024**3)
+            total_deleted += size
+            shutil.rmtree(folder, ignore_errors=True)
+    os.makedirs("/tmp/hf_cache", exist_ok=True)
+    print(f"🧹 Cleaned cache folders (~{total_deleted:.2f} GB removed)")
+
+def check_disk_usage():
+    """Log how much disk space is used (for debugging storage issues)."""
+    st.sidebar.markdown("### 💾 Disk Usage (for debugging)")
+    try:
+        usage = os.popen("du -sh /root/.cache /tmp 2>/dev/null").read()
+        st.sidebar.text(usage if usage else "No cache directories found.")
+    except Exception as e:
+        st.sidebar.text(f"⚠️ Could not fetch disk usage: {e}")
+
+# Run cleanup and diagnostics at startup
+clean_cache()
+check_disk_usage()
+
 import os
 import streamlit as st