Ablility to select additional folders

imagegallery.h

@@ -3,11 +3,12 @@
 
 #include <QWidget>
 #include <QScrollArea>
-#include <QVBoxLayout>
+#include <QGridLayout>
 #include <QLabel>
 #include <QList>
 #include <QDir>
-
+#include <QGridLayout>
+#include <QScrollArea>
 class ImageGallery : public QWidget
 {
     Q_OBJECT
@@ -15,13 +16,15 @@ class ImageGallery : public QWidget
 public:
     explicit ImageGallery(QWidget *parent = nullptr);
     void addImage(const QString &imagePath);
-    void addImagesFromDirectory(const QString &directoryPath);
+    void addImagesFromDirectory(QDir &directory);
+    void extracted(QLayoutItem *&item);
+    void setColumns(int columns);
 
 private:
     QScrollArea *scrollArea;
     QWidget *scrollWidget;
-    QVBoxLayout *scrollLayout;
-    QList<QLabel*> imageLabels;
+    QGridLayout *gridLayout;
+    int currentColumnCount;
 };
 
 #endif // IMAGEGALLERY_H