more UI changes

app.py

@@ -81,14 +81,16 @@ def retrieve_relevant_text(input_text):
     
     # Extract code and similarity scores
     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
+    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)
 
 # Evaluate retrieval efficiency on the validation/test set
 def evaluate_efficiency(dataset_split):
@@ -125,11 +127,7 @@ precision, recall, f1 = evaluate_efficiency(dataset['validation'])
 # Gradio interface
 def gradio_interface(input_text):
     results = retrieve_relevant_text(input_text)
-    formatted_results = [
-        f"Similarity Score: {result['similarity_score']:.2f}, Code: {result['code']}, Description: {result['description']}"
-        for result in results
-    ]
-    return formatted_results
+    return results
 
 # Display retrieval efficiency metrics
 metrics = f"Precision: {precision:.2f}, Recall: {recall:.2f}, F1 Score: {f1:.2f}"
@@ -137,9 +135,12 @@ 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)
-    text_input = gr.Textbox(label="Input Text")
-    text_output = gr.Textbox(label="Retrieved Results")
-    submit_button = gr.Button("Submit")
+    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")
     submit_button.click(fn=gradio_interface, inputs=text_input, outputs=text_output)
 
 interface.launch()