clean the pipeline storage

Author: farshad68

learn2rag/ui/__init__.py

@@ -443,6 +443,16 @@ def pipeline_action(name: str) -> 'str | werkzeug.wrappers.response.Response':
                 app.logger.exception(e)
                 app.logger.error('Could not stop the pipeline')
                 flash(pgettext('flash', 'Could not stop the pipeline: %(message)s', message=e), 'error')
+        elif request.form['action'] == 'clean':
+            try:
+                storage_path = expand_path(pipeline['storage_path'])
+                shutil.rmtree(storage_path)
+                flash(pgettext('flash', 'Successfully clean the pipeline storage: '), 'success')
+            except FileNotFoundError:
+                pass
+            except Exception as e:
+                app.logger.error('Failed to remove directory: %s, %s', storage_path, e)
+                flash(pgettext('flash', 'Could not clean the pipeline storage: %(message)s', message=e), 'error')
         return redirect(url_for('pipelines_list'))
 
     @app.get('/pipelines/<name>/logs/<file>')

learn2rag/ui/templates/_pipelines_list_table.html

@@ -102,6 +102,15 @@
                       </button>
                     </form>
                   </li>
+                  <li><hr class="dropdown-divider"></li>
+                  <li>
+                    <form method="POST" action="{{ url_for('pipeline_action', name=name) }}" hx-boost="true" hx-confirm="{{gettext('Are you sure you want to clean this pipeline storage: %(label)s?', label=pipeline.label)}}">
+                      <input type="hidden" name="action" value="clean"/>
+                      <button type="submit" class="dropdown-item text-bg-danger">
+                        {{pgettext('button', 'Clean storage')}}
+                      </button>
+                    </form>
+                  </li>
                 </ul>
               </div>
               {% endif %}