Update app.py

app.py

@@ -1,141 +1,100 @@
 import streamlit as st
-import random
 import os
-import re
 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 actual key
+# --- Load Groq API Key ---
+GROQ_API_KEY = os.getenv("GROQ_API_KEY", "your_groq_api_key_here")
 client = Groq(api_key=GROQ_API_KEY)
 
-# Stickers for response
+# --- Cute Stickers ---
 stickers = [
-    "https://i.imgur.com/6KZyFxh.png",
-    "https://i.imgur.com/f1Qdqpv.png",
-    "https://i.imgur.com/UuUJqfL.png",
-    "https://i.imgur.com/bGMH89Y.png",
-    "https://i.imgur.com/y0ubkNe.png"
+    "https://media.giphy.com/media/3o6Zt481isNVuQI1l6/giphy.gif",
+    "https://media.giphy.com/media/l3q2XB76CaWPcSspC/giphy.gif",
+    "https://media.giphy.com/media/ICOgUNjpvO0PC/giphy.gif",
+    "https://media.giphy.com/media/26BRuo6sLetdllPAQ/giphy.gif"
 ]
 
-# Streamlit UI setup
-st.set_page_config(page_title="SkinSense AI", layout="centered")
-st.markdown("<h1 style='text-align:center; color:#5E35B1;'>🌸 SkinSense AI – Skincare Buddy</h1>", unsafe_allow_html=True)
-
-# Input form
-name = st.text_input("Your Name")
-gender = st.radio("Gender", ["Female", "Male", "Other"])
-age = st.slider("Age", 12, 100, 25)
-skin_concern = st.selectbox("Main Skin Concern", ["Acne", "Dryness", "Oiliness", "Pores", "Dullness", "Pigmentation", "Other"])
-description = st.text_area("Describe Your Skin Problem", placeholder="e.g. Small white pimples under skin for 2 weeks...")
-image = st.file_uploader("Or upload a picture of your skin", type=["jpg", "jpeg", "png"])
-duration = st.selectbox("How long has this been a concern?", ["Less than 1 week", "1-2 weeks", "More than 2 weeks"])
-sensitivity = st.radio("Is your skin sensitive?", ["Yes", "No", "Not Sure"])
-routine = st.text_input("Do you follow a routine?", placeholder="e.g. Facewash, moisturizer, sunscreen")
-submit = st.button("✨ Generate Skincare Advice")
-
-# Core function
-def skincare_advice(name, gender, age, skin_concern, description, duration, sensitivity, routine, image):
-    if not name.strip():
-        return "<span style='color:red;'>⚠ Please enter your name.</span>"
-
-    sticker = random.choice(stickers)
-
-    # Determine concern text
-    if description.strip():
-        concern_text = description.strip()
-    elif image is not None:
-        concern_text = "The user uploaded an image showing their skin issue. Please base your suggestions on typical visible skin concerns."
-    else:
-        concern_text = skin_concern
-
-    # Prompt for the model
+# --- Title ---
+st.title("💆‍♀️ SkinSense AI - Your Skincare Assistant")
+
+# --- Sidebar Input ---
+with st.sidebar:
+    st.header("📝 User Info")
+    name = st.text_input("Your Name")
+    gender = st.selectbox("Gender", ["Female", "Male", "Other"])
+    age = st.slider("Age", 10, 80, 25)
+    skin_concern = st.selectbox("Main Skin Concern", [
+        "Acne", "Dryness", "Oiliness", "Dark Spots", "Redness", "Wrinkles", "Dullness", "Other"
+    ])
+    description = st.text_area("Describe your skin condition (symptoms, triggers, etc.)")
+    duration = st.text_input("How long has this been happening?")
+    sensitivity = st.radio("Skin Sensitivity?", ["Yes", "No"])
+    routine = st.text_area("Current Skincare Routine (if any)")
+    image = st.file_uploader("Optional: Upload a photo of your skin (jpg/png)", type=["jpg", "png"])
+
+# --- Function to get response from Groq ---
+def skincare_advice(name, gender, age, skin_concern, description, duration, sensitivity, routine, image=None):
     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.
