Update app.py

app.py

@@ -1,4 +1,4 @@
-import os  # Import the os module
+Import the os module
 import gradio as gr
 import cv2
 import numpy as np
@@ -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="", pdf_filepath=""):
+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")
     
@@ -203,11 +203,12 @@ 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>
@@ -300,6 +301,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))])
@@ -319,8 +321,7 @@ def analyze_face(input_data):
         current_patient_details['name'],
         current_patient_details['age'],
         current_patient_details['gender'],
-        current_patient_details['id'],
-        pdf_filepath=""
+        current_patient_details['id']
     )
     
     # Generate PDF and return for download
@@ -332,20 +333,29 @@ 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)""")
@@ -369,7 +379,9 @@ with gr.Blocks() as demo:
             result_html = gr.HTML(label="🧪 Health Report Table")
             result_pdf = gr.File(label="Download Health Report PDF", interactive=False)
 
-            
+            submit_btn.click(fn=route_inputs,
+                             inputs=[mode_selector, image_input, video_input, patient_name, patient_age, patient_gender, patient_id],
+                             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)