Add reference label

app.py

@@ -27,10 +27,10 @@ def retrieve_and_rerank(question: str):
     retrieval_results = es_retriever_client.retrieve(query)
     reranker_results = cross_enc_reranker.rerank(query, retrieval_results)
 
-    return reranker_results[0].text
+    return reranker_results[0].generate_string(), reranker_results[0].text
 
 
-demo = gr.Interface(fn=retrieve_and_rerank, inputs="text", outputs="text")
+demo = gr.Interface(fn=retrieve_and_rerank, inputs="text", outputs=["label", "text"])
 
 # Press the green button in the gutter to run the script.
 demo.launch()

models/law_component.py

@@ -2,6 +2,7 @@ import copy
 
 LAW_PATTERN = '(?P<law_type>[A-Za-z]+) (?P<law_number>\d{1,})/(?P<law_year>\d{4})'
 
+
 class LawComponent:
     law_type: str = None
     law_year: int = 0
@@ -151,3 +152,18 @@ class LawComponent:
 
     def copy(self):
         return copy.deepcopy(self)
+
+    def generate_string(self):
+        res = "{} {}/{}".format(self.law_type.upper(), self.law_number, self.law_year)
+        if self.component_type == 'chapter':
+            return res + " Bab {}".format(self.chapter)
+        else:
+            res += " Pasal {}".format(self.article)
+            if self.subsection != None and self.subsection != 0:
+                res += " Ayat {}".format(self.subsection)
+            if self.letter != None and self.subsection != 0:
+                if self.article != 1 and self.article != "1":
+                    res += " Huruf {}".format(self.letter)
+                else:
+                    res += " Angka {}".format(self.letter)
+            return res

requirements.txt

@@ -1,4 +1,4 @@
 gradio~=3.28.1
-numpy==1.21.4
+numpy>1.21.4
 requests==2.26.0
 sentence-transformers