Update app.py

app.py

@@ -74,20 +74,38 @@ def upload_file():
     logging.info(f"Files successfully uploaded: {uploaded_files}")
     return redirect(url_for('index'))
 
-@app.route('/remove_file')
+@app.route('/remove_file',methods=['POST'])
 def remove_file():
     uploaded_files = session.get('uploaded_files', [])
-    for filename in uploaded_files:
-        file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-        if os.path.exists(file_path):
-            os.remove(file_path)
-            logging.info(f"Removed file: {filename}")
-        else:
-            logging.warning(f"File not found for removal: {file_path}")  # More specific log
+    if uploaded_file:
+        for filename in uploaded_files:
+            file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+            if os.path.exists(file_path):
+                os.remove(file_path)
+                logging.info(f"Removed file: {filename}")
+            else:
+                logging.warning(f"File not found for removal: {file_path}")  # More specific log
+
+        session.pop('uploaded_files', None)  
+        flash('Files successfully removed')
+        logging.info("All uploaded files removed")
+    else:
+        flash('No file to remove.')
+        logging.warning(f"File not found for removal: {file_path") 
+    return redirect(url_for('index'))
 
-    session.pop('uploaded_files', None)  
-    flash('Files successfully removed')
-    logging.info("All uploaded files removed")
+@app.route('/reset_upload')
+def reset_upload():
+    """Reset the uploaded file and the processed data."""
+    uploaded_file = session.get('uploaded_file', [])
+    if uploaded_file:
+        for filename in uploaded_file:
+            os.remove(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+            logging.info(f"Removed file: {filename}")
+        session.pop('uploaded_files', None)
+        flash('Files successfully removed')
+    else:
+        flash('No file to remove.')    
     return redirect(url_for('index'))
 
 @app.route('/process', methods=['POST'])