Update callbackmanager.py

callbackmanager.py

@@ -121,15 +121,14 @@ def analyze_dicom_content_ai(dicom_metadata_json):  # Copied from your code
             top_p=0.9,
         )
         the_response = response.choices[0].message.content
-        return the_response, trace_data_detail_dicom_analysis
+        return the_response, trace_data_detail_dicom_ai
 
     except Exception as e:
         error_message = f"AI Analysis Error in analyze_dicom_content_ai (DICOM Metadata): {e}"
         trace_data_detail_dicom_analysis["error"] = f"AI Analysis Error: {e}"
-        return error_message, trace_data_detail_dicom_analysis
+        return error_message, trace_data_detail_dicom_image_analysis
 
 
-# ... (Paste other AI analysis functions: analyze_hl7_file_with_ai, analyze_cda_xml_file_with_ai, analyze_pdf_file_with_ai, analyze_csv_file_with_ai here - ensure to adapt file reading for Gradio file paths if necessary) ...
 def analyze_hl7_file_with_ai(hl7_file_path):
     """Analyzes HL7 file content using Discharge Guard AI."""
     try:
@@ -293,7 +292,7 @@ def analyze_pdf_content_ai(pdf_text_content):  # Copied from your code
     ```text
     {pdf_text_content}
     ```
