pearmut 1.0.1__tar.gz → 1.0.2__tar.gz

{pearmut-1.0.1 → pearmut-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pearmut
-Version: 1.0.1
+Version: 1.0.2
 Summary: A tool for evaluation of model outputs, primarily MT.
 Author-email: Vilém Zouhar <vilem.zouhar@gmail.com>
 License: MIT
@@ -19,7 +19,7 @@ Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Dynamic: license-file
 
-# 🍐Pearmut &nbsp; &nbsp; [![PyPi version](https://badgen.net/pypi/v/pearmut/)](https://pypi.org/project/pearmut) [![PyPI download/month](https://img.shields.io/pypi/dm/pearmut.svg)](https://pypi.python.org/pypi/pearmut/) [![PyPi license](https://badgen.net/pypi/license/pearmut/)](https://pypi.org/project/pearmut/) [![build status](https://github.com/zouharvi/pearmut/actions/workflows/test.yml/badge.svg)](https://github.com/zouharvi/pearmut/actions/workflows/test.yml)
+# 🍐Pearmut <br> [![PyPi version](https://badgen.net/pypi/v/pearmut/)](https://pypi.org/project/pearmut) [![PyPI download/month](https://img.shields.io/pypi/dm/pearmut.svg)](https://pypi.python.org/pypi/pearmut/) [![PyPi license](https://badgen.net/pypi/license/pearmut/)](https://pypi.org/project/pearmut/) [![build status](https://github.com/zouharvi/pearmut/actions/workflows/test.yml/badge.svg)](https://github.com/zouharvi/pearmut/actions/workflows/test.yml) [![arXiv](https://img.shields.io/badge/arXiv-2601.02933-b31b1b.svg?style=flat)](https://arxiv.org/abs/2601.02933)
 
 **Platform for Evaluation and Reviewing of Multilingual Tasks**: Evaluate model outputs for translation and NLP tasks with support for multimodal data (text, video, audio, images) and multiple annotation protocols ([DA](https://aclanthology.org/N15-1124/), [ESA](https://aclanthology.org/2024.wmt-1.131/), [ESA<sup>AI</sup>](https://aclanthology.org/2025.naacl-long.255/), [MQM](https://doi.org/10.1162/tacl_a_00437), and more!).
 
@@ -142,6 +142,21 @@ The `shuffle` parameter in campaign `info` controls this behavior:
 }
 ```
 
+### Showing Model Names
+
+By default, model names are hidden to avoid biasing annotators. To display model names on top of each output block, set `show_model_names` to `true`:
+```python
+{
+  "info": {
+    "assignment": "task-based",
+    "protocol": "ESA",
+    "show_model_names": true  # Default: false.
+  },
+  "campaign_id": "my_campaign",
+  "data": [...]
+}
+```
+
 ### Custom Score Sliders
 
 For multi-dimensional evaluation tasks (e.g., assessing fluency on a Likert scale), you can define custom sliders with specific ranges and steps:
@@ -163,6 +178,25 @@ For multi-dimensional evaluation tasks (e.g., assessing fluency on a Likert scal
 
 When `sliders` is specified, only the custom sliders are shown. Each slider must have `name`, `min`, `max`, and `step` properties. All sliders must be answered before proceeding.
 
+### Textfield for Post-editing/Translation
+
+Enable a textfield for post-editing or translation tasks using the `textfield` parameter in `info`. The textfield content is stored in annotations alongside scores and error spans.
+
+```python
+{
+  "info": {
+    "protocol": "DA",
+    "textfield": "prefilled"  # Options: null, "hidden", "visible", "prefilled"
+  }
+}
+```
+
+**Textfield modes:**
+- `null` or omitted: No textfield (default)
+- `"hidden"`: Textfield hidden by default, shown by clicking a button
+- `"visible"`: Textfield always visible
+- `"prefilled"`: Textfield visible and pre-filled with model output for post-editing
+
 ### Custom Instructions
 
 Set campaign-level instructions using the `instructions` field in `info` (supports HTML).
@@ -401,7 +435,7 @@ Customize the goodbye message shown to users when they complete all annotations
   - **Score**: Numeric quality rating (0-100)
   - **Error Spans**: Text highlights marking errors with severity (`minor`, `major`)
   - **Error Categories**: MQM taxonomy labels for errors
-- **Template**: The annotation interface type. The `basic` template supports comparing multiple outputs simultaneously.
+- **Template**: The annotation interface type. The `annotate` template supports comparing multiple outputs simultaneously.
 - **Assignment**: The method for distributing items to users:
   - **Task-based**: Each user has predefined items
   - **Single-stream**: Users draw from a shared pool with random assignment
@@ -432,7 +466,7 @@ pearmut run
 2. Add build rule to `webpack.config.js`
 3. Reference as `info->template` in campaign JSON
 
-See [web/src/basic.ts](web/src/basic.ts) for example.
+See [web/src/annotate.ts](web/src/annotate.ts) for example.
 
 ### Deployment
 
@@ -443,68 +477,15 @@ Run on public server or tunnel local port to public IP/domain and run locally.
 If you use this work in your paper, please cite as following.
 ```bibtex
 @misc{zouhar2026pearmut,
-  author = {Zouhar, Vilém},
-  title = {Pearmut: Human Evaluation of Translation Made Trivial},
-  year = {2026}
+      title={Pearmut: Human Evaluation of Translation Made Trivial}, 
+      author={Vilém Zouhar and Tom Kocmi},
+      year={2026},
+      eprint={2601.02933},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2601.02933}, 
 }
 ```
 
 Contributions are welcome! Please reach out to [Vilém Zouhar](mailto:vilem.zouhar@gmail.com).
-
-# Changelog
-
-- v1.0.1
-  - Support RTL languages
-  - Add boxes for references
-  - Add custom score sliders for multi-dimensional evaluation
-  - Make instructions customizable and protocol-dependent
-  - Support custom sliders
-  - Purge/reset whole tasks from dashboard
-  - Fix resetting individual users in single-stream/dynamic
-  - Fix notification stacking
-  - Add campaigns from dashboard
-- v0.3.3
-  - Rename `doc_id` to `item_id`
-  - Add Typst, LaTeX, and PDF export for model ranking tables. Hide them by default.
-  - Add dynamic assignment type with contrastive model comparison
-  - Add `instructions_goodbye` field with variable substitution
-  - Add visual anchors at 33% and 66% on sliders
-  - Add German→English ESA tutorial with attention checks
-  - Validate document model consistency before shuffle
-  - Fix UI block on any interaction
-- v0.3.2
-  - Revert seeding of user IDs
-  - Set ESA (Error Span Annotation) as default
-  - Update server IP address configuration
-  - Show approximate alignment by default
-  - Unify pointwise and listwise interfaces into `basic`
-  - Refactor protocol configuration (breaking change)
-- v0.2.11
-  - Add comment field in settings panel
-  - Add `score_gt` validation for listwise comparisons
-  - Add Content-Disposition headers for proper download filenames
-  - Add model results display to dashboard with rankings
-  - Add campaign file structure validation
-  - Purge command now unlinks assets
-- v0.2.6
-  - Add frozen annotation links feature for view-only mode
-  - Add word-level annotation mode toggle for error spans
-  - Add `[missing]` token support
-  - Improve frontend speed and cleanup toolboxes on item load
-  - Host assets via symlinks
-  - Add validation threshold for success/fail tokens
-  - Implement reset masking for annotations
-  - Allow pre-defined user IDs and tokens in campaign data
-- v0.1.1
-  - Set server defaults and add VM launch scripts
-  - Add warning dialog when navigating away with unsaved work
-  - Add tutorial validation support for pointwise and listwise
-  - Add ability to preview existing annotations via progress bar
-  - Add support for ESA<sup>AI</sup> pre-filled error_spans
-  - Rename pairwise to listwise and update layout
-  - Implement single-stream assignment type
-- v0.0.3
-  - Support multimodal inputs and outputs
-  - Add dashboard
-  - Implement ESA (Error Span Annotation) and MQM support
-
+See changes in [CHANGELOG.md](CHANGELOG.md).

{pearmut-1.0.1 → pearmut-1.0.2}/README.md

@@ -1,4 +1,4 @@
-# 🍐Pearmut     [![PyPi version](https://badgen.net/pypi/v/pearmut/)](https://pypi.org/project/pearmut) [![PyPI download/month](https://img.shields.io/pypi/dm/pearmut.svg)](https://pypi.python.org/pypi/pearmut/) [![PyPi license](https://badgen.net/pypi/license/pearmut/)](https://pypi.org/project/pearmut/) [![build status](https://github.com/zouharvi/pearmut/actions/workflows/test.yml/badge.svg)](https://github.com/zouharvi/pearmut/actions/workflows/test.yml)
+# 🍐Pearmut <br> [![PyPi version](https://badgen.net/pypi/v/pearmut/)](https://pypi.org/project/pearmut) [![PyPI download/month](https://img.shields.io/pypi/dm/pearmut.svg)](https://pypi.python.org/pypi/pearmut/) [![PyPi license](https://badgen.net/pypi/license/pearmut/)](https://pypi.org/project/pearmut/) [![build status](https://github.com/zouharvi/pearmut/actions/workflows/test.yml/badge.svg)](https://github.com/zouharvi/pearmut/actions/workflows/test.yml) [![arXiv](https://img.shields.io/badge/arXiv-2601.02933-b31b1b.svg?style=flat)](https://arxiv.org/abs/2601.02933)
 
 **Platform for Evaluation and Reviewing of Multilingual Tasks**: Evaluate model outputs for translation and NLP tasks with support for multimodal data (text, video, audio, images) and multiple annotation protocols ([DA](https://aclanthology.org/N15-1124/), [ESA](https://aclanthology.org/2024.wmt-1.131/), [ESA<sup>AI</sup>](https://aclanthology.org/2025.naacl-long.255/), [MQM](https://doi.org/10.1162/tacl_a_00437), and more!).
 
@@ -121,6 +121,21 @@ The `shuffle` parameter in campaign `info` controls this behavior:
 }
 ```
 
+### Showing Model Names
+
+By default, model names are hidden to avoid biasing annotators. To display model names on top of each output block, set `show_model_names` to `true`:
+```python
+{
+  "info": {
+    "assignment": "task-based",
+    "protocol": "ESA",
+    "show_model_names": true  # Default: false.
+  },
+  "campaign_id": "my_campaign",
+  "data": [...]
+}
+```
+
 ### Custom Score Sliders
 
 For multi-dimensional evaluation tasks (e.g., assessing fluency on a Likert scale), you can define custom sliders with specific ranges and steps:
@@ -142,6 +157,25 @@ For multi-dimensional evaluation tasks (e.g., assessing fluency on a Likert scal
 
 When `sliders` is specified, only the custom sliders are shown. Each slider must have `name`, `min`, `max`, and `step` properties. All sliders must be answered before proceeding.
 
+### Textfield for Post-editing/Translation
+
+Enable a textfield for post-editing or translation tasks using the `textfield` parameter in `info`. The textfield content is stored in annotations alongside scores and error spans.
+
+```python
+{
+  "info": {
+    "protocol": "DA",
+    "textfield": "prefilled"  # Options: null, "hidden", "visible", "prefilled"
+  }
+}
+```
+
+**Textfield modes:**
+- `null` or omitted: No textfield (default)
+- `"hidden"`: Textfield hidden by default, shown by clicking a button
+- `"visible"`: Textfield always visible
+- `"prefilled"`: Textfield visible and pre-filled with model output for post-editing
+
 ### Custom Instructions
 
 Set campaign-level instructions using the `instructions` field in `info` (supports HTML).
@@ -380,7 +414,7 @@ Customize the goodbye message shown to users when they complete all annotations
   - **Score**: Numeric quality rating (0-100)
   - **Error Spans**: Text highlights marking errors with severity (`minor`, `major`)
   - **Error Categories**: MQM taxonomy labels for errors
-- **Template**: The annotation interface type. The `basic` template supports comparing multiple outputs simultaneously.
+- **Template**: The annotation interface type. The `annotate` template supports comparing multiple outputs simultaneously.
 - **Assignment**: The method for distributing items to users:
   - **Task-based**: Each user has predefined items
   - **Single-stream**: Users draw from a shared pool with random assignment
@@ -411,7 +445,7 @@ pearmut run
 2. Add build rule to `webpack.config.js`
 3. Reference as `info->template` in campaign JSON
 
-See [web/src/basic.ts](web/src/basic.ts) for example.
+See [web/src/annotate.ts](web/src/annotate.ts) for example.
 
 ### Deployment
 
@@ -422,68 +456,15 @@ Run on public server or tunnel local port to public IP/domain and run locally.
 If you use this work in your paper, please cite as following.
 ```bibtex
 @misc{zouhar2026pearmut,
-  author = {Zouhar, Vilém},
-  title = {Pearmut: Human Evaluation of Translation Made Trivial},
-  year = {2026}
+      title={Pearmut: Human Evaluation of Translation Made Trivial}, 
+      author={Vilém Zouhar and Tom Kocmi},
+      year={2026},
+      eprint={2601.02933},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2601.02933}, 
 }
 ```
 
 Contributions are welcome! Please reach out to [Vilém Zouhar](mailto:vilem.zouhar@gmail.com).
-
-# Changelog
-
-- v1.0.1
-  - Support RTL languages
-  - Add boxes for references
-  - Add custom score sliders for multi-dimensional evaluation
-  - Make instructions customizable and protocol-dependent
-  - Support custom sliders
-  - Purge/reset whole tasks from dashboard
-  - Fix resetting individual users in single-stream/dynamic
-  - Fix notification stacking
-  - Add campaigns from dashboard
-- v0.3.3
-  - Rename `doc_id` to `item_id`
-  - Add Typst, LaTeX, and PDF export for model ranking tables. Hide them by default.
-  - Add dynamic assignment type with contrastive model comparison
-  - Add `instructions_goodbye` field with variable substitution
-  - Add visual anchors at 33% and 66% on sliders
-  - Add German→English ESA tutorial with attention checks
-  - Validate document model consistency before shuffle
-  - Fix UI block on any interaction
-- v0.3.2
-  - Revert seeding of user IDs
-  - Set ESA (Error Span Annotation) as default
-  - Update server IP address configuration
-  - Show approximate alignment by default
-  - Unify pointwise and listwise interfaces into `basic`
-  - Refactor protocol configuration (breaking change)
-- v0.2.11
-  - Add comment field in settings panel
-  - Add `score_gt` validation for listwise comparisons
-  - Add Content-Disposition headers for proper download filenames
-  - Add model results display to dashboard with rankings
-  - Add campaign file structure validation
-  - Purge command now unlinks assets
-- v0.2.6
-  - Add frozen annotation links feature for view-only mode
-  - Add word-level annotation mode toggle for error spans
-  - Add `[missing]` token support
-  - Improve frontend speed and cleanup toolboxes on item load
-  - Host assets via symlinks
-  - Add validation threshold for success/fail tokens
-  - Implement reset masking for annotations
-  - Allow pre-defined user IDs and tokens in campaign data
-- v0.1.1
-  - Set server defaults and add VM launch scripts
-  - Add warning dialog when navigating away with unsaved work
-  - Add tutorial validation support for pointwise and listwise
-  - Add ability to preview existing annotations via progress bar
-  - Add support for ESA<sup>AI</sup> pre-filled error_spans
-  - Rename pairwise to listwise and update layout
-  - Implement single-stream assignment type
-- v0.0.3
-  - Support multimodal inputs and outputs
-  - Add dashboard
-  - Implement ESA (Error Span Annotation) and MQM support
-
+See changes in [CHANGELOG.md](CHANGELOG.md).

{pearmut-1.0.1 → pearmut-1.0.2}/pearmut.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pearmut
-Version: 1.0.1
+Version: 1.0.2
 Summary: A tool for evaluation of model outputs, primarily MT.
 Author-email: Vilém Zouhar <vilem.zouhar@gmail.com>
 License: MIT
@@ -19,7 +19,7 @@ Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Dynamic: license-file
 
-# 🍐Pearmut &nbsp; &nbsp; [![PyPi version](https://badgen.net/pypi/v/pearmut/)](https://pypi.org/project/pearmut) [![PyPI download/month](https://img.shields.io/pypi/dm/pearmut.svg)](https://pypi.python.org/pypi/pearmut/) [![PyPi license](https://badgen.net/pypi/license/pearmut/)](https://pypi.org/project/pearmut/) [![build status](https://github.com/zouharvi/pearmut/actions/workflows/test.yml/badge.svg)](https://github.com/zouharvi/pearmut/actions/workflows/test.yml)
+# 🍐Pearmut <br> [![PyPi version](https://badgen.net/pypi/v/pearmut/)](https://pypi.org/project/pearmut) [![PyPI download/month](https://img.shields.io/pypi/dm/pearmut.svg)](https://pypi.python.org/pypi/pearmut/) [![PyPi license](https://badgen.net/pypi/license/pearmut/)](https://pypi.org/project/pearmut/) [![build status](https://github.com/zouharvi/pearmut/actions/workflows/test.yml/badge.svg)](https://github.com/zouharvi/pearmut/actions/workflows/test.yml) [![arXiv](https://img.shields.io/badge/arXiv-2601.02933-b31b1b.svg?style=flat)](https://arxiv.org/abs/2601.02933)
 
 **Platform for Evaluation and Reviewing of Multilingual Tasks**: Evaluate model outputs for translation and NLP tasks with support for multimodal data (text, video, audio, images) and multiple annotation protocols ([DA](https://aclanthology.org/N15-1124/), [ESA](https://aclanthology.org/2024.wmt-1.131/), [ESA<sup>AI</sup>](https://aclanthology.org/2025.naacl-long.255/), [MQM](https://doi.org/10.1162/tacl_a_00437), and more!).
 
@@ -142,6 +142,21 @@ The `shuffle` parameter in campaign `info` controls this behavior:
 }
 ```
 
+### Showing Model Names
+
+By default, model names are hidden to avoid biasing annotators. To display model names on top of each output block, set `show_model_names` to `true`:
+```python
+{
+  "info": {
+    "assignment": "task-based",
+    "protocol": "ESA",
+    "show_model_names": true  # Default: false.
+  },
+  "campaign_id": "my_campaign",
+  "data": [...]
+}
+```
+
 ### Custom Score Sliders
 
 For multi-dimensional evaluation tasks (e.g., assessing fluency on a Likert scale), you can define custom sliders with specific ranges and steps:
@@ -163,6 +178,25 @@ For multi-dimensional evaluation tasks (e.g., assessing fluency on a Likert scal
 
 When `sliders` is specified, only the custom sliders are shown. Each slider must have `name`, `min`, `max`, and `step` properties. All sliders must be answered before proceeding.
 
+### Textfield for Post-editing/Translation
+
+Enable a textfield for post-editing or translation tasks using the `textfield` parameter in `info`. The textfield content is stored in annotations alongside scores and error spans.
+
+```python
+{
+  "info": {
+    "protocol": "DA",
+    "textfield": "prefilled"  # Options: null, "hidden", "visible", "prefilled"
+  }
+}
+```
+
+**Textfield modes:**
+- `null` or omitted: No textfield (default)
+- `"hidden"`: Textfield hidden by default, shown by clicking a button
+- `"visible"`: Textfield always visible
+- `"prefilled"`: Textfield visible and pre-filled with model output for post-editing
+
 ### Custom Instructions
 
 Set campaign-level instructions using the `instructions` field in `info` (supports HTML).
@@ -401,7 +435,7 @@ Customize the goodbye message shown to users when they complete all annotations
   - **Score**: Numeric quality rating (0-100)
   - **Error Spans**: Text highlights marking errors with severity (`minor`, `major`)
   - **Error Categories**: MQM taxonomy labels for errors
-- **Template**: The annotation interface type. The `basic` template supports comparing multiple outputs simultaneously.
+- **Template**: The annotation interface type. The `annotate` template supports comparing multiple outputs simultaneously.
 - **Assignment**: The method for distributing items to users:
   - **Task-based**: Each user has predefined items
   - **Single-stream**: Users draw from a shared pool with random assignment
@@ -432,7 +466,7 @@ pearmut run
 2. Add build rule to `webpack.config.js`
 3. Reference as `info->template` in campaign JSON
 
-See [web/src/basic.ts](web/src/basic.ts) for example.
+See [web/src/annotate.ts](web/src/annotate.ts) for example.
 
 ### Deployment
 
@@ -443,68 +477,15 @@ Run on public server or tunnel local port to public IP/domain and run locally.
 If you use this work in your paper, please cite as following.
 ```bibtex
 @misc{zouhar2026pearmut,
-  author = {Zouhar, Vilém},
-  title = {Pearmut: Human Evaluation of Translation Made Trivial},
-  year = {2026}
+      title={Pearmut: Human Evaluation of Translation Made Trivial}, 
+      author={Vilém Zouhar and Tom Kocmi},
+      year={2026},
+      eprint={2601.02933},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2601.02933}, 
 }
 ```
 
 Contributions are welcome! Please reach out to [Vilém Zouhar](mailto:vilem.zouhar@gmail.com).
-
-# Changelog
-
-- v1.0.1
-  - Support RTL languages
-  - Add boxes for references
-  - Add custom score sliders for multi-dimensional evaluation
-  - Make instructions customizable and protocol-dependent
-  - Support custom sliders
-  - Purge/reset whole tasks from dashboard
-  - Fix resetting individual users in single-stream/dynamic
-  - Fix notification stacking
-  - Add campaigns from dashboard
-- v0.3.3
-  - Rename `doc_id` to `item_id`
-  - Add Typst, LaTeX, and PDF export for model ranking tables. Hide them by default.
-  - Add dynamic assignment type with contrastive model comparison
-  - Add `instructions_goodbye` field with variable substitution
-  - Add visual anchors at 33% and 66% on sliders
-  - Add German→English ESA tutorial with attention checks
-  - Validate document model consistency before shuffle
-  - Fix UI block on any interaction
-- v0.3.2
-  - Revert seeding of user IDs
-  - Set ESA (Error Span Annotation) as default
-  - Update server IP address configuration
-  - Show approximate alignment by default
-  - Unify pointwise and listwise interfaces into `basic`
-  - Refactor protocol configuration (breaking change)
-- v0.2.11
-  - Add comment field in settings panel
-  - Add `score_gt` validation for listwise comparisons
-  - Add Content-Disposition headers for proper download filenames
-  - Add model results display to dashboard with rankings
-  - Add campaign file structure validation
-  - Purge command now unlinks assets
-- v0.2.6
-  - Add frozen annotation links feature for view-only mode
-  - Add word-level annotation mode toggle for error spans
-  - Add `[missing]` token support
-  - Improve frontend speed and cleanup toolboxes on item load
-  - Host assets via symlinks
-  - Add validation threshold for success/fail tokens
-  - Implement reset masking for annotations
-  - Allow pre-defined user IDs and tokens in campaign data
-- v0.1.1
-  - Set server defaults and add VM launch scripts
-  - Add warning dialog when navigating away with unsaved work
-  - Add tutorial validation support for pointwise and listwise
-  - Add ability to preview existing annotations via progress bar
-  - Add support for ESA<sup>AI</sup> pre-filled error_spans
-  - Rename pairwise to listwise and update layout
-  - Implement single-stream assignment type
-- v0.0.3
-  - Support multimodal inputs and outputs
-  - Add dashboard
-  - Implement ESA (Error Span Annotation) and MQM support
-
+See changes in [CHANGELOG.md](CHANGELOG.md).

{pearmut-1.0.1 → pearmut-1.0.2}/pearmut.egg-info/SOURCES.txt

@@ -13,8 +13,8 @@ server/cli.py
 server/constants.py
 server/results_export.py
 server/utils.py
-server/static/basic.bundle.js
-server/static/basic.html
+server/static/annotate.bundle.js
+server/static/annotate.html
 server/static/dashboard.bundle.js
 server/static/dashboard.html
 server/static/favicon.svg

{pearmut-1.0.1 → pearmut-1.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pearmut"
-version = "1.0.1"
+version = "1.0.2"
 description = "A tool for evaluation of model outputs, primarily MT."
 readme = "README.md"
 license = { text = "MIT" }

{pearmut-1.0.1 → pearmut-1.0.2}/server/app.py

@@ -4,7 +4,7 @@ from typing import Any
 
 from fastapi import FastAPI, Query
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse, Response
+from fastapi.responses import FileResponse, JSONResponse, Response
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 
@@ -17,6 +17,7 @@ from .results_export import (
 )
 from .utils import (
     ROOT,
+    TOKEN_MAIN,
     check_validation_threshold,
     load_progress_data,
     save_db_payload,
@@ -192,7 +193,11 @@ async def _dashboard_data(request: DashboardDataRequest):
         progress_new[user_id] = entry
 
     return JSONResponse(
-        content={"data": progress_new, "validation_threshold": validation_threshold},
+        content={
+            "data": progress_new,
+            "validation_threshold": validation_threshold,
+            "assignment": assignment,
+        },
         status_code=200,
     )
 
@@ -288,7 +293,7 @@ class PurgeCampaignRequest(BaseModel):
 @app.post("/purge-campaign")
 async def _purge_campaign(request: PurgeCampaignRequest):
     global progress_data, tasks_data
-    
+
     campaign_id = request.campaign_id
     token = request.token
 
@@ -298,57 +303,69 @@ async def _purge_campaign(request: PurgeCampaignRequest):
         return JSONResponse(content="Invalid token", status_code=400)
 
     # Unlink assets if they exist
-    destination = tasks_data[campaign_id].get("info", {}).get("assets", {}).get("destination")
+    destination = (
+        tasks_data[campaign_id].get("info", {}).get("assets", {}).get("destination")
+    )
     if destination:
         symlink_path = f"{ROOT}/data/{destination}".rstrip("/")
         if os.path.islink(symlink_path):
             os.remove(symlink_path)
-    
+
     # Remove task file
     task_file = f"{ROOT}/data/tasks/{campaign_id}.json"
     if os.path.exists(task_file):
         os.remove(task_file)
-    
+
     # Remove output file
     output_file = f"{ROOT}/data/outputs/{campaign_id}.jsonl"
     if os.path.exists(output_file):
         os.remove(output_file)
-    
+
     # Remove from in-memory data structures
     del tasks_data[campaign_id]
     del progress_data[campaign_id]
-    
+
     # Save updated progress data
     save_progress_data(progress_data)
-    
+
     return JSONResponse(content="ok", status_code=200)
 
 
 class AddCampaignRequest(BaseModel):
     campaign_data: dict[str, Any]
+    token_main: str
 
 
 @app.post("/add-campaign")
 async def _add_campaign(request: AddCampaignRequest):
     global progress_data, tasks_data
-    
+
     from .cli import _add_single_campaign
-    
+
+    if request.token_main != TOKEN_MAIN:
+        return JSONResponse(
+            content={"error": "Invalid main token. Use the latest one."},
+            status_code=400,
+        )
+
     try:
         server = f"{os.environ.get('PEARMUT_SERVER_URL', 'http://localhost:8001')}"
         _add_single_campaign(request.campaign_data, overwrite=False, server=server)
-        
-        campaign_id = request.campaign_data['campaign_id']
+
+        campaign_id = request.campaign_data["campaign_id"]
         with open(f"{ROOT}/data/tasks/{campaign_id}.json", "r") as f:
             tasks_data[campaign_id] = json.load(f)
-        
+
         progress_data = load_progress_data(warn=None)
-        
-        return JSONResponse(content={
-            "status": "ok",
-            "campaign_id": campaign_id,
-            "token": tasks_data[campaign_id]["token"]
-        }, status_code=200)
+
+        return JSONResponse(
+            content={
+                "status": "ok",
+                "campaign_id": campaign_id,
+                "token": tasks_data[campaign_id]["token"],
+            },
+            status_code=200,
+        )
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=400)
 
@@ -418,6 +435,17 @@ if not os.path.exists(static_dir + "index.html"):
         "Static directory not found. Please build the frontend first."
     )
 
+# Serve HTML files directly without redirect
+@app.get("/annotate")
+async def serve_annotate():
+    return FileResponse(static_dir + "annotate.html")
+
+
+@app.get("/dashboard")
+async def serve_dashboard():
+    return FileResponse(static_dir + "dashboard.html")
+
+
 # Mount user assets from data/assets/
 assets_dir = f"{ROOT}/data/assets"
 os.makedirs(assets_dir, exist_ok=True)