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")  # Replace with your actual key or set via environment
+GROQ_API_KEY = os.getenv("GROQ_API_KEY", "your_groq_api_key_here")  # Replace with actual key
 client = Groq(api_key=GROQ_API_KEY)
 
 # Stickers for response
@@ -51,7 +51,7 @@ 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}\".
+    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:
@@ -84,16 +84,32 @@ def skincare_advice(name, gender, age, skin_concern, description, duration, sens
         "🧴": "#F57C00", "🌿": "#43A047", "💡": "#039BE5", "💬": "#6D4C41"
     }
 
+    current_section = None
+    section_buffer = []
+
     for line in lines:
-        stripped = re.sub(r"</div\s*>", "", line.strip(), flags=re.IGNORECASE)
-        if not stripped:
+        line = line.strip()
+        if not line:
             continue
-        if any(stripped.startswith(e) for e in section_colors):
-            emoji = stripped[:2]
-            color = section_colors.get(emoji, "#000")
-            html_output += f"<h4 style='color:{color}; margin-top:20px;'>{stripped}</h4>\n"
+
+        emoji = line[:2]
+        if emoji in section_colors:
+            if current_section:
+                html_output += f"<h4 style='color:{section_colors[current_section]}; margin-top:20px;'>{section_title}</h4>\n"
+                for l in section_buffer:
+                    html_output += f"<p style='color:#111; margin:5px 0;'>{l}</p>\n"
+                section_buffer = []
+
+            current_section = emoji
+            section_title = line
         else:
-            html_output += f"<p style='color:#111; margin:5px 0;'>{stripped}</p>\n"
+            section_buffer.append(line)
+
+    # Flush the final section
+    if current_section:
+        html_output += f"<h4 style='color:{section_colors[current_section]}; margin-top:20px;'>{section_title}</h4>\n"
+        for l in section_buffer:
+            html_output += f"<p style='color:#111; margin:5px 0;'>{l}</p>\n"
 
     # Embed uploaded image
     image_html = ""
@@ -123,7 +139,3 @@ 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)
-
-# Optional: Demonstration Heading
-st.markdown("<h4 style='color:#000; margin-top:20px;'>🌟 Greeting</h4>", unsafe_allow_html=True)
-st.write("Hey! This is how the greeting will render in HTML.")