Update project list

app/server/models.py

@@ -2,6 +2,7 @@ import string
 from django.core.exceptions import ValidationError
 from django.db import models
 from django.contrib.auth.models import User
+from django.contrib.staticfiles.storage import staticfiles_storage
 
 
 class Project(models.Model):
@@ -26,6 +27,17 @@ class Project(models.Model):
     def is_type_of(self, project_type):
         return project_type == self.project_type
 
+    @property
+    def image(self):
+        if self.is_type_of(self.DOCUMENT_CLASSIFICATION):
+            url = staticfiles_storage.url('images/cat-1045782_640.jpg')
+        elif self.is_type_of(self.SEQUENCE_LABELING):
+            url = staticfiles_storage.url('images/cat-3449999_640.jpg')
+        elif self.is_type_of(self.Seq2seq):
+            url = staticfiles_storage.url('images/tiger-768574_640.jpg')
+
+        return url
+
     def __str__(self):
         return self.name
 

app/server/templates/projects.html

@@ -53,7 +53,7 @@
                     <div class="card is-shady">
                         <div class="card-image">
                             <figure class="image is-4by3">
-                                <img src="https://source.unsplash.com/RWnpyGtY1aU" alt="Placeholder image" class="modal-button" data-target="modal-image2">
+                                <img src="{{ project.image }}" alt="Placeholder image" class="modal-button" data-target="modal-image2">
                             </figure>
                         </div>
                         <div class="card-content">