Update app.py

app.py

@@ -160,7 +160,7 @@ def save_results_to_pdf(test_results, filename):
         return f"Error saving PDF: {str(e)}", None
 
 # Build health card layout
-def build_health_card(profile_image, test_results, summary, patient_name="", patient_age="", patient_gender="", patient_id=""):
+def build_health_card(profile_image, test_results, summary, patient_name="", patient_age="", patient_gender="", patient_id="", pdf_filepath=""):
     from datetime import datetime
     current_date = datetime.now().strftime("%B %d, %Y")
     
@@ -319,7 +319,8 @@ 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=""
     )
     
     # Generate PDF and return for download
@@ -331,29 +332,20 @@ def analyze_face(input_data):
         temp_pdf_path = "/tmp/" + os.path.basename(pdf_filepath)
         shutil.copy(pdf_filepath, temp_pdf_path)
         
+        # Update the HTML to include the link to download the PDF
+        health_card_html = build_health_card(
+            profile_image_base64, 
+            test_results,
+            summary, 
+            current_patient_details['name'],
+            current_patient_details['age'],
+            current_patient_details['gender'],
+            current_patient_details['id'],
+            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
-
-
 # Create Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("""# 🧠 Face-Based Lab Test AI Report (Video Mode)""")