Update callbackmanager.py

callbackmanager.py

@@ -875,7 +875,7 @@ cyberpunk_theme = gr.themes.Monochrome(
 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
+    with gr.Tab("Authenticate with MeldRx", elem_classes="cyberpunk-tab", js=True): # 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
@@ -891,7 +891,7 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
         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_button = gr.Button("Generate PDF from MeldRx Data (No AI)", elem_classes="cyberpunk-button", js=True) # 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
 
@@ -907,13 +907,15 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
         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],
+            js=True# Reusing auth_result for extraction status
         )
 
         auth_submit.click(
             fn=CALLBACK_MANAGER.set_auth_code,
             inputs=extracted_code_output,  # Using extracted code as input for authentication
             outputs=auth_result,
+            js=True
         )
 
     with gr.Tab("Patient Dashboard", elem_classes="cyberpunk-tab"): # Optional: Class for tab styling
@@ -973,38 +975,38 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                 return f"<p style='color:#FF6347;'>Error fetching patient data: {str(e)}</p>" # Tomato Error
 
 
-    with gr.Tab("Discharge Form", elem_classes="cyberpunk-tab"): # Optional: Class for tab styling
+    with gr.Tab("Discharge Form", elem_classes="cyberpunk-tab", js=True): # 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():
+        with gr.Row(js=True):
             first_name = gr.Textbox(label="First Name")
             last_name = gr.Textbox(label="Last Name")
             middle_initial = gr.Textbox(label="Middle Initial")
-        with gr.Row():
+        with gr.Row(js=True):
             dob = gr.Textbox(label="Date of Birth")
             age = gr.Textbox(label="Age")
             sex = gr.Textbox(label="Sex")
         address = gr.Textbox(label="Address")
-        with gr.Row():
+        with gr.Row(js=True):
             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
-        with gr.Row():
+        with gr.Row(js=True):
             doctor_first_name = gr.Textbox(label="Doctor's First Name")
             doctor_last_name = gr.Textbox(label="Doctor's Last Name")
             doctor_middle_initial = gr.Textbox(label="Doctor's Middle Initial")
         hospital_name = gr.Textbox(label="Hospital/Clinic Name")
         doctor_address = gr.Textbox(label="Address")
-        with gr.Row():
+        with gr.Row(js=True):
             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
-        with gr.Row():
+        with gr.Row(js=True):
             admission_date = gr.Textbox(label="Date of Admission")
             referral_source = gr.Textbox(label="Source of Referral")
         admission_method = gr.Textbox(label="Method of Admission")
-        with gr.Row():
+        with gr.Row(js=True):
             discharge_date = gr.Textbox(label="Date of Discharge")
             discharge_reason = gr.Radio(
                 ["Treated", "Transferred", "Discharge Against Advice", "Patient Died"],
@@ -1017,12 +1019,12 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
         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
-        with gr.Row():
+        with gr.Row(js=True):
             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():
+        with gr.Row(js=True):
             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
 
@@ -1065,6 +1067,7 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                 preparer_job_title,
             ],
             outputs=form_output,
+            js=True
         )
 
         # Connect the generate PDF button (No AI version)
@@ -1102,11 +1105,12 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                 preparer_job_title,
             ],
             outputs=pdf_output,
+            js=True
         )
 
     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():
+        with gr.Column(js=True):
             dicom_file = gr.File(
                 file_types=[".dcm"], label="Upload DICOM File (.dcm)"
             )
@@ -1156,36 +1160,41 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
             analyze_dicom_file_with_ai,  # Call REAL AI function
             inputs=dicom_file,
             outputs=dicom_ai_output,
+            js=True
         )
         analyze_hl7_button.click(
             analyze_hl7_file_with_ai,  # Call REAL AI function
             inputs=hl7_file,
             outputs=hl7_ai_output,
+            js=True
         )
         analyze_xml_button.click(
             analyze_cda_xml_file_with_ai,  # Call REAL AI function
             inputs=xml_file,
             outputs=xml_ai_output,
+            js=True
         )
         analyze_ccda_button.click(
             analyze_cda_xml_file_with_ai,  # Call REAL AI function
             inputs=ccda_file,
             outputs=ccda_ai_output,
+            js=True
         )
         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,
+            js=True
         )
         analyze_pdf_button.click(
-            analyze_pdf_file_with_ai, inputs=pdf_file, outputs=pdf_ai_output
+            analyze_pdf_file_with_ai, inputs=pdf_file, outputs=pdf_ai_output, js=True
         )
         analyze_csv_button.click(
-            analyze_csv_file_with_ai, inputs=csv_file, outputs=csv_ai_output
+            analyze_csv_file_with_ai, inputs=csv_file, outputs=csv_ai_output, js=True
         )
 
     with gr.Tab(
-        "One-Click Discharge Paper (AI)", elem_classes="cyberpunk-tab"
+        "One-Click Discharge Paper (AI)", elem_classes="cyberpunk-tab", js=True
     ):  # 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
         one_click_ai_pdf_button = gr.Button(
@@ -1209,11 +1218,12 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
         fn=CALLBACK_MANAGER.get_patient_data,
         inputs=None,
         outputs=patient_data_output,
+        js=True
     )
 
     # Connect refresh button to update dashboard
     refresh_btn.click(
-        fn=update_dashboard, inputs=None, outputs=dashboard_output
+        fn=update_dashboard, inputs=None, outputs=dashboard_output, js=True
     )
 
     # Corrected the button click function name here to `generate_pdf_from_meldrx` (No AI PDF)
@@ -1221,12 +1231,13 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
         fn=generate_pdf_from_meldrx,
         inputs=patient_data_output,
         outputs=[meldrx_pdf_download, meldrx_pdf_status],
+        js=True
     )
 
     # Connect patient data updates to dashboard
     patient_data_button.click(
-        fn=update_dashboard, inputs=None, outputs=dashboard_output
+        fn=update_dashboard, inputs=None, outputs=dashboard_output, js=True
     )
 
 # Launch with sharing enabled for public access
-demo.launch(share=True)
+demo.launch(ssr_mode=False)