improve one click discharge paper

app.py

@@ -12,13 +12,14 @@ from utils.callbackmanager import CallbackManager
 from utils.meldrx import MeldRxAPI
 from utils.prompts import system_instructions
 from old.extractcode import 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
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Import PDF utilities
 from utils.pdfutils import PDFGenerator, generate_discharge_summary
-from utils.oneclick import generate_discharge_paper_one_click
+from utils.oneclick import generate_discharge_paper_one_click, generate_ai_discharge_summary_one_click
 # Import necessary libraries for new file types and AI analysis functions
 import pydicom  # For DICOM
 import hl7  # For HL7
@@ -395,16 +396,19 @@ with gr.Blocks(theme=cyberpunk_theme) as demo: # Apply the theme here
         )
 
 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 Paper Generation with AI Content</h2>")
+    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_pdf_button = gr.Button(
-        "Generate Discharge Paper with AI (One-Click)", elem_classes="cyberpunk-button"
+    one_click_ai_button = gr.Button(
+        "Generate Discharge Summary with AI (One-Click)", elem_classes="cyberpunk-button"
+    )
+    one_click_ai_output = gr.Textbox(
+        label="AI-Generated Discharge Summary",
+        lines=20,
+        placeholder="Discharge summary will appear here..."
     )
-    one_click_ai_pdf_status = gr.Textbox(label="Discharge Paper Generation Status (AI)")
-    one_click_ai_pdf_download = gr.File(label="Download Discharge Paper (AI)")
 
     # Initialize MeldRxAPI (ensure client_id, client_secret, workspace_id are set in environment variables)
     client_id = os.getenv("APPID")
@@ -413,10 +417,10 @@ with gr.Tab("One-Click Discharge Paper (AI)", elem_classes="cyberpunk-tab"):
     redirect_uri = "https://multitransformer-discharge-guard.hf.space/callback"
     meldrx_api = MeldRxAPI(client_id, client_secret, workspace_id, redirect_uri)
 
-    one_click_ai_pdf_button.click(
-        fn=lambda pid, fname, lname: generate_discharge_paper_one_click(meldrx_api, pid, fname, lname),
+    one_click_ai_button.click(
+        fn=lambda pid, fname, lname: generate_discharge_summary_one_click(meldrx_api, pid, fname, lname),
         inputs=[patient_id_input, first_name_input, last_name_input],
-        outputs=[one_click_ai_pdf_download, one_click_ai_pdf_status],
+        outputs=one_click_ai_output,
     )
 
     # Connect the patient data buttons