Update app.py

app.py

@@ -4102,6 +4102,11 @@ async def handle_veterinary_product_followup(selection: str, from_number: str) -
             send_whatsjet_message(from_number, "❌ No product selected. Please search for a product first.")
             return
         
+        # Validate that the selection is a valid follow-up option (1, 2, or 3)
+        if selection not in ['1', '2', '3']:
+            send_whatsjet_message(from_number, "❌ Invalid selection. Please choose 1, 2, or 3.")
+            return
+        
         if selection == '1':
             # Talk to Veterinary Consultant
             product_name = current_product.get('Product Name', 'the selected product')
@@ -4136,9 +4141,6 @@ async def handle_veterinary_product_followup(selection: str, from_number: str) -
                 current_menu_options=list(MENU_CONFIG['main_menu']['option_descriptions'].values())
             )
             return
-        else:
-            send_whatsjet_message(from_number, "❌ Invalid selection. Please choose 1, 2, or 3.")
-            return
     except Exception as e:
         logger.error(f"Error in product follow-up: {e}")
         user_context = context_manager.get_context(from_number)