Update app.py

app.py

@@ -147,13 +147,11 @@ MENU_CONFIG = {
     },
     'product_inquiry': {
         'name': 'Product Inquiry Menu',
-        'valid_options': ['1', '2', '3', '4', '5'],
+        'valid_options': ['1', '2', '3'],
         'option_descriptions': {
             '1': 'Talk to Veterinary Consultant',
             '2': 'Inquire about Product Availability',
-            '3': 'View Product Image',
-            '4': 'Download Detailed PDF',
-            '5': 'Back to Main Menu'
+            '3': 'Back to Main Menu'
         }
     },
     'ai_chat': {
@@ -647,22 +645,18 @@ context_manager = VeterinaryContextManager()
 # Enhanced product response with veterinary domain expertise
 def generate_veterinary_product_response(product_info: Dict[str, Any], user_context: Dict[str, Any]) -> str:
     """Generate comprehensive veterinary product response with intelligent information handling"""
-    
     def clean_text(text):
         if pd.isna(text) or text is None:
             return "Not specified"
         return str(text).strip()
-    
     # Extract product details
     product_name = clean_text(product_info.get('Product Name', ''))
     product_type = clean_text(product_info.get('Type', ''))
     category = clean_text(product_info.get('Category', ''))
     indications = clean_text(product_info.get('Indications', ''))
-    
     # Check for PDF link in the CSV data
     pdf_link = ""
     try:
-        # Load CSV data to check for PDF link
         csv_data = pd.read_csv('Veterinary.csv')
         product_row = csv_data[csv_data['Product Name'] == product_name]
         if not product_row.empty:
@@ -671,27 +665,15 @@ def generate_veterinary_product_response(product_info: Dict[str, Any], user_cont
                 pdf_link = brochure_link.strip()
     except Exception as e:
         logger.warning(f"Error checking PDF link for {product_name}: {e}")
-    
-    # Build the response
-    response = f"""🧪 *Name:* {product_name}
-📦 *Type:* {product_type}
-🏥 *Category:* {category}
-💊 *Used For:* {indications}"""
-    
-    # Add PDF link if available, in the requested format
+    response = f"""🧪 *Name:* {product_name}\n📦 *Type:* {product_type}\n🏥 *Category:* {category}\n💊 *Used For:* {indications}"""
     if pdf_link:
         response += f"\n\n📄 Product Brochure Available\n🔗 {product_name} PDF:\n{pdf_link}"
-    
-    # Add menu options
     response += f"""
-
-💬 *Available Actions:*
+\n💬 *Available Actions:*
 1️⃣ Talk to Veterinary Consultant
 2️⃣ Inquire About Availability
 3️⃣ Back to Main Menu
-
-💬 Select an option or ask about related products"""
-    
+\n💬 Select an option or ask about related products"""
     return response
 
 def clean_text_for_pdf(text: str) -> str:
@@ -2982,33 +2964,6 @@ async def handle_veterinary_product_followup(selection: str, from_number: str) -
             await handle_availability_inquiry(from_number, user_context)
             
         elif selection == '3':
-            # View Product Image
-            await send_product_with_image(from_number, current_product, user_context)
-            # Return to product inquiry menu
-            response_with_media = generate_veterinary_product_response_with_media(current_product, user_context)
-            send_whatsjet_message(from_number, response_with_media['text'])
-            context_manager.update_context(
-                from_number,
-                current_state='product_inquiry',
-                current_menu='product_inquiry',
-                current_menu_options=list(MENU_CONFIG['product_inquiry']['option_descriptions'].values())
-            )
-            
-        elif selection == '4':
-            # Download Detailed PDF
-            await send_enhanced_pdf(from_number, current_product)
-            # Return to product inquiry menu
-            response_with_media = generate_veterinary_product_response_with_media(current_product, user_context)
-            send_whatsjet_message(from_number, response_with_media['text'])
-            context_manager.update_context(
-                from_number,
-                current_state='product_inquiry',
-                current_menu='product_inquiry',
-                current_menu_options=list(MENU_CONFIG['product_inquiry']['option_descriptions'].values())
-            )
-            
-        elif selection == '5':
-            # Back to Main Menu
             welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
             send_whatsjet_message(from_number, welcome_msg)
             context_manager.update_context(
@@ -3019,7 +2974,7 @@ async def handle_veterinary_product_followup(selection: str, from_number: str) -
             )
             return
         else:
-            send_whatsjet_message(from_number, "❌ Invalid selection. Please choose 1, 2, 3, 4, or 5.")
+            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}")
@@ -3365,7 +3320,7 @@ def get_menu_validation_message(current_state: str, user_context: dict) -> str:
             return "❌ No products available in this category. Please type 'main' to return to main menu."
     
     elif current_state == 'product_inquiry':
-        return "❌ Invalid selection for Product Options. Please choose:\n1️⃣ Talk to Veterinary Consultant\n2️⃣ Inquire about Product Availability\n3️⃣ View Product Image\n4️⃣ Download Detailed PDF\n5️⃣ Back to Main Menu"
+        return "❌ Invalid selection for Product Options. Please choose:\n1️⃣ Talk to Veterinary Consultant\n2️⃣ Inquire about Product Availability\n3️⃣ Back to Main Menu"
     
     elif current_state == 'ai_chat_mode':
         return "💬 *You're in AI Chat mode. Ask me anything about veterinary care!*\n\nType 'main' to return to main menu."
@@ -3402,7 +3357,7 @@ def is_valid_menu_selection(selection: str, current_state: str, user_context: di
         return 1 <= selection_num <= total_products
     
     elif current_state == 'product_inquiry':
-        return 1 <= selection_num <= 5
+        return 1 <= selection_num <= 3
     
     elif current_state == 'ai_chat_mode':
         return mapped_selection == 'main'
@@ -3573,22 +3528,16 @@ def generate_veterinary_product_response_with_media(product_info: Dict[str, Any]
     Generate comprehensive veterinary product response with media information
     Returns a dictionary with text response and media info
     """
-    
     def clean_text(text):
         if pd.isna(text) or text is None:
             return "Not specified"
         return str(text).strip()
-    
-    # Extract product details
     product_name = clean_text(product_info.get('Product Name', ''))
     product_type = clean_text(product_info.get('Type', ''))
     category = clean_text(product_info.get('Category', ''))
     indications = clean_text(product_info.get('Indications', ''))
-    
-    # Check for PDF link in the CSV data
     pdf_link = ""
     try:
-        # Load CSV data to check for PDF link
         csv_data = pd.read_csv('Veterinary.csv')
         product_row = csv_data[csv_data['Product Name'] == product_name]
         if not product_row.empty:
@@ -3597,33 +3546,17 @@ def generate_veterinary_product_response_with_media(product_info: Dict[str, Any]
                 pdf_link = brochure_link.strip()
     except Exception as e:
         logger.warning(f"Error checking PDF link for {product_name}: {e}")
-    
-    # Build the response
-    response_text = f"""🧪 *Name:* {product_name}
-📦 *Type:* {product_type}
-🏥 *Category:* {category}
-💊 *Used For:* {indications}"""
-    
-    # Add PDF link if available
+    response_text = f"""🧪 *Name:* {product_name}\n📦 *Type:* {product_type}\n🏥 *Category:* {category}\n💊 *Used For:* {indications}"""
     if pdf_link:
         response_text += f"\n\n📄 Product Brochure Available\n🔗 {product_name} PDF:\n{pdf_link}"
-    
-    # Add menu options
     response_text += f"""
-
-💬 *Available Actions:*
+\n💬 *Available Actions:*
 1️⃣ Talk to Veterinary Consultant
 2️⃣ Inquire About Availability
-3️⃣ View Product Image
-4️⃣ Download Detailed PDF
-5️⃣ Back to Main Menu
-
-💬 Select an option or ask about related products"""
-    
-    # Check for product image
+3️⃣ Back to Main Menu
+\n💬 Select an option or ask about related products"""
     image_path = get_product_image_path(product_name)
     has_image = image_path is not None and os.path.exists(image_path)
-    
     return {
         'text': response_text,
         'has_image': has_image,