Update app.py

app.py

@@ -45,14 +45,19 @@ def categorize_weather(description):
         return "Uncategorized", "🔍"
 
 # Function to get outfit suggestion using Groq's LLaMA model
-def get_outfit_suggestion(temperature, description, style, fabric, weather_category, weather_icon):
+def get_outfit_suggestion(temperature, description, style, fabric, weather_category, weather_icon, gender):
     # Initialize Groq's API
     try:
         client = groq.Groq(api_key=groq_api_key)  # Use the secret API key
         
-        # Adjust the prompt based on the weather category
-        prompt = f"The current weather is {description} with a temperature of {temperature}°C. The weather category is {weather_category}. Suggest an outfit. The user prefers a {style} style and {fabric} fabric."
-
+        # Adjust the prompt based on the weather category and gender
+        if gender == "Male":
+            prompt = f"The current weather is {description} with a temperature of {temperature}°C. The weather category is {weather_category}. Suggest a stylish outfit for a man who prefers a {style} style and {fabric} fabric."
+        elif gender == "Female":
+            prompt = f"The current weather is {description} with a temperature of {temperature}°C. The weather category is {weather_category}. Suggest a stylish outfit for a woman who prefers a {style} style and {fabric} fabric."
+        else:
+            prompt = f"The current weather is {description} with a temperature of {temperature}°C. The weather category is {weather_category}. Suggest a stylish outfit for someone who prefers a {style} style and {fabric} fabric."
+        
         # Use Groq's chat completion to get the text response
         response = client.chat.completions.create(
             messages=[{"role": "user", "content": prompt}],
@@ -67,7 +72,7 @@ def get_outfit_suggestion(temperature, description, style, fabric, weather_categ
 st.set_page_config(page_title="Weather-Based Outfit Suggestion", page_icon="🌤️", layout="wide")
 
 # Custom styles
-st.markdown("""
+st.markdown(""" 
     <style>
     .reportview-container {
         background: linear-gradient(135deg, #ffcc00, #ff7b00);
@@ -114,6 +119,9 @@ st.title("🌤️ Weather-Based Outfit Suggestion App")
 # User input
 city = st.text_input("Enter your location:", placeholder="E.g. Peshawar")
 
+# Gender selection
+gender = st.selectbox("Select your gender", ["Male", "Female"])
+
 # Style and fabric preferences
 style = st.selectbox("Select your preferred style", ["Casual", "Formal", "Sporty", "Business", "Chic"])
 fabric = st.selectbox("Select your preferred fabric", ["Cotton", "Linen", "Wool", "Polyester", "Silk", "Leather"])
@@ -131,8 +139,8 @@ if city:
         st.write(f"Weather: {description} {weather_icon}")
         st.write(f"Weather Category: {weather_category} {weather_icon}")
         
-        # Get outfit suggestion based on user preferences
-        outfit_suggestion, weather_icon = get_outfit_suggestion(temperature, description, style, fabric, weather_category, weather_icon)
+        # Get outfit suggestion based on user preferences and gender
+        outfit_suggestion, weather_icon = get_outfit_suggestion(temperature, description, style, fabric, weather_category, weather_icon, gender)
         
         if outfit_suggestion:
             # Display outfit suggestion
@@ -171,6 +179,8 @@ st.markdown(
 
 
 
+
+
 # import requests
 # import streamlit as st
 # import groq