Update app.py

app.py

@@ -4,7 +4,6 @@ import os
 import re
 from groq import Groq
 import base64
-import html
 
 # Set your GROQ API Key
 GROQ_API_KEY = os.getenv("GROQ_API_KEY", "your_groq_api_key_here")  # Replace with your actual key or set via environment
@@ -82,9 +81,9 @@ def skincare_advice(name, gender, age, skin_concern, description, duration, sens
     raw = html.unescape(raw)  # Decode HTML entities
     raw = re.sub(r'<[^>]+>', '', raw)  # Remove any HTML tags
     
-    # Format response into HTML
+    # Format response using markdown and simple text formatting
     lines = raw.strip().split("\n")
-    html_output = ""
+    formatted_content = ""
     section_colors = {
         "🌟": "#E53935", "🩺": "#D81B60", "📊": "#5E35B1",
         "🧴": "#F57C00", "🌿": "#43A047", "💡": "#039BE5", "💬": "#6D4C41"
@@ -109,16 +108,22 @@ 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]
-                    # 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"
+                    # Use HTML span for colored headers
+                    formatted_content += f"<br><span style='color:{color}; font-weight:bold; font-size:18px;'>{stripped}</span><br>\n"
                     break
         
         # If not a section header, treat as regular content
         if not is_section_header:
-            # 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"
+            # Clean content and add as regular text
+            clean_content = stripped.replace('<', '&lt;').replace('>', '&gt;')
+            formatted_content += f"{clean_content}<br>\n"
+    
+    # Build the final response with proper HTML structure
+    html_output = f"""
+<div style='margin: 10px 0;'>
+    {formatted_content}
+</div>
+"""
 
     # Embed uploaded image
     image_html = ""
@@ -147,4 +152,7 @@ def skincare_advice(name, gender, age, skin_concern, description, duration, sens
 if submit:
     with st.spinner("Generating advice..."):
         result = skincare_advice(name, gender, age, skin_concern, description, duration, sensitivity, routine, image)
-        st.markdown(result, unsafe_allow_html=True)
+        if result.startswith("<span style='color:red;'>"):
+            st.markdown(result, unsafe_allow_html=True)
+        else:
+            st.markdown(result, unsafe_allow_html=True)