Update app.py

app.py

@@ -39,7 +39,7 @@ def problem_categorization(problem_description):
     for department, terms in department_keywords.items():
         if any(term in problem_description for term in terms):
             return department
-    return "General"
+    return "NA"
 
 # Define the chatbot model
 # os.environ["GROQ_API_KEY"] = "GROQQQ"
@@ -86,6 +86,8 @@ Answer:You can check the PNR status of your train ticket by visiting https://www
  
 replace /PNR/ with the pnr number if the user provided otherwise use PNR
 
+Question: How do i cancel my ticket?
+Answer:Click on the "Cancel Ticket" option, and follow the prompts to complete the cancellation process.Please note that canceling a ticket is subject to IRCTC's cancellation policy and specific conditions.For more information on IRCTC's cancellation policy, please refer to: https://contents.irctc.co.in/en/CancellationRulesforIRCTCTrain.pdf You can also cancel your ticket by visiting the IRCTC e-Ticket Cancellation Website: https://contents.irctc.co.in/en/eticketCancel.html
 
 Question: What should I do if my train is delayed?
 Answer: We sincerely apologize for the inconvenience caused by the delay of your train. We understand how frustrating this situation can be. Please stay informed by checking the live train status through https://enquiry.indianrail.gov.in/mntes/ or mobile app. If the delay is significant and you decide not to continue your journey, you may be eligible for a full refund. Additionally, if the delay exceeds three hours, you can file for a refund even after the scheduled departure time. We deeply regret the disruption to your travel plans and appreciate your understanding. Please keep your ticket and PNR number handy if you need further assistance 
@@ -137,7 +139,10 @@ def handle_input(image, text, session_id):
         # Handle image input
         image_desc = image_description(image)
         category = problem_categorization(image_desc)
-        response = f"Image description: {image_desc}. Categorized as: {category}."
+        if category=='NA':
+            response="Please upload a better image"
+        else:
+            response=f"Your request has been forwarded to the {category} department"
     else:
         # Handle text input
         response = chain.invoke(