Update app.py

app.py

@@ -98,6 +98,13 @@ def get_risk_color(value, normal_range):
 
 # Function to build table for test results
 def build_table(title, rows):
+    # Define color mapping
+    color_map = {
+        "Normal": "#28a745",
+        "High": "#dc3545",
+        "Low": "#ffc107"
+    }
+    
     html = (
         f'<div style="margin-bottom: 25px; border-radius: 8px; overflow: hidden; border: 1px solid #e0e0e0;">'
         f'<div style="background: linear-gradient(135deg, #f5f7fa, #c3cfe2); padding: 12px 16px; border-bottom: 1px solid #e0e0e0;">'
@@ -117,7 +124,7 @@ def build_table(title, rows):
         else:
             value_str = f"{value:.2f}"
             
-        html += f'<tr style="background:{row_bg};border-bottom:1px solid #e9ecef;"><td style="padding:10px 8px;color:#2c3e50;font-weight:500;">{label}</td><td style="padding:10px 8px;text-align:center;color:#2c3e50;font-weight:600;">{value_str}</td><td style="padding:10px 8px;text-align:center;color:#6c757d;font-size:12px;">{ref[0]} - {ref[1]}</td><td style="padding:10px 8px;text-align:center;font-weight:600;color:{"#28a745" if level == "Normal" else "#dc3545" if level == "High" else "#ffc107"};">{icon} {level}</td></tr>'
+        html += f'<tr style="background:{row_bg};border-bottom:1px solid #e9ecef;"><td style="padding:10px 8px;color:#2c3e50;font-weight:500;">{label}</td><td style="padding:10px 8px;text-align:center;color:#2c3e50;font-weight:600;">{value_str}</td><td style="padding:10px 8px;text-align:center;color:#6c757d;font-size:12px;">{ref[0]} - {ref[1]}</td><td style="padding:10px 8px;text-align:center;font-weight:600;color:{color_map[level]};">{icon} {level}</td></tr>'
     html += '</tbody></table></div>'
     return html
 
@@ -199,6 +206,13 @@ def build_health_card(profile_image, test_results, summary, pdf_filepath, patien
     
     pdf_filename = os.path.basename(pdf_filepath) if pdf_filepath else "health_report.pdf"
     
+    # Define color mapping for consistency
+    color_map = {
+        "Normal": "#28a745",
+        "High": "#dc3545",
+        "Low": "#ffc107"
+    }
+    
     html = f"""
     <div id="health-card" style="font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; max-width: 700px; margin: 20px auto; border-radius: 16px; background: linear-gradient(135deg, #e3f2fd 0%, #f3e5f5 100%); border: 2px solid #ddd; box-shadow: 0 8px 32px rgba(0, 0, 0, 0.15); padding: 30px; color: #1a1a1a;">
         
@@ -247,12 +261,12 @@ def build_health_card(profile_image, test_results, summary, pdf_filepath, patien
         </div>
     </div>
     <style>
-    @media print {
+    @media print {{
         /* Hide input sections during print */
-        .gradio-container { display: block; }
+        .gradio-container {{ display: block; }}
         /* Keep only the health card visible */
-        #health-card { display: block; }
-    }
+        #health-card {{ display: block; }}
+    }}
     </style>
     """
     return html