Update app.py

app.py

@@ -12,12 +12,6 @@ import pprint
 # Initialize the translator
 translator = Translator()
 
-def print_results(hits):
-    results = ""
-    for hit in hits:
-        results += pprint.pformat(hit, indent=4) + '\n'
-    return results
-
 async def translate_bulk(bulk: list) -> list:
     """
     Translate the given text to English and return the translated text.
@@ -159,12 +153,15 @@ async def predict(query):
         process_end_time = time.time()
         process_time_taken = process_end_time - process_time
         time_taken = time.time() - start_time
-        return {"results": results, "time_taken": time_taken, 
+
+        hits = {"results": results, "time_taken": time_taken, 
                 "request_time": request_time, "normalization_time": normalization_time,
                 "translation_time": translation_time, "categorize_time": categorize_time, 
                 "tokenization_time": tokenization_time, "encode_time": encode_time, 
                 "calculate_interrelations_time": calculate_interrelations_time,
                 "process_time": process_time_taken}
+        
+        return pprint.pformat(hits, indent=4)
     
     except Exception as e:
         error_message = f"An error occurred during processing: {e}"