added more context to the LLM from the BLIP/CLIP analys

app.py

@@ -166,9 +166,16 @@ with gr.Blocks() as demo:
     # analysis_output = gr.Textbox(label="AI Analysis")
     advice_output = gr.Textbox(label="AI Advice")
 
+
+    # Store the latest analysis for chatbot context
+    latest_analysis = {"caption": "", "fashion_analysis": {}}
+
     def tryon_workflow(cloth, avatar):
         img = tryon_api(cloth, avatar)
         analysis = analyze_fashion_image(img)
+        # Save for chatbot context
+        latest_analysis["caption"] = analysis["caption"]
+        latest_analysis["fashion_analysis"] = analysis["fashion_analysis"]
         sys_prompt = 'You are a professional fashion stylist. Provide constructive, specific, and friendly advice about fit, colors, style, and accessories.'
         user_msg = f"Analyze and advise based on: {json.dumps(analysis)}"
         advice = llm_advise(sys_prompt, user_msg)
@@ -210,7 +217,11 @@ with gr.Blocks() as demo:
     # --- Chatbot ---
     def chatbot(message, history=None):
         sys_prompt = 'You are a helpful, up-to-date fashion stylist assistant.'
-        return llm_advise(sys_prompt, message)
+        # Add latest analysis as context if available
+        context = ""
+        if latest_analysis["caption"] or latest_analysis["fashion_analysis"]:
+            context = f"\n\nCurrent try-on analysis: {json.dumps(latest_analysis)}"
+        return llm_advise(sys_prompt, message + context)
 
     # --- Gradio UI: Tabs for all features ---
     with gr.Tab("Chatbot"):