Update app.py

app.py

@@ -181,6 +181,24 @@ def download_latest_model():
         flash(f"Error while downloading the model: {str(e)}", 'error')
         return redirect(request.referrer)
 
+@app.route('/remove_file', methods=['POST'])
+def remove_file():
+    try:
+        file_to_remove = session.get('uploaded_file', None)
+        if file_to_remove:
+            file_path = os.path.join(app.config['UPLOAD_FOLDER'], file_to_remove)
+            if os.path.exists(file_path):
+                os.remove(file_path)
+                flash(f"{file_to_remove} removed successfully.", "success")
+                session.pop('uploaded_file', None)
+            else:
+                flash("File not found.", "error")
+        else:
+            flash("No file selected for removal.", "error")
+    except Exception as e:
+        flash(f"Error while removing file: {str(e)}", "error")
+    return redirect(url_for('index'))
+
 # Route to serve uploaded files
 '''
 @app.route('/uploads/<filename>')