Update app.py

app.py

@@ -3,20 +3,21 @@ import pandas as pd
 import requests
 import json
 from datetime import datetime
+import PyPDF2
 
 st.title("FHIR Converter")
 
 # Dropdown for conversion type
 conversion_type = st.selectbox(
     "Select Conversion Type",
-    ["CSV to FHIR", "CCDA to FHIR","Clinical Notes to FHIR"]
+    ["CSV to FHIR", "CCDA to FHIR", "Clinical Notes to FHIR"]
 )
 
 # Dropdown for resource type
 if conversion_type == "CSV to FHIR":
     res_type = st.selectbox(
         "Select Resource Type",
-        ["Patient","CarePlan", "MedicationRequest", "Immunization", "Encounter", "Claim", "Procedure", "Observation"]
+        ["Patient", "CarePlan", "MedicationRequest", "Immunization", "Encounter", "Claim", "Procedure", "Observation"]
     )
 
     # File uploader for CSV file
@@ -84,7 +85,7 @@ if conversion_type == "CSV to FHIR":
 elif conversion_type == "CCDA to FHIR":
     res_type = st.selectbox(
         "Select Resource Type",
-        ["Patient", "Encounter", "Observation","MedicationRequest","Procedure","Immunization","Condition"]
+        ["Patient", "Encounter", "Observation", "MedicationRequest", "Procedure", "Immunization", "Condition"]
     )
 
     cda_content = st.text_area("Enter CCDA Content in XML format")
@@ -111,28 +112,51 @@ elif conversion_type == "CCDA to FHIR":
             st.error(f"An error occurred: {str(e)}")
     if conversion_successful:
         st.markdown("[Go to FHIR Validator](https://validator.fhir.org)", unsafe_allow_html=True)
-elif conversion_type == "Clinical Notes to FHIR":
 
-    note_content = st.text_area("Copy and Paste your clinical notes for a patient")
+elif conversion_type == "Clinical Notes to FHIR":
+    uploaded_pdf = st.file_uploader("Upload PDF with Clinical Notes", type=["pdf"])
 
     # Variables to store the result and state
     result = st.empty()
     conversion_successful = False
+    progress_bar = st.progress(0)
+    time_placeholder = st.empty()
 
-    # Button to trigger conversion
-    if st.button("Convert"):
+    if uploaded_pdf is not None and st.button("Convert"):
         try:
-            data = {"note_content": note_content}
+            # Extract text from PDF
+            pdf_reader = PyPDF2.PdfFileReader(uploaded_pdf)
+            paragraphs = []
+            for page_num in range(pdf_reader.getNumPages()):
+                page = pdf_reader.getPage(page_num)
+                text = page.extract_text()
+                paragraphs.extend(text.split("\n\n"))  # Split by double newline for paragraph separation
+
+            # Send paragraphs to API
+            total_paragraphs = len(paragraphs)
+            start_time = datetime.now()
+            result_text = ""
+
+            for idx, paragraph in enumerate(paragraphs):
+                data = {"note_content": paragraph}
+                response = requests.post("https://fhir-api-9jsn.onrender.com/convert_notes", json=data)
+
+                if response.status_code == 200:
+                    result_text += response.json().get("output", "") + "\n"
+                else:
+                    st.error(f"An error occurred at paragraph {idx + 1}: {response.json().get('error', 'Unknown error')}")
+                    break
+
+                # Update progress
+                elapsed_time = datetime.now() - start_time
+                progress_bar.progress(int(((idx + 1) / total_paragraphs) * 100))
+                time_placeholder.text(f"Time elapsed: {elapsed_time}")
+
+            if result_text:
+                result.success("Conversion successful!")
+                result.text_area("FHIR Output", result_text, height=400)
+                conversion_successful = True
 
-            response = requests.post("https://fhir-api-9jsn.onrender.com/convert_notes", json=data)
-            if response.status_code == 200:
-                result_text = response.json().get("output", "")
-                if result_text:
-                    result.success("Conversion successful!")
-                    result.text_area("FHIR Output", result_text, height=400)
-                    conversion_successful = True
-            else:
-                st.error(f"An error occurred: {response.json().get('error', 'Unknown error')}")
         except Exception as e:
             st.error(f"An error occurred: {str(e)}")