revert

app.py

@@ -81,16 +81,14 @@ def retrieve_relevant_text(input_text):
     
     # Extract code and similarity scores
     output = []
-    for i, (metadata, distance) in enumerate(zip(results['metadatas'][0], results['distances'][0])):
-        output.append(
-            f"Result {i + 1}:\n"
-            f"Similarity Score: {distance:.2f}\n"
-            f"Code: {metadata['code']}\n"
-            f"Code System: {metadata['code_system']}\n"
-            f"Description: {metadata['description']}\n"
-            "-------------------"
-        )
-    return "\n".join(output)
+    for metadata, distance in zip(results['metadatas'][0], results['distances'][0]):
+        output.append({
+            "similarity_score": distance,
+            "code": metadata['code'],
+            "code_system": metadata['code_system'],
+            "description": metadata['description']
+        })
+    return output
 
 # Evaluate retrieval efficiency on the validation/test set
 def evaluate_efficiency(dataset_split):
@@ -127,7 +125,11 @@ precision, recall, f1 = evaluate_efficiency(dataset['validation'])
 # Gradio interface
 def gradio_interface(input_text):
     results = retrieve_relevant_text(input_text)
-    return results
+    formatted_results = [
+        f"Similarity Score: {result['similarity_score']:.2f}, Code: {result['code']}, Description: {result['description']}"
+        for result in results
+    ]
+    return formatted_results
 
 # Display retrieval efficiency metrics
 metrics = f"Precision: {precision:.2f}, Recall: {recall:.2f}, F1 Score: {f1:.2f}"
@@ -135,12 +137,9 @@ metrics = f"Precision: {precision:.2f}, Recall: {recall:.2f}, F1 Score: {f1:.2f}
 with gr.Blocks() as interface:
     gr.Markdown("# Text Retrieval with Efficiency Metrics")
     gr.Markdown(metrics)
-    with gr.Row():
-        with gr.Column():
-            text_input = gr.Textbox(label="Input Text")
-            submit_button = gr.Button("Submit")
-        with gr.Column():
-            text_output = gr.Textbox(label="Retrieved Results")
+    text_input = gr.Textbox(label="Input Text")
+    text_output = gr.Textbox(label="Retrieved Results")
+    submit_button = gr.Button("Submit")
     submit_button.click(fn=gradio_interface, inputs=text_input, outputs=text_output)
 
 interface.launch()