-    """
-
-    try:
-        response = client.chat.completions.create(
-            model="llama3-8b-8192",
-            messages=[{"role": "user", "content": prompt}]
-        )
-        raw = response.choices[0].message.content
-    except Exception as e:
-        return f"<span style='color:red;'>❌ Error: {str(e)}</span>"
-
-    # Format response into HTML
-    lines = raw.strip().split("\n")
-    html_output = ""
-    section_colors = {
-        "🌟": "#E53935", "🩺": "#D81B60", "📊": "#5E35B1",
-        "🧴": "#F57C00", "🌿": "#43A047", "💡": "#039BE5", "💬": "#6D4C41"
-    }
-
-    current_section = None
-    section_buffer = []
-
-    for line in lines:
-        line = line.strip()
-        if not line:
-            continue
-
-        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:
-            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 = ""
-    if image is not None:
-        mime = image.type
-        img_bytes = image.read()
-        img_b64 = base64.b64encode(img_bytes).decode()
-        image_html = f"<img src='data:{mime};base64,{img_b64}' style='max-width:100%; border-radius:12px; margin:15px 0;'>"
-
-    # Final HTML layout
-    final_html = f"""
-<div style='font-family:Arial, sans-serif; color:#111; background:#fff; padding:20px;
-            border-radius:12px; border:1px solid #ccc; box-shadow:0 2px 6px rgba(0,0,0,0.1);'>
-    <div style='display:flex; align-items:center; justify-content:space-between;'>
-        <h2 style='color:#000; margin:0;'>👤 <b>{name}</b> | {gender}, {age} yrs</h2>
-        <img src="{sticker}" style="width:70px; border-radius:10px;">
-    </div>
-    {image_html}
-    {html_output}
-</div>
+You are a skincare expert chatbot. A user named {name}, a {gender}, aged {age}, has the following skin concern: {skin_concern}.
+They described the issue as: {description}. The issue has been happening for: {duration}.
+Sensitive skin: {sensitivity}. Current routine: {routine}.
+Provide a friendly and professional skin analysis, possible causes, and a step-by-step personalized skincare routine.
+Include emojis to make it friendly. Use headings like "🌟 Greeting", "🩺 Skin Issue Summary", "📊 Possible Causes", "🧴 Skincare Routine", "🌿 Natural Remedies", "💡 Tips", "💬 Final Note".
+Don't recommend medicines.
 """
-
-    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)
+    completion = client.chat.completions.create(
+        model="llama3-8b-8192",
+        messages=[
+            {"role": "system", "content": "You are a friendly and expert skincare assistant."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return completion.choices[0].message.content
+
+# --- Generate Button ---
+if st.button("✨ Get My Skincare Plan"):
+    with st.spinner("Analyzing your skin profile..."):
+        raw = skincare_advice(name, gender, age, skin_concern, description, duration, sensitivity, routine, image)
+
+        # --- Parse and Beautify Output ---
+        lines = raw.strip().split("\n")
+        html_output = ""
+        section_colors = {
+            "🌟": "#E53935", "🩺": "#D81B60", "📊": "#5E35B1",
+            "🧴": "#F57C00", "🌿": "#43A047", "💡": "#039BE5", "💬": "#6D4C41"
+        }
+
+        current_section = None
+        section_title = ""
+        section_buffer = []
+
+        for line in lines:
+            line = line.strip()
+            if not line:
+                continue
+
+            emoji = line[:2]
+            if emoji in section_colors:
+                # Flush previous section
+                if current_section:
+                    html_output += f"<h4 style='color:{section_colors[current_section]}; margin-top:20px;'>{section_title}</h4>\n"
+                    for item in section_buffer:
+                        html_output += f"<p style='margin:4px 0;'>{item}</p>\n"
+                    section_buffer = []
+
+                current_section = emoji
+                section_title = line
+            else:
+                section_buffer.append(line)
+
+        # Flush final section
+        if current_section:
+            html_output += f"<h4 style='color:{section_colors[current_section]}; margin-top:20px;'>{section_title}</h4>\n"
+            for item in section_buffer:
+                html_output += f"<p style='margin:4px 0;'>{item}</p>\n"
+
+        # --- Display Final HTML Output ---
+        st.markdown(html_output, unsafe_allow_html=True)
+
+        # --- Show Sticker ---
+        st.image(random.choice(stickers), use_column_width=True)