doccano · Hironsan · Mar 27, 2023 · Mar 27, 2023
diff --git a/doccano_mini/examples.py b/doccano_mini/examples.py
diff --git a/doccano_mini/examples/question_answering.json b/doccano_mini/examples/question_answering.json
@@ -0,0 +1,7 @@
+[
+    {
+        "context": "Google was founded by computer scientists Larry Page and Sergey Brin.",
+        "question": "Who founded Google?",
+        "answer": "Larry Page and Sergey Brin"
+    }
+]
diff --git a/doccano_mini/examples/task_free.json b/doccano_mini/examples/task_free.json
@@ -0,0 +1 @@
+[{"Column 1": "", "Column 2": ""}]
diff --git a/doccano_mini/examples/text_classification.json b/doccano_mini/examples/text_classification.json
@@ -0,0 +1,5 @@
+[
+    {"text": "That would be awesome!", "label": "positive"},
+    {"text": "This is awful!", "label": "negative"},
+    {"text": "Today is hot day.", "label": "neutral"}
+]
diff --git a/doccano_mini/layout.py b/doccano_mini/layout.py
@@ -1,6 +1,8 @@
 from abc import ABC, abstractmethod
+from pathlib import Path
 from typing import Dict, List
 
+import pandas as pd
 import streamlit as st
 from langchain.chains import LLMChain
 from langchain.prompts.few_shot import FewShotPromptTemplate
@@ -21,6 +23,10 @@ def __init__(self, title: str) -> None:
     def columns(self) -> List[str]:
         return []
 
+    def load_examples(self, filename: str) -> pd.DataFrame:
+        filepath = Path(__file__).parent.resolve().joinpath("examples", filename)
+        return pd.read_json(filepath)
+
     @abstractmethod
     def make_examples(self, columns: List[str]) -> List[Dict]:
         raise NotImplementedError()

diff --git a/doccano_mini/pages/01_Text_Classification.py b/doccano_mini/pages/01_Text_Classification.py
@@ -2,14 +2,13 @@
 
 import streamlit as st
 
-from doccano_mini.examples import make_classification_example
 from doccano_mini.layout import BasePage
 from doccano_mini.prompts import make_classification_prompt
 
 
 class TextClassificationPage(BasePage):
     def make_examples(self, columns: List[str]):
-        df = make_classification_example()
+        df = self.load_examples("text_classification.json")
         edited_df = st.experimental_data_editor(df, num_rows="dynamic", width=1000)
         examples = edited_df.to_dict(orient="records")
         return examples

diff --git a/doccano_mini/pages/02_Question_Answering.py b/doccano_mini/pages/02_Question_Answering.py
@@ -2,14 +2,13 @@
 
 import streamlit as st
 
-from doccano_mini.examples import make_question_answering_example
 from doccano_mini.layout import BasePage
 from doccano_mini.prompts import make_question_answering_prompt
 
 
 class QuestionAnsweringPage(BasePage):
     def make_examples(self, columns: List[str]):
-        df = make_question_answering_example()
+        df = self.load_examples("question_answering.json")
         edited_df = st.experimental_data_editor(df, num_rows="dynamic", width=1000)
         examples = edited_df.to_dict(orient="records")
         return examples

diff --git a/doccano_mini/pages/09_Task_Free.py b/doccano_mini/pages/09_Task_Free.py
@@ -2,7 +2,6 @@
 
 import streamlit as st
 
-from doccano_mini.examples import make_task_free_example
 from doccano_mini.layout import BasePage
 from doccano_mini.prompts import make_task_free_prompt
 
@@ -15,7 +14,7 @@ def columns(self) -> List[str]:
         return columns
 
     def make_examples(self, columns: List[str]):
-        df = make_task_free_example()
+        df = self.load_examples("task_free.json")
         df = df.reindex(columns, axis="columns", fill_value="")
         edited_df = st.experimental_data_editor(df, num_rows="dynamic", width=1000)
         examples = edited_df.to_dict(orient="records")