app.py updated

app.py

@@ -14,7 +14,7 @@ from utils import (
     get_clinical_trials_related_to_diseases,
     get_clinical_records_by_ids
 )
-from llm_res import get_short_summary_out_of_json_files
+from llm_res import get_short_summary_out_of_json_files, tagging_insights_from_json
 import json
 import numpy as np
 from sentence_transformers import SentenceTransformer
@@ -82,10 +82,13 @@ with st.container():
             # 7. Use an LLM to get a summary of the clinical trials, in plain text format.
             status.write("Getting a summary of the clinical trials...")
             response = get_short_summary_out_of_json_files(json_of_clinical_trials)
-            print(f'Response from LLM: {response}')
-            status.write(f'Response from LLM: {response}')
+            print(f'Response from LLM summarization: {response}')
+            status.write(f'Response from LLM summarization: {response}')
             # 8. Use an LLM to extract numerical data from the clinical trials (e.g. number of patients, number of deaths, etc.). Get summary statistics out of that.
             status.write("Getting summary statistics of the clinical trials...")
+            response = tagging_insights_from_json(json_of_clinical_trials)
+            print(f'Response from LLM tagging: {response}')
+            status.write(f'Response from LLM tagging: {response}')
             # 9. Show the results to the user: graph of the diseases chosen, summary of the clinical trials, summary statistics of the clinical trials, and list of the details of the clinical trials considered
             status.update(label="Done!", state="complete")
             time.sleep(1)