implemented deleting collections

database_implementation/sqlite_api.py

@@ -57,6 +57,20 @@ class SqliteDatabaseAPI(database_api.CrisisEventsDatabase):
         self._disconnect(connection, cur)
         return collection_id
     
+    def delete_collection(self, collection_name):
+        connection, cur = self._connect()
+        cur.execute("DELETE FROM event_collections WHERE collection_name = ?", (collection_name,))
+
+        # Check if any row was affected
+        if cur.rowcount > 0:
+            print(f"Deleted collection '{collection_name}'")
+            self._disconnect(connection, cur)
+            return True
+        else:
+            print(f"Collection '{collection_name}' not found")
+            self._disconnect(connection, cur)
+            return False
+    
     def create_raw_text_file(self,collection_id,path):
         connection, cur = self._connect()
         cur.execute("INSERT INTO raw_text_files VALUES (NULL, ?, ?)", (collection_id,path,))

flask_backend.py

@@ -186,6 +186,28 @@ def v1_create_collection():
         "collection_id":id
         }, 200
 
+@app.route('/api/v1/delete_collection', methods=['DELETE'])
+def v1_delete_collection():
+    data = request.json
+
+    if "collection_info" not in data:
+        abort(400)
+
+    collection_info = data["collection_info"]
+
+    if "collection_name" not in collection_info:
+        abort(400)
+
+    collection_name = collection_info["collection_name"]
+
+    # Call the database function to delete the collection
+    success = database.delete_collection(collection_name)
+
+    if success:
+        return jsonify({"status": "success", "message": f"Collection '{collection_name}' deleted"}), 200
+    else:
+        abort(404, f"Collection '{collection_name}' not found")
+
 @app.route('/api/v1/update_collection_glob', methods=['POST'])
 def v1_update_collection_glob():
 

frontend/crisis-events-text-summarization-frontend/src/App.js

@@ -127,6 +127,11 @@ function App() {
   updateFileLists();
   }
 
+  const handleDeletedCollection = () => {
+    setSelectedCollection(null);
+    showEditor();
+  }
+
   const logoutFunction = () => {
     setAuthenticated(false);
     sessionStorage.setItem('authenticated', 'false');
@@ -192,7 +197,7 @@ function App() {
         {/* <Box sx={{border: '1px dashed grey', padding:1}}> */}
           <Grid container spacing={2}>
             <Grid item xs={2}>
-              <CollectionList collections={loadedCollections} onSelectCollection={setSelectedCollection} onListUpdate={updateCollectionList} onFileSelect={onFileSelect} collectionFileCache={collectionFileCache}/>
+              <CollectionList collections={loadedCollections} onSelectCollection={setSelectedCollection} onListUpdate={updateCollectionList} onFileSelect={onFileSelect} collectionFileCache={collectionFileCache} onDeletedCollection={handleDeletedCollection}/>
             </Grid>
             <Grid item xs={10}>
               {

frontend/crisis-events-text-summarization-frontend/src/collection_list.js

 import * as React from 'react'
 import './Sidebar.css'
-import {dbCreateCollection, dbGetCollection, dbUploadRawTextFiles, dbUploadUrlFile, dbUploadRawHTMLFiles} from './database_util'
+import {dbCreateCollection, dbGetCollection, dbUploadRawTextFiles, dbUploadUrlFile, dbUploadRawHTMLFiles, dbDeleteCollection} from './database_util'
 
 
 //General MUI imports
@@ -85,6 +85,18 @@ const CollectionList = (props) => {
         props.onSelectCollection(detailCollection);
     }
 
+    const handleDelete = async (collectionName) => {
+        const isDeleted = await dbDeleteCollection(collectionName);
+        if (isDeleted) {
+          console.log(`Collection "${collectionName}" deleted successfully`);
+          props.onDeletedCollection(true);
+          props.onListUpdate();
+        } else {
+          console.error(`Failed to delete collection "${collectionName}"`);
+        }
+        handleDeleteClose();
+    }
+
     return (
         <div className="sidebar-container">
             <aside id="aside" className="sidebar">
@@ -184,7 +196,7 @@ const CollectionList = (props) => {
                 </DialogContent>
                 <DialogActions>
                     <Button onClick={handleDeleteClose}>Cancel</Button>
-                    <Button >Delete</Button>
+                    <Button onClick={() => handleDelete(displayedCollection)}>Delete</Button>
                 </DialogActions>
             </Dialog>
         </div>

frontend/crisis-events-text-summarization-frontend/src/database_util.js

@@ -29,6 +29,31 @@ export const  dbCreateCollection = async (newCollectionName, newCollectionType)
   return result
 };
 
+export const dbDeleteCollection = async (collectionName) => {
+  console.log("dbDeleteCollection");
+  const response = await fetch(
+    "http://127.0.0.1:5000/api/v1/delete_collection",
+    {
+      method: "DELETE",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({
+        "collection_info": { "collection_name": collectionName }
+      }),
+    }
+  );
+
+  const responseData = await response.json();
+  if (responseData.status === "success") {
+    console.log("dbDeleteCollection Success");
+    return true;
+  } else {
+    console.error("dbDeleteCollection Failed");
+    return false;
+  }
+};
+
 export const dbGetCollection = async (collection_id) => {
   console.log("dbGetCollection");
   let result = await fetch("http://127.0.0.1:5000//api/v1/get_collection?collection=" + collection_id,)