Update app.py

app.py

@@ -302,6 +302,13 @@ def analyze_face(input_data):
     _, buffer = cv2.imencode('.png', frame_rgb)
     profile_image_base64 = base64.b64encode(buffer).decode('utf-8')
 
+    # Create a temporary file path for the PDF
+    pdf_filename = f"Health_Report_{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.pdf"
+    pdf_filepath = f"/tmp/{pdf_filename}"
+
+    # Save the PDF
+    pdf_result, _ = save_results_to_pdf(test_results, pdf_filepath)
+
     # Use global patient details
     global current_patient_details
     health_card_html = build_health_card(
@@ -311,39 +318,11 @@ def analyze_face(input_data):
         current_patient_details['name'],
         current_patient_details['age'],
         current_patient_details['gender'],
-        current_patient_details['id']
+        current_patient_details['id'],
+        pdf_filepath=pdf_filepath
     )
     
-    # Generate PDF and return for download
-    pdf_filename = f"Health_Report_{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.pdf"
-    pdf_result, pdf_filepath = save_results_to_pdf(test_results, pdf_filename)
-    
-    if pdf_filepath:
-        # Copy the PDF to a temporary directory for Gradio to serve it
-        temp_pdf_path = "/tmp/" + os.path.basename(pdf_filepath)
-        shutil.copy(pdf_filepath, temp_pdf_path)
-        
-        return health_card_html, temp_pdf_path
-
-
-# Modified route_inputs function
-def route_inputs(mode, image, video, patient_name, patient_age, patient_gender, patient_id):
-    if mode == "Image" and image is None:
-        return "<div style='color:red;'>⚠️ Error: No image provided.</div>", None
-    if mode == "Video" and video is None:
-        return "<div style='color:red;'>⚠️ Error: No video provided.</div>", None
-    
-    # Store patient details globally for use in analyze_face
-    global current_patient_details
-    current_patient_details = {
-        'name': patient_name,
-        'age': patient_age, 
-        'gender': patient_gender,
-        'id': patient_id
-    }
-    
-    health_card_html, pdf_file_path = analyze_face(image if mode == "Image" else video)
-    return health_card_html, pdf_file_path
+    return health_card_html, pdf_filepath
 
 
 # Create Gradio interface