Update app.py

app.py

@@ -6,7 +6,7 @@ from groq import Groq
 import base64
 
 # Set your GROQ API Key
-GROQ_API_KEY = os.getenv("GROQ_API_KEY", "your_groq_api_key_here")
+GROQ_API_KEY = os.getenv("GROQ_API_KEY", "your_groq_api_key_here")  # Replace with your actual key or set via environment
 client = Groq(api_key=GROQ_API_KEY)
 
 # Stickers for response
@@ -51,21 +51,21 @@ def skincare_advice(name, gender, age, skin_concern, description, duration, sens
 
     # Prompt for the model
     prompt = f"""
-A user named {name} (Gender: {gender}, Age: {age}) is experiencing a skin issue: "{concern_text}".
-Duration: {duration}, Sensitive skin: {sensitivity}, Routine followed: {routine}.
-
-Write a response in sections using emojis and headings only for:
-🌟 Greeting
-🩺 Skin Issue Summary
-📊 Global Insight
-🧴 Daily Skincare Routine
-🌿 Natural Remedy
-💡 Lifestyle Tips
-💬 Motivational Quote
-
-Each section should start with a title (e.g., 🌿 Natural Remedy) on one line, then 2–4 lines of helpful content below it in plain text or bullet points.
-Do not use bold, underline, or any extra styling. Do NOT return HTML.
-"""
+    A user named {name} (Gender: {gender}, Age: {age}) is experiencing a skin issue: "{concern_text}".
+    Duration: {duration}, Sensitive skin: {sensitivity}, Routine followed: {routine}.
+
+    Write a response in sections using emojis and headings only for:
+    🌟 Greeting
+    🩺 Skin Issue Summary
+    📊 Global Insight
+    🧴 Daily Skincare Routine
+    🌿 Natural Remedy
+    💡 Lifestyle Tips
+    💬 Motivational Quote
+
+    Each section should start with a title (e.g., 🌿 Natural Remedy) on one line, then 2–4 lines of helpful content below it in plain text or bullet points.
+    Do not use bold, underline, or any extra styling.
+    """
 
     try:
         response = client.chat.completions.create(
@@ -76,7 +76,7 @@ Do not use bold, underline, or any extra styling. Do NOT return HTML.
     except Exception as e:
         return f"<span style='color:red;'>❌ Error: {str(e)}</span>"
 
-    # Format response into styled HTML (cleaned, with no HTML tags from model)
+    # Format response into HTML
     lines = raw.strip().split("\n")
     html_output = ""
     section_colors = {
@@ -85,15 +85,26 @@ Do not use bold, underline, or any extra styling. Do NOT return HTML.
     }
 
     for line in lines:
-        clean_line = re.sub(r"<.*?>", "", line.strip())
-        if not clean_line:
+        stripped = line.strip()
+        if not stripped:
             continue
-        emoji = clean_line[:2]
-        if emoji in section_colors:
-            color = section_colors[emoji]
-            html_output += f"<h4 style='color:{color}; margin-top:20px;'>{clean_line}</h4>\n"
-        else:
-            html_output += f"<p style='color:#111; margin:5px 0;'>{clean_line}</p>\n"
+        
+        # Check if this line is a section header (starts with emoji and contains expected keywords)
+        is_section_header = False
+        for emoji in section_colors:
+            if stripped.startswith(emoji):
+                # Additional check to ensure it's actually a header and not content
+                header_keywords = ["Greeting", "Skin Issue Summary", "Global Insight", "Daily Skincare Routine", 
+                                 "Natural Remedy", "Lifestyle Tips", "Motivational Quote"]
+                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"
+                    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"
 
     # Embed uploaded image
     image_html = ""
@@ -115,10 +126,11 @@ Do not use bold, underline, or any extra styling. Do NOT return HTML.
     {html_output}
 </div>
 """
+
     return final_html
 
 # Handle button press
 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)
+        st.markdown(result, unsafe_allow_html=True)