Update app.py

app.py

@@ -190,13 +190,49 @@ Medication: {d['medication']}
         d["factors"]
     )
 
-    formatted = last_llm_report.replace("\n", "<br>")
+    lines = last_llm_report.split("\n")
+
+    formatted_parts = []
+    disclaimer_html = ""
+
+    for line in lines:
+        line = line.strip()
+        if not line:
+            continue
+
+        # Detect disclaimer line
+        if "consult a qualified doctor" in line.lower():
+            disclaimer_html = f"""
+            <div style="
+                margin-top:15px;
+                padding:12px;
+                background:#fff3cd;
+                border-left:5px solid #ff9800;
+                border-radius:6px;
+                font-size:14px;">
+                <b>Disclaimer:</b><br>
+                {line}
+            </div>
+            """
+            continue
+
+        # Detect headings starting with ##
+        if line.startswith("##"):
+            clean_heading = line.replace("##", "").strip()
+            formatted_parts.append(
+                f"<br><b style='font-size:16px;color:#003366;'>{clean_heading}:</b><br>"
+            )
+        else:
+            formatted_parts.append(f"{line}<br>")
+
+    formatted = "".join(formatted_parts)
 
     return basic_html + f"""
     <hr>
     <h3 style="color:#003366;">Detailed AI Report & Recommendations</h3>
     <div style="background:#f9f9f9;padding:15px;border-radius:8px;">
     {formatted}
+    {disclaimer_html}
     </div>
     """