Add clear_collection function and integrate delete button in DigitalFootprintDashboard

helper/telemetry.py

@@ -39,3 +39,23 @@ def collect_telemetry(data):
         print(f"Error sending data to MongoDB: {e}")
     finally:
         client.close()
+
+def clear_collection(data):
+    """
+    Deletes all documents from a specified MongoDB collection.
+    """
+    mongodb_uri = os.getenv("MONGODB_URI")
+    if not mongodb_uri:
+        print("Deletion skipped: No database configured.")
+        return
+
+    try:
+        client = MongoClient(mongodb_uri)
+        db = client.get_default_database()
+        collection = db[f"{data}"]  # Replace with your collection name
+        result = collection.delete_many({})
+        print(f"Deleted {result.deleted_count} documents from the collection.")
+    except Exception as e:
+        print(f"Error deleting documents: {e}")
+    finally:
+        client.close()

pages/home.py

@@ -21,6 +21,7 @@ from classes.amazon import Amazon
 from classes.ebay import eBay
 import asyncio
 from helper.upload_button import hide_button, unhide_button
+from helper.telemetry import clear_collection
 import time
 
 class DigitalFootprintDashboard:
@@ -131,6 +132,12 @@ class DigitalFootprintDashboard:
             self.content = Content(os.getenv('Model_Content'))
         return col1, col2, col3, col4
     
+    async def delete_button(self):
+        reset_button = st.button("RESET ALL",icon="🗑️", use_container_width=True)
+
+        if reset_button:
+            clear_collection("telemetry")
+
     async def run_analysis(self):
         result = await asyncio.gather(
             self.gtmetrix.process(), 
@@ -144,11 +151,12 @@ class DigitalFootprintDashboard:
             self.tiktok.process(), 
         )
         st.session_state.analyze = False
-
+        
     async def main(self):
         """Main method to run the dashboard"""
         await self.create_row1()
         await self.create_row2()
+        await self.delete_button()
         #self.run_analysis()
 
 # Main execution