revert function

app.py

@@ -19,7 +19,7 @@ logger = logging.getLogger(__name__)
 
 # Import PDF utilities
 from utils.pdfutils import PDFGenerator, generate_discharge_summary
-from utils.oneclick import generate_discharge_paper_one_click, generate_ai_discharge_summary_one_click
+from utils.oneclick import generate_discharge_paper_one_click
 # Import necessary libraries for new file types and AI analysis functions
 import pydicom  # For DICOM
 import hl7  # For HL7
@@ -94,42 +94,42 @@ CALLBACK_MANAGER = CallbackManager(
 #             return f"Authentication successful! Access Token: {self.access_token[:10]}... (truncated)"
 #         return "Authentication failed. Please check the authorization code."
     
-def generate_discharge_paper_one_click():
-    """One-click function to fetch patient data and generate discharge paper with AI Content."""
-    patient_data_str = CALLBACK_MANAGER.get_patient_data()
-    if (
-        patient_data_str.startswith("Not authenticated")
-        or patient_data_str.startswith("Failed")
-        or patient_data_str.startswith("Error")
-    ):
-        return None, patient_data_str  # Return error message if authentication or data fetch fails
+# def generate_discharge_paper_one_click():
+#     """One-click function to fetch patient data and generate discharge paper with AI Content."""
+#     patient_data_str = CALLBACK_MANAGER.get_patient_data()
+#     if (
+#         patient_data_str.startswith("Not authenticated")
+#         or patient_data_str.startswith("Failed")
+#         or patient_data_str.startswith("Error")
+#     ):
+#         return None, patient_data_str  # Return error message if authentication or data fetch fails
 
-    try:
-        patient_data = json.loads(patient_data_str)
+#     try:
+#         patient_data = json.loads(patient_data_str)
 
-        # --- AI Content Generation for Discharge Summary ---
-        # This is a placeholder - Replace with actual AI call using InferenceClient and patient_data to generate content
-        ai_generated_content = generate_ai_discharge_content(
-            patient_data
-        )  # Placeholder AI function
+#         # --- AI Content Generation for Discharge Summary ---
+#         # This is a placeholder - Replace with actual AI call using InferenceClient and patient_data to generate content
+#         ai_generated_content = generate_ai_discharge_content(
+#             patient_data
+#         )  # Placeholder AI function
 
-        if not ai_generated_content:
-            return None, "Error: AI content generation failed."
+#         if not ai_generated_content:
+#             return None, "Error: AI content generation failed."
 
-        # --- PDF Generation with AI Content ---
-        pdf_path, status_message = generate_pdf_from_meldrx_with_ai_content(
-            patient_data, ai_generated_content
-        )  # Function to generate PDF with AI content
+#         # --- PDF Generation with AI Content ---
+#         pdf_path, status_message = generate_pdf_from_meldrx_with_ai_content(
+#             patient_data, ai_generated_content
+#         )  # Function to generate PDF with AI content
 
-        if pdf_path:
-            return pdf_path, status_message
-        else:
-            return None, status_message  # Return status message if PDF generation fails
+#         if pdf_path:
+#             return pdf_path, status_message
+#         else:
+#             return None, status_message  # Return status message if PDF generation fails
 
-    except json.JSONDecodeError:
-        return None, "Error: Patient data is not in valid JSON format."
-    except Exception as e:
-        return None, f"Error during discharge paper generation: {str(e)}"
+#     except json.JSONDecodeError:
+#         return None, "Error: Patient data is not in valid JSON format."
+#     except Exception as e:
+#         return None, f"Error during discharge paper generation: {str(e)}"
 
 
 
@@ -418,7 +418,7 @@ with gr.Tab("One-Click Discharge Paper (AI)", elem_classes="cyberpunk-tab"):
     meldrx_api = MeldRxAPI(client_id, client_secret, workspace_id, redirect_uri)
 
     one_click_ai_button.click(
-        fn=lambda pid, fname, lname: generate_discharge_summary_one_click(meldrx_api, pid, fname, lname),
+        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,
     )