Update app.py

app.py

@@ -76,6 +76,11 @@ def skincare_advice(name, gender, age, skin_concern, description, duration, sens
     except Exception as e:
         return f"<span style='color:red;'>❌ Error: {str(e)}</span>"
 
+    # Clean the raw response first - remove any HTML tags that might be in the AI response
+    import html
+    raw = html.unescape(raw)  # Decode HTML entities
+    raw = re.sub(r'<[^>]+>', '', raw)  # Remove any HTML tags
+    
     # Format response into HTML
     lines = raw.strip().split("\n")
     html_output = ""
@@ -89,6 +94,10 @@ def skincare_advice(name, gender, age, skin_concern, description, duration, sens
         if not stripped:
             continue
         
+        # Skip lines that look like HTML code or contain style attributes
+        if '<' in stripped and ('style=' in stripped or '/>' in stripped or '</p>' in stripped):
+            continue
+            
         # Check if this line is a section header (starts with emoji and contains expected keywords)
         is_section_header = False
         for emoji in section_colors:
@@ -99,12 +108,16 @@ def skincare_advice(name, gender, age, skin_concern, description, duration, sens
                 if any(keyword.lower() in stripped.lower() for keyword in header_keywords):
                     is_section_header = True
                     color = section_colors[emoji]
-                    html_output += f"<h4 style='color:{color}; margin-top:20px; margin-bottom:10px;'>{stripped}</h4>\n"
+                    # Escape any remaining HTML characters in the header text
+                    safe_header = html.escape(stripped)
+                    html_output += f"<h4 style='color:{color}; margin-top:20px; margin-bottom:10px;'>{safe_header}</h4>\n"
                     break
         
         # If not a section header, treat as regular content
         if not is_section_header:
-            html_output += f"<p style='color:#111; margin:5px 0; line-height:1.5;'>{stripped}</p>\n"
+            # Escape any HTML characters in the content
+            safe_content = html.escape(stripped)
+            html_output += f"<p style='color:#111; margin:5px 0; line-height:1.5;'>{safe_content}</p>\n"
 
     # Embed uploaded image
     image_html = ""