Update app.py

app.py

@@ -42,7 +42,13 @@ def retrieve_and_rerank_faiss(job_duties, model_name="BAAI/bge-m3", top_k=4):
     query_embedding = np.array(query_embedding).astype("float32")
     distances, indices = index.search(query_embedding, top_k)
     results = [
-        (isco_taxonomy[str(idx)]["ESCO_DESCRIPTION"], distances[0][i])
+        {
+            "ISCO_CODE_4": isco_taxonomy[str(idx)]["ISCO_CODE_4"],
+            "ISCO_LABEL_4": isco_taxonomy[str(idx)]["ISCO_LABEL_4"],
+            "ESCO_OCCUPATION": isco_taxonomy[str(idx)]["ESCO_OCCUPATION"],
+            "ESCO_DESCRIPTION": isco_taxonomy[str(idx)]["ESCO_DESCRIPTION"],
+            "Similarity": distances[0][i]
+        }
         for i, idx in enumerate(indices[0])
     ]
     return results
@@ -50,7 +56,7 @@ def retrieve_and_rerank_faiss(job_duties, model_name="BAAI/bge-m3", top_k=4):
 # Gradio Interface
 def gradio_interface(job_duties):
     results = retrieve_and_rerank_faiss(job_duties)
-    return [f"Description: {desc}, Distance: {dist}" for desc, dist in results]
+    return results
 
-iface = gr.Interface(fn=gradio_interface, inputs="text", outputs="text", title="Job Duties to ISCO Descriptions")
+iface = gr.Interface(fn=gradio_interface, inputs="text", outputs=gr.outputs.Dataframe(type="pandas"), title="Semantic similarity matches with ESCO descriptions")
 iface.launch()