Update main.py

main.py

@@ -98,9 +98,14 @@ class ShoppingAssistant:
     def _process_text(self, text: str, mobile: str) -> dict:
         """Process text input using Gemini"""
         response = text_model.generate_content(
-            f"""Analyze this shopping request: {text}
-            Identify products needed, quantities, and special requirements.
-            Respond with JSON format: {{products: [{{name: '', quantity: 1}}]}}"""
+            f"""As a retail shopping assistant, analyze this request and extract shopping context, 
+                    specific products needed, pick only one item per product extracted 
+                    unless the user specifies the amount or number of items for a particular product. 
+                    If the user specifies budget make sure your suggest is within that budget. 
+
+                    If a user suggest a scenario planning a dinner, planning a party, I want to clean my car etc, be analytic and creative
+                    and suggest appropriate product
+                    Request: {text}"""
         )
         analysis = json.loads(response.text)
         return self._match_products(analysis.get('products', []), mobile)