revert function

app.py

@@ -395,27 +395,27 @@ 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.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")  # Optional, set if required
-    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)
+    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.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,
+        )
 
     one_click_ai_button.click(
         fn=lambda pid, fname, lname: generate_discharge_paper_one_click(meldrx_api, pid, fname, lname),