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")
     
@@ -203,19 +203,13 @@ 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;">
-            <button onclick="downloadHealthCard()" 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 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>
         </div>
     </div>
-    <script>
-    function downloadHealthCard() {{
-        const link = document.createElement('a');
-        link.href = '{pdf_filepath}';
-        link.download = 'Health_Report.pdf';
-        link.click();
-    }}
-    </script>
     """
     return html
 
@@ -318,7 +312,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="/tmp/Health_Report.pdf"
     )
     
     # Generate PDF and return for download
@@ -333,29 +328,10 @@ def analyze_face(input_data):
         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
+# Gradio interface setup
 with gr.Blocks() as demo:
-    gr.Markdown("""# 🧠 Face-Based Lab Test AI Report (Video Mode)""")
+    gr.Markdown("""# 🧠 Face-Based Lab Test AI Report""")
+    
     with gr.Row():
         with gr.Column():
             gr.Markdown("### Patient Information")
@@ -376,7 +352,7 @@ 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,
+            submit_btn.click(fn=analyze_face,
                              inputs=[mode_selector, image_input, video_input, patient_name, patient_age, patient_gender, patient_id],
                              outputs=[result_html, result_pdf])