added object detection imporved

QA_bot.py

@@ -44,7 +44,7 @@ def tyre_synap_bot(filter_agent,image_file_path):
         with st.chat_message("assistant"):
             message_placeholder = st.empty()
             full_response = ""
-            if 'mask' in ai_response['output'] or 'predicted_image' in ai_response['output']:
+            if ('Segmentation image' in ai_response['output']) or ('predicted_image' in ai_response['output']):
                 display_mask_image('final_mask.png')
             
             for chunk in re.split(r'(\s+)', response):

tool_utils/yolo_world.py

@@ -83,7 +83,7 @@ class YoloWorld:
                          )
         
         cv2.imwrite('final_mask.png', cv2.cvtColor(detected_image,cv2.COLOR_BGR2RGB))
-        return "Predicted image : final_mask.jpg . Details :{}".format(processed_predictions[0])
+        return "Predicted image : final_mask.png . Details :{}".format(processed_predictions[0])
                 
 # if __name__ == "__main__":
 #     yolo = YoloWorld()