Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os  # Import the os module
 import gradio as gr
 import cv2
@@ -160,8 +159,8 @@ def save_results_to_pdf(test_results, filename):
     except Exception as e:
         return f"Error saving PDF: {str(e)}", None
 
-# Function to generate the health card layout
-def build_health_card(profile_image, test_results, summary, patient_name="", patient_age="", patient_gender="", patient_id="", pdf_filepath=""):
+# Build health card layout
+def build_health_card(profile_image, test_results, summary, patient_name="", patient_age="", patient_gender="", patient_id=""):
     from datetime import datetime
     current_date = datetime.now().strftime("%B %d, %Y")
     
@@ -204,13 +203,20 @@ def build_health_card(profile_image, test_results, summary, patient_name="", pat
         </div>
 
         <div style="display: flex; gap: 15px; justify-content: center; flex-wrap: wrap;">
-            <a href="{pdf_filepath}" download="Health_Report.pdf">
-                <button style="padding: 12px 24px; background: linear-gradient(135deg, #4caf50, #45a049); color: white; border: none; border-radius: 8px; cursor: pointer; font-weight: 600; font-size: 14px; box-shadow: 0 4px 12px rgba(76, 175, 80, 0.3); transition: all 0.3s;">
-                    📥 Download Report
-                </button>
-            </a>
+            <button onclick="window.print()" style="padding: 12px 24px; background: linear-gradient(135deg, #4caf50, #45a049); color: white; border: none; border-radius: 8px; cursor: pointer; font-weight: 600; font-size: 14px; box-shadow: 0 4px 12px rgba(76, 175, 80, 0.3); transition: all 0.3s;">
+                📥 Download Report
+            </button>
+            <button style="padding: 12px 24px; background: linear-gradient(135deg, #2196f3, #1976d2); color: white; border: none; border-radius: 8px; cursor: pointer; font-weight: 600; font-size: 14px; box-shadow: 0 4px 12px rgba(33, 150, 243, 0.3);">
+                📞 Find Labs Near Me
+            </button>
         </div>
     </div>
+    <style>
+    @media print {{
+        /* Hide input sections during print */
+        .gradio-container {{ display: none; }}
+    }}
+    </style>
     """
     return html
 
@@ -293,6 +299,7 @@ def analyze_face(input_data):
         build_table("❤️ Vitals",
                     [("SpO2", spo2, (95, 100)),
                      ("Heart Rate", heart_rate, (60, 100)),
+                     ("Respiratory Rate", rr, (12, 20)),
                      ("Temperature", test_values["Temperature"], (97, 99)),
                      ("BP Systolic", test_values["BP Systolic"], (90, 120)),
                      ("BP Diastolic", test_values["BP Diastolic"], (60, 80))])
@@ -303,13 +310,6 @@ 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(
@@ -319,11 +319,39 @@ def analyze_face(input_data):
         current_patient_details['name'],
         current_patient_details['age'],
         current_patient_details['gender'],
-        current_patient_details['id'],
-        pdf_filepath=pdf_filepath
+        current_patient_details['id']
     )
     
-    return health_card_html, 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
 
 
 # Create Gradio interface
@@ -354,4 +382,4 @@ with gr.Blocks() as demo:
                              outputs=[result_html, result_pdf])
 
 # Launch Gradio for Replit
-demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(server_name="0.0.0.0", server_port=7860)