Update app.py

app.py

@@ -4,6 +4,12 @@ from helper_functions import *
 from rank_bm25 import BM25L
 import time
 import pprint
+
+def print_results(results):
+    result_string = ''
+    for hit in results:
+        result_string += pprint.pformat(hit, indent=4) + "\n"
+    return result_string.strip()
         
 def predict(query):
     start_time = time.time()
@@ -15,7 +21,6 @@ def predict(query):
     
     # Base URL for the search API
     base_url = "https://api.omaline.dev/search/product/search"
-    results = {}
     
     # Construct query string for API request
     query_string = "&".join([f"k={item}" for item in normalized_query_list])
@@ -98,7 +103,7 @@ def predict(query):
             semantic_scores = semantic_search(query, doc_embeddings)
             hybrid_scores = hybrid_search(keyword_scores, semantic_scores)
             is_cheapest(query, request_json)
-            results[query] = rerank_results(request_json, hybrid_scores)
+            results = rerank_results(request_json, hybrid_scores)
     
         process_end_time = time.time()
         process_time_taken = process_end_time - process_time
@@ -110,7 +115,7 @@ def predict(query):
         #         "calculate_interrelations_time": calculate_interrelations_time,
         #         "process_time": process_time_taken}
         
-        return pprint.pformat(results, indent=4)
+        return print_results(results)
     
     except Exception as e:
         error_message = f"An error occurred during processing: {e}"