-        *   Remember, this analysis is for conceptual informational purposes only and **NOT medical advice.** Focus on summarizing and structuring the extracted data in a clinically relevant way based on the PDF content.
+        *   Remember, this analysis is for conceptual informational purposes only and **NOT medical advice.** Focus on deep summarization and structuring the extracted data in a clinically relevant way based on the PDF content.
     """
 
     trace_data_detail_pdf_analysis = {
@@ -387,7 +386,6 @@ def analyze_csv_content_ai(csv_content_string):  # Copied from your code
         return error_message, trace_data_detail_csv_analysis
 
 
-# ... (CallbackManager, display_form, generate_pdf_from_form, generate_pdf_from_meldrx, generate_discharge_paper_one_click, client initialization remain the same) ...
 class CallbackManager:
     def __init__(self, redirect_uri: str, client_secret: str = None):
         client_id = os.getenv("APPID")
@@ -408,71 +406,16 @@ class CallbackManager:
         if self.api.authenticate_with_code(code):
             self.access_token = self.api.access_token
             return (
-                f"<span style='color:#00FF7F;'>Authentication successful!</span> Access Token: {self.access_token[:10]}... (truncated)" # Neon Green Success
+                f"<span style='color:#00FF7F;'>Authentication successful!</span> Access Token: {self.access_token[:10]}... (truncated)"  # Neon Green Success
             )
-        return "<span style='color:#FF4500;'>Authentication failed. Please check the code.</span>" # Neon Orange Error
+        return "<span style='color:#FF4500;'>Authentication failed. Please check the code.</span>"  # Neon Orange Error
 
     def get_patient_data(self) -> str:
         """Fetch patient data from MeldRx"""
         try:
             if not self.access_token:
                 logger.warning("Not authenticated when getting patient data")
-                return "<span style='color:#FF8C00;'>Not authenticated. Please provide a valid authorization code first.</span>" # Neon Dark Orange
-
-            # For demo purposes, if there's no actual API connected, return mock data
-            # Remove this in production and use the real API call
-            if not hasattr(self.api, "get_patients") or self.api.get_patients is None:
-                logger.info("Using mock patient data (no API connection)")
-                # Return mock FHIR bundle with patient data
-                mock_data = {
-                    "resourceType": "Bundle",
-                    "type": "searchset",
-                    "total": 2,
-                    "link": [],
-                    "entry": [
-                        {
-                            "resource": {
-                                "resourceType": "Patient",
-                                "id": "patient1",
-                                "name": [
-                                    {
-                                        "use": "official",
-                                        "family": "Smith",
-                                        "given": ["John"],
-                                    }
-                                ],
-                                "gender": "male",
-                                "birthDate": "1970-01-01",
-                                "address": [
-                                    {"city": "Boston", "state": "MA", "postalCode": "02108"}
-                                ],
-                            }
-                        },
-                        {
-                            "resource": {
-                                "resourceType": "Patient",
-                                "id": "patient2",
-                                "name": [
-                                    {
-                                        "use": "official",
-                                        "family": "Johnson",
-                                        "given": ["Jane"],
-                                    }
-                                ],
-                                "gender": "female",
-                                "birthDate": "1985-05-15",
-                                "address": [
-                                    {
-                                        "city": "Cambridge",
-                                        "state": "MA",
-                                        "postalCode": "02139",
-                                    }
-                                ],
-                            }
-                        },
-                    ],
-                }
-                return json.dumps(mock_data, indent=2)
+                return "<span style='color:#FF8C00;'>Not authenticated. Please provide a valid authorization code first.</span>"  # Neon Dark Orange
 
             # Real implementation with API call
             logger.info("Calling Meldrx API to get patients")
@@ -481,19 +424,18 @@ class CallbackManager:
                 return (
                     json.dumps(patients, indent=2)
                     if patients
-                    else "<span style='color:#FFFF00;'>No patient data returned.</span>" # Neon Yellow
+                    else "<span style='color:#FFFF00;'>No patient data returned.</span>"  # Neon Yellow
                 )
-            return "<span style='color:#DC143C;'>Failed to retrieve patient data.</span>" # Crimson Error
+            return "<span style='color:#DC143C;'>Failed to retrieve patient data.</span>"  # Crimson Error
         except Exception as e:
             error_msg = f"Error in get_patient_data: {str(e)}"
             logger.error(error_msg)
-            return f"<span style='color:#FF6347;'>Error retrieving patient data: {str(e)}</span> {str(e)}" # Tomato Error
-
+            return f"<span style='color:#FF6347;'>Error retrieving patient data: {str(e)}</span> {str(e)}"  # Tomato Error
 
     def get_patient_documents(self, patient_id: str = None):
         """Fetch patient documents from MeldRx"""
         if not self.access_token:
-            return "<span style='color:#FF8C00;'>Not authenticated. Please provide a valid authorization code first.</span>" # Neon Dark Orange
+            return "<span style='color:#FF8C00;'>Not authenticated. Please provide a valid authorization code first.</span>"  # Neon Dark Orange
 
         try:
             # This would call the actual MeldRx API to get documents for a specific patient
@@ -515,7 +457,7 @@ class CallbackManager:
                 },
             ]
         except Exception as e:
-            return f"<span style='color:#FF6347;'>Error retrieving patient documents: {str(e)}</span>: {str(e)}" # Tomato Error
+            return f"<span style='color:#FF6347;'>Error retrieving patient documents: {str(e)}</span>: {str(e)}"  # Tomato Error
 
 
 def display_form(
@@ -867,47 +809,76 @@ cyberpunk_theme = gr.themes.Monochrome(
     primary_hue="cyan",
     secondary_hue="pink",
     neutral_hue="slate",
-    font=["Source Code Pro", "monospace"], # Retro monospace font
-    font_mono=["Source Code Pro", "monospace"]
+    font=["Source Code Pro", "monospace"],  # Retro monospace font
+    font_mono=["Source Code Pro", "monospace"],
 )
 
 # Create the UI with the cyberpunk theme
-with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
-    gr.Markdown("<h1 style='color:#00FFFF; text-shadow: 0 0 5px #00FFFF;'>Discharge Guard <span style='color:#FF00FF; text-shadow: 0 0 5px #FF00FF;'>Cyber</span></h1>") # Cyberpunk Title
-
-    with gr.Tab("Authenticate with MeldRx", elem_classes="cyberpunk-tab"): # Optional: Class for tab styling
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>SMART on FHIR Authentication</h2>") # Neon Tab Header
-        auth_url_output = gr.Textbox(label="Authorization URL", value=CALLBACK_MANAGER.get_auth_url(), interactive=False)
-        gr.Markdown("<p style='color:#A9A9A9;'>Copy the URL above, open it in a browser, log in, and paste the <span style='color:#00FFFF;'>entire redirected URL</span> from your browser's address bar below.</p>") # Subdued instructions with neon highlight
-        redirected_url_input = gr.Textbox(label="Redirected URL") # New textbox for redirected URL
-        extract_code_button = gr.Button("Extract Authorization Code", elem_classes="cyberpunk-button") # Cyberpunk button style
-        extracted_code_output = gr.Textbox(label="Extracted Authorization Code", interactive=False) # Textbox to show extracted code
-
-        auth_code_input = gr.Textbox(label="Authorization Code (from above, or paste manually if extraction fails)", interactive=True) # Updated label to be clearer
-        auth_submit = gr.Button("Submit Code for Authentication", elem_classes="cyberpunk-button") # Cyberpunk button style
-        auth_result = gr.HTML(label="Authentication Result") # Use HTML for styled result
-
-        patient_data_button = gr.Button("Fetch Patient Data", elem_classes="cyberpunk-button") # Cyberpunk button style
+with gr.Blocks(theme=cyberpunk_theme) as demo:  # Apply the theme here
+    gr.Markdown(
+        "<h1 style='color:#00FFFF; text-shadow: 0 0 5px #00FFFF;'>Discharge Guard <span style='color:#FF00FF; text-shadow: 0 0 5px #FF00FF;'>Cyber</span></h1>"
+    )  # Cyberpunk Title
+
+    with gr.Tab("Authenticate with MeldRx", elem_classes="cyberpunk-tab"):  # Optional: Class for tab styling
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>SMART on FHIR Authentication</h2>"
+        )  # Neon Tab Header
+        auth_url_output = gr.Textbox(
+            label="Authorization URL",
+            value=CALLBACK_MANAGER.get_auth_url(),
+            interactive=False,
+        )
+        gr.Markdown(
+            "<p style='color:#A9A9A9;'>Copy the URL above, open it in a browser, log in, and paste the <span style='color:#00FFFF;'>entire redirected URL</span> from your browser's address bar below.</p>"
+        )  # Subdued instructions with neon highlight
+        redirected_url_input = gr.Textbox(label="Redirected URL")  # New textbox for redirected URL
+        extract_code_button = gr.Button(
+            "Extract Authorization Code", elem_classes="cyberpunk-button"
+        )  # Cyberpunk button style
+        extracted_code_output = gr.Textbox(
+            label="Extracted Authorization Code", interactive=False
+        )  # Textbox to show extracted code
+
+        auth_code_input = gr.Textbox(
+            label="Authorization Code (from above, or paste manually if extraction fails)",
+            interactive=True,
+        )  # Updated label to be clearer
+        auth_submit = gr.Button(
+            "Submit Code for Authentication", elem_classes="cyberpunk-button"
+        )  # Cyberpunk button style
+        auth_result = gr.HTML(label="Authentication Result")  # Use HTML for styled result
+
+        patient_data_button = gr.Button(
+            "Fetch Patient Data", elem_classes="cyberpunk-button"
+        )  # Cyberpunk button style
         patient_data_output = gr.Textbox(label="Patient Data", lines=10)
 
         # Add button to generate PDF from MeldRx data (No AI)
-        meldrx_pdf_button = gr.Button("Generate PDF from MeldRx Data (No AI)", elem_classes="cyberpunk-button") # Renamed button
-        meldrx_pdf_status = gr.Textbox(label="PDF Generation Status (No AI)") # Renamed status
-        meldrx_pdf_download = gr.File(label="Download Generated PDF (No AI)") # Renamed download
+        meldrx_pdf_button = gr.Button(
+            "Generate PDF from MeldRx Data (No AI)", elem_classes="cyberpunk-button"
+        )  # Renamed button
+        meldrx_pdf_status = gr.Textbox(
+            label="PDF Generation Status (No AI)"
+        )  # Renamed status
+        meldrx_pdf_download = gr.File(
+            label="Download Generated PDF (No AI)"
+        )  # Renamed download
 
         def process_redirected_url(redirected_url):
             """Processes the redirected URL to extract and display the authorization code."""
             auth_code, error_message = extract_auth_code_from_url(redirected_url)
             if auth_code:
-                return auth_code, "<span style='color:#00FF7F;'>Authorization code extracted!</span>" # Neon Green Success
+                return auth_code, "<span style='color:#00FF7F;'>Authorization code extracted!</span>"  # Neon Green Success
             else:
-                return "", f"<span style='color:#FF4500;'>Could not extract authorization code.</span> {error_message or ''}" # Neon Orange Error
-
+                return "", f"<span style='color:#FF4500;'>Could not extract authorization code.</span> {error_message or ''}"  # Neon Orange Error
 
         extract_code_button.click(
             fn=process_redirected_url,
             inputs=redirected_url_input,
-            outputs=[extracted_code_output, auth_result],# Reusing auth_result for extraction status
+            outputs=[
+                extracted_code_output,
+                auth_result,
+            ],  # Reusing auth_result for extraction status
         )
 
         auth_submit.click(
@@ -916,11 +887,19 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
             outputs=auth_result,
         )
 
-    with gr.Tab("Patient Dashboard", elem_classes="cyberpunk-tab"): # Optional: Class for tab styling
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Patient Data</h2>") # Neon Tab Header
-        dashboard_output = gr.HTML("<p style='color:#A9A9A9;'>Fetch patient data from the Authentication tab first.</p>") # Subdued placeholder text
-
-        refresh_btn = gr.Button("Refresh Data", elem_classes="cyberpunk-button") # Cyberpunk button style
+    with gr.Tab(
+        "Patient Dashboard", elem_classes="cyberpunk-tab"
+    ):  # Optional: Class for tab styling
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Patient Data</h2>"
+        )  # Neon Tab Header
+        dashboard_output = gr.HTML(
+            "<p style='color:#A9A9A9;'>Fetch patient data from the Authentication tab first.</p>"
+        )  # Subdued placeholder text
+
+        refresh_btn = gr.Button(
+            "Refresh Data", elem_classes="cyberpunk-button"
+        )  # Cyberpunk button style
 
         # Simple function to update dashboard based on fetched data
         def update_dashboard():
@@ -931,7 +910,7 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                     or data.startswith("<span style='color:#DC143C;'>Failed")
                     or data.startswith("<span style='color:#FF6347;'>Error")
                 ):
-                    return f"<p style='color:#FF8C00;'>{data}</p>" # Show auth errors in orange
+                    return f"<p style='color:#FF8C00;'>{data}</p>"  # Show auth errors in orange
 
                 try:
                     # Parse the data
@@ -945,7 +924,7 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                             patients.append(resource)
 
                     # Generate HTML card
-                    html = "<h3 style='color:#00FFFF; text-shadow: 0 0 2px #00FFFF;'>Patients</h3>" # Neon Sub-header
+                    html = "<h3 style='color:#00FFFF; text-shadow: 0 0 2px #00FFFF;'>Patients</h3>"  # Neon Sub-header
                     for patient in patients:
                         # Extract name
                         name = patient.get("name", [{}])[0]
@@ -968,13 +947,16 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
 
                     return html
                 except Exception as e:
-                    return f"<p style='color:#FF6347;'>Error parsing patient data: {str(e)}</p>" # Tomato Error
+                    return f"<p style='color:#FF6347;'>Error parsing patient data: {str(e)}</p>"  # Tomato Error
             except Exception as e:
-                return f"<p style='color:#FF6347;'>Error fetching patient data: {str(e)}</p>" # Tomato Error
+                return f"<p style='color:#FF6347;'>Error fetching patient data: {str(e)}</p>"  # Tomato Error
 
+        refresh_btn.click(fn=update_dashboard, inputs=None, outputs=dashboard_output)
 
-    with gr.Tab("Discharge Form", elem_classes="cyberpunk-tab"): # Optional: Class for tab styling
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Patient Details</h2>") # Neon Tab Header
+    with gr.Tab("Discharge Form", elem_classes="cyberpunk-tab"):  # Optional: Class for tab styling
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Patient Details</h2>"
+        )  # Neon Tab Header
         with gr.Row():
             first_name = gr.Textbox(label="First Name")
             last_name = gr.Textbox(label="Last Name")
@@ -988,7 +970,9 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
             city = gr.Textbox(label="City")
             state = gr.Textbox(label="State")
             zip_code = gr.Textbox(label="Zip Code")
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Primary Healthcare Professional Details</h2>") # Neon Sub-header
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Primary Healthcare Professional Details</h2>"
+        )  # Neon Sub-header
         with gr.Row():
             doctor_first_name = gr.Textbox(label="Doctor's First Name")
             doctor_last_name = gr.Textbox(label="Doctor's Last Name")
@@ -999,7 +983,9 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
             doctor_city = gr.Textbox(label="City")
             doctor_state = gr.Textbox(label="State")
             doctor_zip = gr.Textbox(label="Zip Code")
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Admission and Discharge Details</h2>") # Neon Sub-header
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Admission and Discharge Details</h2>"
+        )  # Neon Sub-header
         with gr.Row():
             admission_date = gr.Textbox(label="Date of Admission")
             referral_source = gr.Textbox(label="Source of Referral")
@@ -1011,24 +997,36 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                 label="Discharge Reason",
             )
         date_of_death = gr.Textbox(label="Date of Death (if applicable)")
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Diagnosis & Procedures</h2>") # Neon Sub-header
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Diagnosis & Procedures</h2>"
+        )  # Neon Sub-header
         diagnosis = gr.Textbox(label="Diagnosis")
         procedures = gr.Textbox(label="Operation & Procedures")
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Medication Details</h2>") # Neon Sub-header
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Medication Details</h2>"
+        )  # Neon Sub-header
         medications = gr.Textbox(label="Medication on Discharge")
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Prepared By</h2>") # Neon Sub-header
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Prepared By</h2>"
+        )  # Neon Sub-header
         with gr.Row():
             preparer_name = gr.Textbox(label="Name")
             preparer_job_title = gr.Textbox(label="Job Title")
 
         # Add buttons for both display form and generate PDF
         with gr.Row():
-            submit_display = gr.Button("Display Form", elem_classes="cyberpunk-button") # Cyberpunk button style
-            submit_pdf = gr.Button("Generate PDF (No AI)", elem_classes="cyberpunk-button") # Renamed button to clarify no AI and styled
+            submit_display = gr.Button(
+                "Display Form", elem_classes="cyberpunk-button"
+            )  # Cyberpunk button style
+            submit_pdf = gr.Button(
+                "Generate PDF (No AI)", elem_classes="cyberpunk-button"
+            )  # Renamed button to clarify no AI and styled
 
         # Output areas
-        form_output = gr.HTML() # Use HTML to render styled form
-        pdf_output = gr.File(label="Download PDF (No AI)")  # Renamed output to clarify no AI
+        form_output = gr.HTML()  # Use HTML to render styled form
+        pdf_output = gr.File(
+            label="Download PDF (No AI)"
+        )  # Renamed output to clarify no AI
 
         # Connect the display form button
         submit_display.click(
@@ -1064,7 +1062,7 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                 preparer_name,
                 preparer_job_title,
             ],
-            outputs=form_output
+            outputs=form_output,
         )
 
         # Connect the generate PDF button (No AI version)
@@ -1101,35 +1099,44 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                 preparer_name,
                 preparer_job_title,
             ],
-            outputs=pdf_output
+            outputs=pdf_output,
         )
 
-    with gr.Tab("Medical File Analysis", elem_classes="cyberpunk-tab"): # Optional: Class for tab styling
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Analyze Medical Files with Discharge Guard AI</h2>") # Neon Tab Header
+    with gr.Tab(
+        "Medical File Analysis", elem_classes="cyberpunk-tab"
+    ):  # Optional: Class for tab styling
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>Analyze Medical Files with Discharge Guard AI</h2>"
+        )  # Neon Tab Header
         with gr.Column():
             dicom_file = gr.File(
                 file_types=[".dcm"], label="Upload DICOM File (.dcm)"
             )
             dicom_ai_output = gr.Textbox(label="DICOM Analysis Report", lines=5)
-            analyze_dicom_button = gr.Button("Analyze DICOM with AI", elem_classes="cyberpunk-button") # Cyberpunk button style
+            analyze_dicom_button = gr.Button(
+                "Analyze DICOM with AI", elem_classes="cyberpunk-button"
+            )  # Cyberpunk button style
 
-            hl7_file = gr.File(
-                file_types=[".hl7"], label="Upload HL7 File (.hl7)"
-            )
+            hl7_file = gr.File(file_types=[".hl7"], label="Upload HL7 File (.hl7)")
             hl7_ai_output = gr.Textbox(label="HL7 Analysis Report", lines=5)
-            analyze_hl7_button = gr.Button("Analyze HL7 with AI", elem_classes="cyberpunk-button") # Cyberpunk button style
+            analyze_hl7_button = gr.Button(
+                "Analyze HL7 with AI", elem_classes="cyberpunk-button"
+            )  # Cyberpunk button style
 
-            xml_file = gr.File(
-                file_types=[".xml"], label="Upload XML File (.xml)"
-            )
+            xml_file = gr.File(file_types=[".xml"], label="Upload XML File (.xml)")
             xml_ai_output = gr.Textbox(label="XML Analysis Report", lines=5)
-            analyze_xml_button = gr.Button("Analyze XML with AI", elem_classes="cyberpunk-button") # Cyberpunk button style
+            analyze_xml_button = gr.Button(
+                "Analyze XML with AI", elem_classes="cyberpunk-button"
+            )  # Cyberpunk button style
 
             ccda_file = gr.File(
-                file_types=[".xml", ".cda", ".ccd"], label="Upload CCDA File (.xml, .cda, .ccd)"
+                file_types=[".xml", ".cda", ".ccd"],
+                label="Upload CCDA File (.xml, .cda, .ccd)",
             )
             ccda_ai_output = gr.Textbox(label="CCDA Analysis Report", lines=5)
-            analyze_ccda_button = gr.Button("Analyze CCDA with AI", elem_classes="cyberpunk-button") # Cyberpunk button style
+            analyze_ccda_button = gr.Button(
+                "Analyze CCDA with AI", elem_classes="cyberpunk-button"
+            )  # Cyberpunk button style
 
             ccd_file = gr.File(
                 file_types=[".ccd"],
@@ -1138,44 +1145,46 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
             ccd_ai_output = gr.Textbox(
                 label="CCD Analysis Report", lines=5
             )  # Redundant
-            analyze_ccd_button = gr.Button("Analyze CCD with AI", elem_classes="cyberpunk-button") # Cyberpunk button style # Redundant
-            pdf_file = gr.File(
-                file_types=[".pdf"], label="Upload PDF File (.pdf)"
-            )
+            analyze_ccd_button = gr.Button(
+                "Analyze CCD with AI", elem_classes="cyberpunk-button"
+            )  # Cyberpunk button style # Redundant
+            pdf_file = gr.File(file_types=[".pdf"], label="Upload PDF File (.pdf)")
             pdf_ai_output = gr.Textbox(label="PDF Analysis Report", lines=5)
-            analyze_pdf_button = gr.Button("Analyze PDF with AI", elem_classes="cyberpunk-button") # Cyberpunk button style
+            analyze_pdf_button = gr.Button(
+                "Analyze PDF with AI", elem_classes="cyberpunk-button"
+            )  # Cyberpunk button style
 
-            csv_file = gr.File(
-                file_types=[".csv"], label="Upload CSV File (.csv)"
-            )
+            csv_file = gr.File(file_types=[".csv"], label="Upload CSV File (.csv)")
             csv_ai_output = gr.Textbox(label="CSV Analysis Report", lines=5)
-            analyze_csv_button = gr.Button("Analyze CSV with AI", elem_classes="cyberpunk-button") # Cyberpunk button style
+            analyze_csv_button = gr.Button(
+                "Analyze CSV with AI", elem_classes="cyberpunk-button"
+            )  # Cyberpunk button style
 
         # Connect AI Analysis Buttons - using REAL AI functions now
         analyze_dicom_button.click(
             analyze_dicom_file_with_ai,  # Call REAL AI function
             inputs=dicom_file,
-            outputs=dicom_ai_output
+            outputs=dicom_ai_output,
         )
         analyze_hl7_button.click(
             analyze_hl7_file_with_ai,  # Call REAL AI function
             inputs=hl7_file,
-            outputs=hl7_ai_output
+            outputs=hl7_ai_output,
         )
         analyze_xml_button.click(
             analyze_cda_xml_file_with_ai,  # Call REAL AI function
             inputs=xml_file,
-            outputs=xml_ai_output
+            outputs=xml_ai_output,
         )
         analyze_ccda_button.click(
             analyze_cda_xml_file_with_ai,  # Call REAL AI function
             inputs=ccda_file,
-            outputs=ccda_ai_output
+            outputs=ccda_ai_output,
         )
         analyze_ccd_button.click(  # Redundant button, but kept for UI if needed
             analyze_cda_xml_file_with_ai,  # Call REAL AI function
             inputs=ccd_file,
-            outputs=ccd_ai_output
+            outputs=ccd_ai_output,
         )
         analyze_pdf_button.click(
             analyze_pdf_file_with_ai, inputs=pdf_file, outputs=pdf_ai_output
@@ -1187,9 +1196,12 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
     with gr.Tab(
         "One-Click Discharge Paper (AI)", elem_classes="cyberpunk-tab"
     ):  # New Tab for One-Click Discharge Paper with AI, styled
-        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>One-Click Medical Discharge Paper Generation with AI Content</h2>") # Neon Tab Header
+        gr.Markdown(
+            "<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>One-Click Medical Discharge Paper Generation with AI Content</h2>"
+        )  # Neon Tab Header
         one_click_ai_pdf_button = gr.Button(
-            "Generate Discharge Paper with AI (One-Click)", elem_classes="cyberpunk-button"
+            "Generate Discharge Paper with AI (One-Click)",
+            elem_classes="cyberpunk-button",
         )  # Updated button label and styled
         one_click_ai_pdf_status = gr.Textbox(
             label="Discharge Paper Generation Status (AI)"
@@ -1206,21 +1218,17 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
 
     # Connect the patient data buttons
     patient_data_button.click(
-        fn=CALLBACK_MANAGER.get_patient_data,
-        inputs=None,
-        outputs=patient_data_output
+        fn=CALLBACK_MANAGER.get_patient_data, inputs=None, outputs=patient_data_output
     )
 
     # Connect refresh button to update dashboard
-    refresh_btn.click(
-        fn=update_dashboard, inputs=None, outputs=dashboard_output
-    )
+    refresh_btn.click(fn=update_dashboard, inputs=None, outputs=dashboard_output)
 
     # Corrected the button click function name here to `generate_pdf_from_meldrx` (No AI PDF)
     meldrx_pdf_button.click(
         fn=generate_pdf_from_meldrx,
         inputs=patient_data_output,
-        outputs=[meldrx_pdf_download, meldrx_pdf_status]
+        outputs=[meldrx_pdf_download, meldrx_pdf_status],
     )
 
     # Connect patient data updates to dashboard
@@ -1228,5 +1236,6 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
         fn=update_dashboard, inputs=None, outputs=dashboard_output
     )
 
+
 # Launch with sharing enabled for public access
 demo.launch(ssr_mode=False)