modifications

app.py

@@ -6,7 +6,6 @@ from torchvision import transforms, models
 from segment_anything import sam_model_registry, SamPredictor
 from io import BytesIO
 import base64
-import requests
 from google.generativeai import configure, GenerativeModel
 
 # Configure Gemini API
@@ -51,14 +50,17 @@ def analyze_garment(image, style_pref=None, feedback=None, generate_variations=F
 
     style_input = f"\nStyle preference: {style_pref}" if style_pref else ""
     feedback_input = f"\nUser feedback: {feedback}" if feedback else ""
-    variation_input = "\nGenerate 3 variations of this design based on the given preferences." if generate_variations else ""
+    variation_input = "\nGenerate 3 creative variations based on this mixed or hybrid style." if generate_variations else ""
 
     prompt = {
         "parts": [
             {
                 "text": (
-                    "Analyze the garment in this image, describing its style, fabric, and design elements. "
-                    "Suggest the best occasions to wear it and recommend complementary fashion pieces." +
+                    "You are a professional fashion consultant. "
+                    "Analyze the uploaded garment image and describe its style, fabric, and design elements. "
+                    "Then, suggest complementary pieces and occasions to wear it. "
+                    "The user might provide unique or combined style preferences such as 'elegant streetwear' or 'casual luxury'. "
+                    "In such cases, blend both styles and explain how to balance or merge them." +
                     style_input + feedback_input + variation_input
                 )
             },
@@ -94,6 +96,8 @@ if uploaded_file:
         st.session_state.processed_image = None
     if "segmented_image" not in st.session_state:
         st.session_state.segmented_image = None
+    if "last_analysis_result" not in st.session_state:
+        st.session_state.last_analysis_result = None
 
     if st.button("⚙️ Preprocess Image"):
         st.session_state.processed_image = preprocess_image(image)
@@ -107,12 +111,11 @@ if uploaded_file:
     # 🧑‍🤝‍🧑 Human-AI Interaction
     st.markdown("### 🧠 Customize Your Style & Get AI Suggestions")
     style_pref = st.text_input("🔖 Describe your style preference (e.g., elegant, streetwear, minimalist)")
-    feedback = st.text_input("🗣️ Provide feedback to AI (e.g., Make it more vibrant, Use denim)")
-
     generate_variations = st.checkbox("🎨 Generate design variations")
     
     if st.button("🤖 Analyze & Collaborate with AI"):
-        result = analyze_garment(image, style_pref, feedback, generate_variations)
+        result = analyze_garment(image, style_pref)
+        st.session_state.last_analysis_result = result
         st.success(result)
 
         # Fabric suggestion extraction (Optional basic parsing)
@@ -120,4 +123,13 @@ if uploaded_file:
         for line in result.splitlines():
             if "fabric" in line.lower() or "material" in line.lower():
                 st.info(line)
-            
+
+    # Feedback AFTER analysis
+    if st.session_state.last_analysis_result:
+        st.markdown("### 🗣️ Give Feedback to Improve AI Suggestions")
+        feedback = st.text_input("What would you like to adjust or improve?", key="feedback_input")
+
+        if st.button("🔁 Refine Based on Feedback"):
+            updated_result = analyze_garment(image, style_pref, feedback, generate_variations)
+            st.session_state.last_analysis_result = updated_result
+            st.success(updated_result)