Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import pandas as pd
-import json
 import fitz  # PyMuPDF
 import pytesseract
 from pdf2image import convert_from_path
@@ -333,8 +332,26 @@ def show_to_UI(pdf):
 
     # Extract metadata
     metadata = output["metadata"]
-    labtests = pd.DataFrame(output["lab_tests"])
-    reds = pd.DataFrame(output["reds"])
+    # labtests = pd.DataFrame(output["lab_tests"])
+    # reds = pd.DataFrame(output["reds"])
+
+    try:
+      labtests = pd.DataFrame(output["lab_tests"],)
+    except Exception as e:
+      print(f"Error creating lab tests DataFrame: {e}")
+      labtests = pd.DataFrame()  # Return empty DataFrame
+    
+    try:
+      highs = pd.DataFrame(output["reds"]["high"],index=True)
+    except Exception as e:
+      print(f"Error creating highs DataFrame: {e}")
+      highs = pd.DataFrame()  # Return empty DataFrame
+    
+    try:
+      lows = pd.DataFrame(output["reds"]["low"],)
+    except Exception as e:
+      print(f"Error creating lowss DataFrame: {e}")
+      lows = pd.DataFrame()  # Return empty DataFrame
 
     metadata_str = f"**Patient Name:** {metadata['patient_name']}\n\n" \
                    f"**Age:** {metadata['age']}\n\n" \
@@ -342,7 +359,9 @@ def show_to_UI(pdf):
                    f"**Lab Name:** {metadata['lab_name']}\n\n" \
                    f"**Report Date:** {metadata['report_date']}"
 
-    return metadata_str, labtests, reds, output  
+    
+
+    return metadata_str,highs,lows, labtests,output  
     
 
 
@@ -356,9 +375,11 @@ with gr.Blocks() as demo:
         submit_btn = gr.Button("Process")
 
     metadata_output = gr.Markdown("**Patient Name: Prashasst...**")
+    with gr.Row():
+      high_output = gr.Dataframe(label="High Values")
+      low_output = gr.Dataframe(label="Low Values")
     lab_test_output = gr.Dataframe(label="Lab Test Results")
-    reds_output = gr.Dataframe(label="Reds")
     output_JSON = gr.JSON(label="Extracted Report")  # Show JSON output
 
-    submit_btn.click(show_to_UI, inputs=[pdf_input], outputs=[metadata_output, lab_test_output, reds_output,output_JSON])
-demo.launch()
+    submit_btn.click(show_to_UI, inputs=[pdf_input], outputs=[metadata_output, high_output, low_output,lab_test_output, output_JSON])
+demo.launch(debug=True,share=True)