1 files changed, 12 insertions, 3 deletions
diff --git a/archivecontroller.cpp b/archivecontroller.cpp
index 3cdcd4a..50e9982 100644
--- a/archivecontroller.cpp
+++ b/archivecontroller.cpp
@@ -11,6 +11,7 @@
 #include <QMessageBox>
 #include <QInputDialog>
 #include <QTextEdit>
+#include <QStandardItemModel>
 
 #include "archivecontroller.h"
 #include "archivemodel.h"
@@ -46,7 +47,7 @@ void ArchiveController::setModels(ArchiveModel *amodel, ArchiveFilesModel *afile
     mArchiveFilesModel = afilesmodel;
 }
 
-void ArchiveController::setMappingModels(QStringListModel *actorModel, QStringListModel *genreModel){
+void ArchiveController::setMappingModels(QStandardItemModel *actorModel, QStandardItemModel *genreModel){
     mActorModel = actorModel;
     mGenreModel = genreModel;
 }
@@ -176,9 +177,17 @@ void ArchiveController::treeSelectionChanged(const QItemSelection &selected, con
     mArchiveFiles->expandAll();
     mArchiveView->setCurrentArchivePath(mArchiveModel->indexToPath(sel.first()));
     QStringList actors = mArchiveModel->actors(ids);
-    mActorModel->setStringList(actors);
+    mActorModel->clear();
+    foreach(QString actor, actors){
+        QStandardItem *newItem = new QStandardItem(QIcon(":/spreadingpants.png"), actor);
+        mActorModel->appendRow(newItem);
+    }
     QStringList genres = mArchiveModel->genres(ids);
-    mGenreModel->setStringList(genres);
+    mGenreModel->clear();
+    foreach(QString genre, genres){
+        QStandardItem *newItem = new QStandardItem(QIcon(":/spreadingpants.png"), genre);
+        mGenreModel->appendRow(newItem);
+    }
     QString metaData = mArchiveModel->metadata(*ids.begin());
     mMetadataView->setHtml(metaData);
 }