attempt to generate discharge paper with auth

app.py

@@ -333,37 +333,37 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
             analyze_csv_file_with_ai, inputs=csv_file, outputs=csv_ai_output
         )
 
-with gr.Tab("One-Click Discharge Paper (AI)", elem_classes="cyberpunk-tab"):
-    gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>One-Click Medical Discharge Summary with AI Content</h2>")
-    with gr.Row():
-        patient_id_input = gr.Textbox(label="Patient ID (Optional)", placeholder="Enter Patient ID")
-        first_name_input = gr.Textbox(label="First Name (Optional)", placeholder="Enter First Name")
-        last_name_input = gr.Textbox(label="Last Name (Optional)", placeholder="Enter Last Name")
-    one_click_ai_button = gr.Button("Generate Discharge Summary with AI (One-Click)", elem_classes="cyberpunk-button")
-    one_click_ai_output = gr.File(label="Download AI-Generated Discharge PDF")  # Changed to File output for PDF
-    one_click_ai_status = gr.Textbox(label="Generation Status", lines=2, placeholder="Status will appear here...")
-
-    # Use the global CALLBACK_MANAGER instead of creating a new MeldRxAPI instance
-    def one_click_handler(patient_id, first_name, last_name):
-        try:
-            # Check if CALLBACK_MANAGER is authenticated
-            if not CALLBACK_MANAGER.access_token:
-                return None, "Error: Not authenticated. Please authenticate in the 'Authenticate with MeldRx' tab first."
-            
-            # Call the one-click function with the existing authenticated CALLBACK_MANAGER.api
-            pdf_path, status = generate_discharge_paper_one_click(
-                CALLBACK_MANAGER.api, patient_id, first_name, last_name
-            )
-            return pdf_path, status
-        except Exception as e:
-            logger.error(f"One-click handler error: {str(e)}")
-            return None, f"Error: {str(e)}"
-
-    one_click_ai_button.click(
-        fn=one_click_handler,
-        inputs=[patient_id_input, first_name_input, last_name_input],
-        outputs=[one_click_ai_output, one_click_ai_status],
-    )
+    with gr.Tab("One-Click Discharge Paper (AI)", elem_classes="cyberpunk-tab"):
+        gr.Markdown("<h2 style='color:#00FFFF; text-shadow: 0 0 3px #00FFFF;'>One-Click Medical Discharge Summary with AI Content</h2>")
+        with gr.Row():
+            patient_id_input = gr.Textbox(label="Patient ID (Optional)", placeholder="Enter Patient ID")
+            first_name_input = gr.Textbox(label="First Name (Optional)", placeholder="Enter First Name")
+            last_name_input = gr.Textbox(label="Last Name (Optional)", placeholder="Enter Last Name")
+        one_click_ai_button = gr.Button("Generate Discharge Summary with AI (One-Click)", elem_classes="cyberpunk-button")
+        one_click_ai_output = gr.File(label="Download AI-Generated Discharge PDF")  # Changed to File output for PDF
+        one_click_ai_status = gr.Textbox(label="Generation Status", lines=2, placeholder="Status will appear here...")
+
+        # Use the global CALLBACK_MANAGER instead of creating a new MeldRxAPI instance
+        def one_click_handler(patient_id, first_name, last_name):
+            try:
+                # Check if CALLBACK_MANAGER is authenticated
+                if not CALLBACK_MANAGER.access_token:
+                    return None, "Error: Not authenticated. Please authenticate in the 'Authenticate with MeldRx' tab first."
+                
+                # Call the one-click function with the existing authenticated CALLBACK_MANAGER.api
+                pdf_path, status = generate_discharge_paper_one_click(
+                    CALLBACK_MANAGER.api, patient_id, first_name, last_name
+                )
+                return pdf_path, status
+            except Exception as e:
+                logger.error(f"One-click handler error: {str(e)}")
+                return None, f"Error: {str(e)}"
+
+        one_click_ai_button.click(
+            fn=one_click_handler,
+            inputs=[patient_id_input, first_name_input, last_name_input],
+            outputs=[one_click_ai_output, one_click_ai_status],
+        )
 
 
     # Connect the patient data buttons