attempt to generate discharge paper with auth

app.py

@@ -29,7 +29,7 @@ import csv  # For CSV
 import io  # For IO operations
 from PIL import Image  # For image handling
 from utils.callbackmanager import extract_auth_code_from_url, extract_code_from_url
-from utils.generators import  generate_pdf_from_meldrx, generate_ai_discharge_content, generate_pdf_from_meldrx_with_ai_content, extract_section,  generate_pdf_from_form, generate_discharge_summary, generate_ai_discharge_content, analyze_dicom_file_with_ai, analyze_hl7_file_with_ai, analyze_cda_xml_file_with_ai, analyze_pdf_file_with_ai, analyze_csv_file_with_ai, generate_pdf_from_form , generate_ai_discharge_content , extract_section , generate_pdf_from_meldrx_with_ai_content 
+from utils.generators import  generate_pdf_from_meldrx, generate_ai_discharge_content, generate_pdf_from_meldrx_with_ai_content, extract_section,  generate_pdf_from_form, generate_discharge_summary, generate_ai_discharge_content, analyze_dicom_file_with_ai, analyze_hl7_file_with_ai, analyze_cda_xml_file_with_ai, analyze_pdf_file_with_ai, analyze_csv_file_with_ai, generate_pdf_from_form , generate_ai_discharge_content , extract_section , generate_pdf_from_meldrx_with_ai_content
 
 
 # Initialize Inference Client - Ensure YOUR_HF_TOKEN is set in environment variables or replace with your actual token
@@ -340,16 +340,27 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
             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...")
+        one_click_ai_output = gr.Textbox(label="AI-Generated Discharge Summary", lines=20, placeholder="Discharge summary will appear here...")
 
+        # Initialize MeldRxAPI (ensure client_id, client_secret, workspace_id are set in environment variables)
+        client_id = os.getenv("APPID")
+        client_secret = os.getenv("CLIENT_SECRET")
+        workspace_id = os.getenv("WORKSPACE_URL")
+        redirect_uri = "https://multitransformer-discharge-guard.hf.space/callback"
+        meldrx_api = MeldRxAPI(client_id, client_secret, workspace_id, redirect_uri)
+
+        one_click_ai_button.click(
+            fn=lambda pid, fname, lname: generate_discharge_paper_one_click(meldrx_api, pid, fname, lname),
+            inputs=[patient_id_input, first_name_input, last_name_input],
+            outputs=one_click_ai_output,
+        )
         # 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
@@ -359,11 +370,11 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
                 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],
-        )
+        # one_click_ai_button.click(
+        #     fn=lambda pid, fname, lname: generate_discharge_paper_one_click(meldrx_api, pid, fname, lname),
+        #     inputs=[patient_id_input, first_name_input, last_name_input],
+        #     outputs=one_click_ai_output,
+        # )
 
 
     # Connect the patient data buttons