Change "reset database" button location for more security

plugins/common.py

@@ -2,18 +2,25 @@ from global_vars import t, translations
 from app import Plugin
 import streamlit as st
 import torch
+from plugins.scansite import ScansitePlugin
 
 # Ajout des traductions spécifiques à ce plugin
 translations["en"].update({
     "work_directory": "Work Directory",
     "page_title": "News watcher",
+    "reset_database": "Reset database",
 })
 translations["fr"].update({
     "work_directory": "Répertoire de travail",
     "page_title": "Outil de veille",
+    "reset_database": "Réinitialiser la base de données",
 })
 
 class CommonPlugin(Plugin):
+    def __init__(self, name, plugin_manager):
+        super().__init__(name, plugin_manager)
+        self.scansite_plugin = ScansitePlugin('scansite', plugin_manager)
+
     def get_config_fields(self):
         return {
             "work_directory": {
@@ -38,6 +45,10 @@ class CommonPlugin(Plugin):
         torch.cuda.empty_cache()
         st.write("CUDA memory reset")
 
+        if st.button(t("reset_database")):
+            self.scansite_plugin.reset_database()
+            st.success(t("database_reset_success"))
+
 def remove_quotes(s):
     if s.startswith('"') and s.endswith('"'):
         return s[1:-1]

plugins/scansite.py

@@ -13,7 +13,6 @@ translations["en"].update({
     "total_links": "Total number of links",
     "annotated_links": "Number of annotated links",
     "known_tags": "Known tags",
-    "reset_database": "Reset database",
     "database_reset_success": "Database reset successfully",
     "launch_scan": "Launch scan",
     "scan_complete": "Scan complete",
@@ -46,7 +45,6 @@ translations["fr"].update({
     "total_links": "Nombre total de liens",
     "annotated_links": "Nombre de liens annotés",
     "known_tags": "Tags connus",
-    "reset_database": "Réinitialiser la base de données",
     "database_reset_success": "Base de données réinitialisée",
     "launch_scan": "Lancer le scan",
     "scan_complete": "Scan terminé",
@@ -128,10 +126,6 @@ class ScansitePlugin(Plugin):
         all_tags = self.get_all_tags()
         st.write(f"{t('known_tags')} :", ", ".join(all_tags))
 
-        if st.button(t("reset_database")):
-            self.reset_database()
-            st.success(t("database_reset_success"))
-
         if st.button(t("launch_scan")):
             self.launch_scan()
             st.success(t("scan_complete"))