Update app.py

app.py

@@ -45,18 +45,13 @@ 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, gender):
+def get_outfit_suggestion(temperature, description, style, fabric, weather_category, weather_icon):
     # 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 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."
+        # Adjust the prompt based on the weather category and custom style
+        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."
         
         # Use Groq's chat completion to get the text response
         response = client.chat.completions.create(
@@ -72,7 +67,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);
@@ -120,10 +115,15 @@ st.title("🌤️ Weather-Based Outfit Suggestion App")
 city = st.text_input("Enter your location:", placeholder="E.g. Peshawar")
 
 # Gender selection
-gender = st.selectbox("Select your gender", ["Male", "Female"])
+gender = st.selectbox("Select your gender", ["Male", "Female", "Other"])
 
-# Style and fabric preferences
-style = st.selectbox("Select your preferred style", ["Casual", "Formal", "Sporty", "Business", "Chic"])
+# Custom style input
+custom_style = st.text_input("Enter your personalized style (optional)", placeholder="E.g. Vintage, Sporty")
+
+# If no custom style is provided, default to the selected style
+style = custom_style if custom_style else st.selectbox("Select your preferred style", ["Casual", "Formal", "Sporty", "Business", "Chic"])
+
+# Fabric preference
 fabric = st.selectbox("Select your preferred fabric", ["Cotton", "Linen", "Wool", "Polyester", "Silk", "Leather"])
 
 if city:
@@ -139,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 and gender
-        outfit_suggestion, weather_icon = get_outfit_suggestion(temperature, description, style, fabric, weather_category, weather_icon, gender)
+        # Get outfit suggestion based on user preferences
+        outfit_suggestion, weather_icon = get_outfit_suggestion(temperature, description, style, fabric, weather_category, weather_icon)
         
         if outfit_suggestion:
             # Display outfit suggestion
@@ -181,6 +181,8 @@ st.markdown(
 
 
 
+
+
 # import requests
 # import streamlit as st
 # import groq