add storage indicator

app.py

@@ -4,6 +4,7 @@ from huggingface_hub import hf_hub_download
 import os
 import gc
 import shutil
+import subprocess
 
 # Available models
 MODELS = {
@@ -34,6 +35,24 @@ with st.sidebar:
     top_p = st.slider("Top-P", 0.1, 1.0, 0.95)
     repeat_penalty = st.slider("Repetition Penalty", 1.0, 2.0, 1.1)
 
+    if st.button("🧹 Clear All Cached Models"):
+        try:
+            for f in os.listdir("models"):
+                if f.endswith(".gguf"):
+                    os.remove(os.path.join("models", f))
+            st.success("Model cache cleared.")
+        except Exception as e:
+            st.error(f"Failed to clear models: {e}")
+
+    if st.button("📦 Show Disk Usage"):
+        try:
+            usage = shutil.disk_usage(".")
+            used = usage.used / (1024**3)
+            free = usage.free / (1024**3)
+            st.info(f"Disk Used: {used:.2f} GB | Free: {free:.2f} GB")
+        except Exception as e:
+            st.error(f"Disk usage error: {e}")
+
 # Model info
 selected_model = MODELS[selected_model_name]
 model_path = os.path.join("models", selected_model["filename"])