Update app.py

app.py

@@ -223,6 +223,22 @@ def show_preview_from_url(url_input):
         return gr.update(visible=False), gr.update(value=url_input, visible=True)
     return gr.update(visible=False), gr.update(visible=False)
 
+
+def clear_model_cache():
+    """
+    Deletes all model weight folders so they are redownloaded fresh.
+    """
+    folders = [
+        "models/detection/weights",
+        "models/segmentation/weights",
+        "models/depth/weights"
+    ]
+    for folder in folders:
+        shutil.rmtree(folder, ignore_errors=True)
+        logger.info(f" Cleared: {folder}")
+    return " Model cache cleared. Models will be reloaded on next run."
+
+
 # Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## Unified Visual Intelligence System (UVIS)")
@@ -315,6 +331,14 @@ with gr.Blocks() as demo:
             json_out = gr.JSON(label="Scene JSON")
             zip_out = gr.File(label="Download Results")
 
+
+        with gr.Row():
+            clear_button = gr.Button("🧹 Clear Model Cache")
+            status_box = gr.Textbox(label="Status", interactive=False)
+        
+        clear_button.click(fn=clear_model_cache, inputs=[], outputs=[status_box])
+
+
     media_upload.change(show_preview_from_upload, inputs=media_upload, outputs=[img_out, vid_out])
     url.submit(show_preview_from_url, inputs=url, outputs=[img_out, vid_out])