From 3fcdfc5cb1d18b69938187ecff331265dd694b4f Mon Sep 17 00:00:00 2001
From: Zdenek Kasner <zdenek.kasner@gmail.com>
Date: Mon, 20 Jan 2025 10:16:15 +0100
Subject: [PATCH] Display resource size (hardcoded in the config for
 simplicity)

---
 factgenie/config/resources.yml                     | 11 ++++++++++-
 factgenie/static/css/custom.css                    |  4 ++++
 factgenie/templates/include/add_dataset_modal.html | 10 +++++++++-
 3 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/factgenie/config/resources.yml b/factgenie/config/resources.yml
index 83eef67c..25eb1d52 100644
--- a/factgenie/config/resources.yml
+++ b/factgenie/config/resources.yml
@@ -7,6 +7,7 @@ easy-vqa:
   source: https://github.com/vzhou842/easy-VQA
   splits:
   - test
+  download-size: 34 kB
 logicnlg-100:
   class: logicnlg.LogicNLG
   name: logicnlg-100
@@ -18,6 +19,7 @@ logicnlg-100:
   source: https://huggingface.co/datasets/kasnerz/logicnlg
   splits:
   - test
+  download-size: 122 kB
 quintd1-gsmarena:
   name: quintd1-gsmarena
   annotations:
@@ -36,6 +38,7 @@ quintd1-gsmarena:
   splits:
   - dev
   - test
+  download-size: 728 kB
 quintd1-ice-hockey:
   name: quintd1-ice-hockey
   annotations:
@@ -54,6 +57,7 @@ quintd1-ice-hockey:
   splits:
   - dev
   - test
+  download-size: 728 kB
 quintd1-openweather:
   name: quintd1-openweather
   annotations:
@@ -68,10 +72,11 @@ quintd1-openweather:
   - mistral
   - zephyr
   - gpt-3.5
-  source: https://gitQuintd1-hub.com/kasnerz/quintd/tree/main/data/quintd-1
+  source: https://github.com/kasnerz/quintd/tree/main/data/quintd-1
   splits:
   - dev
   - test
+  download-size: 728 kB
 quintd1-owid:
   name: quintd1-owid
   annotations:
@@ -91,6 +96,7 @@ quintd1-owid:
   splits:
   - dev
   - test
+  download-size: 728 kB
 quintd1-wikidata:
   name: quintd1-wikidata
   annotations:
@@ -110,6 +116,7 @@ quintd1-wikidata:
   splits:
   - dev
   - test
+  download-size: 728 kB
 xsum-debug:
   name: xsum-debug
   class: basic.PlainTextDataset
@@ -122,6 +129,7 @@ xsum-debug:
   source: https://github.com/EdinburghNLP/XSum
   splits:
   - test
+  download-size: 5 kB
 propaganda-techniques:
   name: propaganda-techniques
   class: propaganda_techniques.PropagandaTechniques
@@ -140,3 +148,4 @@ Loaded Language, Name Calling&Labeling, Repetition, Exaggeration&Minimization, D
     - train
     - dev
     - test
+  download-size: 1.49 MB
diff --git a/factgenie/static/css/custom.css b/factgenie/static/css/custom.css
index 818d08fa..c83d3fce 100755
--- a/factgenie/static/css/custom.css
+++ b/factgenie/static/css/custom.css
@@ -696,6 +696,10 @@ a:hover {
     max-width: 700px;
 }
 
+.modal-xl {
+    max-width: 1000px !important;
+}
+
 #about-btn {
     padding: 5px;
     /* border-radius: 15px; */
diff --git a/factgenie/templates/include/add_dataset_modal.html b/factgenie/templates/include/add_dataset_modal.html
index 3e1ea95f..3d4cf23a 100644
--- a/factgenie/templates/include/add_dataset_modal.html
+++ b/factgenie/templates/include/add_dataset_modal.html
@@ -1,5 +1,5 @@
 <div class="modal fade" id="add-dataset-modal" tabindex="-1">
-    <div class="modal-dialog">
+    <div class="modal-dialog modal-xl">
         <div class="modal-content">
             <div class="modal-header">
                 <h5 class="modal-title" id="about-modal-label">Add dataset</h5>
@@ -53,6 +53,7 @@ <h5 class="modal-title" id="about-modal-label">Add dataset</h5>
                                     <th scope="col">Campaigns</th>
                                     <th scope="col">Source</th>
                                     <th scope="col">Download</th>
+                                    <th scope="col">Size</th>
                                     <th scope="col" data-visible="false">Description</th>
                                 </tr>
                             </thead>
@@ -103,6 +104,13 @@ <h5 class="modal-title" id="about-modal-label">Add dataset</h5>
                                             style="margin: 10px 12px 10px 12px; {% if not dataset.downloaded %}display: none;{% endif %}"
                                             id="check-downloaded-{{ dataset_id }}"></i>
                                     </td>
+                                    <td class="download-size">
+                                            {% if dataset['download-size'] %}
+                                            {{ dataset['download-size']|safe }}
+                                            {% else %}
+                                            <i>-</i>
+                                            {% endif %}
+                                    </td>
                                     <td class="dataset-description" style="display: none;">
                                         <div class="blue-link">
                                             {% if dataset.description %}