Update app.py

app.py

@@ -67,12 +67,13 @@ class ParrotletRetriever:
 
         docs = []
         for i, match in enumerate(results["matches"]):
-            text = match["metadata"].get("text", "[No text metadata]")
+            metadata = match["metadata"]
+            text = metadata.get("text")
             docs.append({
                 "Rank": i + 1,
-                "Score": f"{match['score']:.4f}",
+                "Score": f"{match['score']:.2f}",
                 "Document": text,
-                "ID": match["concept_id"],
+                "ID": metadata.get("concept_id")
             })
 
         return docs
@@ -126,13 +127,12 @@ with gr.Blocks(title="Parrotlet-e Retrieval", theme=gr.themes.Soft()) as demo:
                 inputs=query_input,
                 label="Example Queries"
             )
-
-            top_k_slider = gr.Slider(
-                minimum=1,
-                maximum=10,
+            
+            top_k_input = gr.Number(
+                label="K",
                 value=5,
-                step=1,
-                label="Top-K Retrieved"
+                precision=0,
+                interactive=True
             )
 
             search_btn = gr.Button("retrieve", variant="primary")
@@ -148,13 +148,9 @@ with gr.Blocks(title="Parrotlet-e Retrieval", theme=gr.themes.Soft()) as demo:
 
     search_btn.click(
         fn=retrieve_documents,
-        inputs=[query_input, top_k_slider],
+        inputs=[query_input, top_k_input],
         outputs=[results_output, status_box],
     )
 
-
-# =========================
-# Run App
-# =========================
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)