Add Problem and Sentence models to admin interface

Author: XanderVertegaal

backend/problem/admin.py

@@ -1,3 +1,18 @@
-from django.contrib import admin
-
-# Register your models here.
+from django.contrib import admin
+
+from problem.models import Problem, Sentence
+
+
+@admin.register(Problem)
+class ProblemAdmin(admin.ModelAdmin):
+    search_fields = ("entailment_label", "dataset")
+    list_display = ("__str__", "dataset", "entailment_label", "base", "get_hypothesis_text")
+
+    @admin.display(description="hypothesis text")
+    def get_hypothesis_text(self, obj):
+        return obj.hypothesis.text if obj.hypothesis else None
+
+
+@admin.register(Sentence)
+class SentenceAdmin(admin.ModelAdmin):
+    search_fields = ("text",)

backend/problem/models.py

@@ -7,6 +7,9 @@
 class Sentence(models.Model):
     text = models.TextField()
 
+    def __str__(self):
+        return self.text
+
 
 class Problem(models.Model):
     class Dataset(models.TextChoices):
@@ -74,3 +77,6 @@ def get_index(self, qs: QuerySet) -> int | None:
         except Exception as e:
             logger.exception(f"Error getting index for problem {self.pk}: {e}")
             return None
+
+    def __str__(self):
+        return f"Problem {self.pk} ({self.get_dataset_display()}): {self.get_entailment_label_display()}"  # type: ignore