Update connect_doctor.py

connect_doctor.py

@@ -39,15 +39,15 @@ def handle_connection(patient_id):
     """Generate connection link based on the assigned doctor and confirm appointment."""
     patient_name = get_patient_details(patient_id)
     if patient_name in ["Patient ID not found", "Patients file not found."]:
-        return f"Error: {patient_name}", ""
+        return f"Error: {patient_name}", "", ""
     
     doctor_name = get_assigned_doctor(patient_id)
     if doctor_name in ["Doctor not found for this Patient ID.", "No appointments found."]:
-        return f"Error: {doctor_name}", ""
+        return f"Error: {doctor_name}", "", ""
     
     connection_link = generate_jitsi_link(doctor_name)
     
-    return f"Connecting {patient_name} with {doctor_name}.", connection_link
+    return f"Connecting {patient_name} with {doctor_name}.", connection_link, doctor_name
 
 def generate_jitsi_link(doctor_name):
     """Generate a Jitsi link to connect to the selected doctor."""
@@ -71,6 +71,7 @@ def create_gradio_interface():
                 
                 doctor_name_output = gr.Textbox(label="Assigned Doctor", placeholder="Doctor will auto-populate", interactive=False)
                 connect_button = gr.Button("Connect to Doctor")
+                disconnect_button = gr.Button("Disconnect")
                 connection_text = gr.Markdown("Connection Link will appear here")
                 connection_iframe = gr.HTML()  # To display the iframe
                 
@@ -88,17 +89,27 @@ def create_gradio_interface():
 
                 # Action to connect to doctor
                 def connect_to_doctor(patient_id):
-                    message, connection_link = handle_connection(patient_id)
+                    message, connection_link, doctor_name = handle_connection(patient_id)
                     if connection_link:
                         iframe_html = f'<iframe src="{connection_link}" width="800" height="600" allow="camera; microphone; fullscreen; display-capture" style="border:none;"></iframe>'
-                        return message, iframe_html
+                        return message, iframe_html, gr.update(visible=True), gr.update(visible=False)
                     else:
-                        return message, ""
+                        return message, "", gr.update(visible=False), gr.update(visible=True)
+
+                # Action to disconnect
+                def disconnect_meeting():
+                    return "", "", gr.update(visible=False), gr.update(visible=True)
 
                 connect_button.click(
                     fn=connect_to_doctor,
                     inputs=patient_id_input,
-                    outputs=[connection_text, connection_iframe]
+                    outputs=[connection_text, connection_iframe, disconnect_button, connect_button]
+                )
+
+                disconnect_button.click(
+                    fn=disconnect_meeting,
+                    inputs=[],
+                    outputs=[connection_text, connection_iframe, disconnect_button, connect_button]
                 )
 
     return demo