Add query examples

app.py

@@ -21,8 +21,10 @@ RERANKER_MODEL_NAME = "douglasfaisal/granularity-legal-reranker-cross-encoder-in
 es_retriever_client = ESRetriever(ES_HOST, ES_INDEX_NAME, ES_USERNAME, ES_PASSWORD)
 cross_enc_reranker = CrossEncReranker(RERANKER_MODEL_NAME, 512)
 
-def retrieve_and_rerank(question: str):
+def retrieve_and_rerank(question: str, example: str):
 
+    if (question == None or question == ""):
+        question = example
     query = question_to_statement(question)
 
     try:
@@ -33,8 +35,23 @@ def retrieve_and_rerank(question: str):
     except:
         return "-", "(Result Not Found)"
 
-
-demo = gr.Interface(fn=retrieve_and_rerank, inputs="text", outputs=["label", "text"])
+with gr.Blocks() as demo:
+    with gr.Row():
+        text_input = gr.Textbox()
+
+demo = gr.Interface(
+    fn=retrieve_and_rerank,
+    inputs=[
+        "text",
+        gr.Dropdown(
+            [
+                "Apa yang dimaksud dengan pemberi kerja?",
+                "Berapa paling lama waktu kerja lembur?",
+                "Apa bentuk pendapatan non-upah?"
+            ]
+        )
+    ],
+    outputs=["label", "text"])
 
 if __name__ == "__main__":
     demo.launch()