Update app.py

app.py

@@ -320,28 +320,27 @@ Expected JSON format:
 
 # Gradio interface function
 def process_pdf(pdf):
-    # text = read_pdf(pdf.path)  # Extract text from PDF
-    # output = generate(text)  # Generate structured JSON
+    text = read_pdf(pdf)  # Extract text from PDF
+    output = generate(text)  # Generate structured JSON
 
-    # return output
-    return f"name {pdf.name} and  path {pdf.path}"
+    return output
 
-# def show_to_UI(pdf):
-#     # output = process_pdf(pdf)  # Call process_pdf to get JSON
-#     output= {"metadata": {"name": "John Doe", "age": 30}, "lab_tests": [{"test": "Blood Sugar", "value": "120 mg/dL"}]}
+def show_to_UI(pdf):
+    output = process_pdf(pdf)  # Call process_pdf to get JSON
+    # output= {"metadata": {"name": "John Doe", "age": 30}, "lab_tests": [{"test": "Blood Sugar", "value": "120 mg/dL"}]}
 
-#     # Extract metadata
-#     metadata = output["metadata"]
-#     labtests = pd.DataFrame(output["lab_tests"])
-#     # reds = pd.DataFrame(output["reds"])
+    # Extract metadata
+    metadata = output["metadata"]
+    labtests = pd.DataFrame(output["lab_tests"])
+    reds = pd.DataFrame(output["reds"])
 
-#     metadata_str = f"**Patient Name:** {metadata['patient_name']}\n\n" \
-#                    f"**Age:** {metadata['age']}\n\n" \
-#                    # f"**Gender:** {metadata['gender']}\n\n" \
-#                    # f"**Lab Name:** {metadata['lab_name']}\n\n" \
-#                    # f"**Report Date:** {metadata['report_date']}"
+    metadata_str = f"**Patient Name:** {metadata['patient_name']}\n\n" \
+                   f"**Age:** {metadata['age']}\n\n" \
+                   f"**Gender:** {metadata['gender']}\n\n" \
+                   f"**Lab Name:** {metadata['lab_name']}\n\n" \
+                   f"**Report Date:** {metadata['report_date']}"
 
-#     return metadata_str, labtests  
+    return metadata_str, labtests, reds, output  
 
 # Define Gradio interface
 with gr.Blocks() as demo:
@@ -351,11 +350,11 @@ with gr.Blocks() as demo:
         pdf_input = gr.File(label="Upload PDF Report")
         submit_btn = gr.Button("Process")
 
-    # metadata_output = gr.Markdown(label="Metadata")
-    # lab_test_output = gr.Dataframe(label="Lab Test Results")
-    # output_display = gr.JSON(label="Extracted Report")  # Show JSON output
-    output_display = gr.Text(label="Extracted Report")  # Show JSON output
+    metadata_output = gr.Markdown("**Patient Name: Prashasst...**")
+    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(process_pdf, inputs=[pdf_input], outputs=[output_display])
+    submit_btn.click(process_pdf, inputs=[pdf_input], outputs=[metadata_output, lab_test_output, reds_output,output_JSON])
 
 demo.launch()